1
0

confd.c 6.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329
  1. /*
  2. * clish_config_callback.c
  3. *
  4. *
  5. * Callback hook to execute config operations.
  6. */
  7. #include <stdio.h>
  8. #include <stdlib.h>
  9. #include <unistd.h>
  10. #include <sys/types.h>
  11. #include <sys/wait.h>
  12. #include <errno.h>
  13. #include <assert.h>
  14. #include <sys/socket.h>
  15. #include <sys/un.h>
  16. #include <string.h>
  17. #include <sys/select.h>
  18. #include <signal.h>
  19. #include "clish/private.h"
  20. #include "cliconf/conf.h"
  21. #include "cliconf/query.h"
  22. #include "cliconf/buf.h"
  23. #include "lub/argv.h"
  24. #include "lub/string.h"
  25. #define CONFD_SOCKET_PATH "/tmp/confd.socket"
  26. #define CONFD_CONFIG_PATH "/tmp/running-config"
  27. #ifndef UNIX_PATH_MAX
  28. #define UNIX_PATH_MAX 108
  29. #endif
  30. #define MAXMSG 1024
  31. /* Global signal vars */
  32. static volatile int sigterm = 0;
  33. static void sighandler(int signo);
  34. static char * process_query(int sock, cliconf_t * conf, char *str);
  35. int answer_send(int sock, char *command);
  36. static int dump_running_config(int sock, cliconf_t *conf, query_t *query);
  37. /*--------------------------------------------------------- */
  38. int main(int argc, char **argv)
  39. {
  40. int retval = 0;
  41. unsigned i;
  42. char *str;
  43. cliconf_t *conf;
  44. lub_bintree_t bufs;
  45. conf_buf_t *tbuf;
  46. /* Network vars */
  47. int sock;
  48. struct sockaddr_un laddr;
  49. struct sockaddr_un raddr;
  50. fd_set active_fd_set, read_fd_set;
  51. /* Signal vars */
  52. struct sigaction sig_act;
  53. sigset_t sig_set;
  54. /* Set signal handler */
  55. sigemptyset(&sig_set);
  56. sigaddset(&sig_set, SIGTERM);
  57. sigaddset(&sig_set, SIGINT);
  58. sigaddset(&sig_set, SIGQUIT);
  59. sig_act.sa_flags = 0;
  60. sig_act.sa_mask = sig_set;
  61. sig_act.sa_handler = &sighandler;
  62. sigaction(SIGTERM, &sig_act, NULL);
  63. sigaction(SIGINT, &sig_act, NULL);
  64. sigaction(SIGQUIT, &sig_act, NULL);
  65. /* Configuration tree */
  66. conf = cliconf_new("", 0);
  67. /* Initialize the tree of buffers */
  68. lub_bintree_init(&bufs,
  69. conf_buf_bt_offset(),
  70. conf_buf_bt_compare, conf_buf_bt_getkey);
  71. /* Create listen socket */
  72. if ((sock = socket(AF_UNIX, SOCK_STREAM, 0)) == -1) {
  73. fprintf(stderr, "Cannot create socket: %s\n", strerror(errno));
  74. /* syslog(LOG_ERR, "Cannot create socket: %s\n", strerror(errno));
  75. */ return -1;
  76. }
  77. unlink(CONFD_SOCKET_PATH);
  78. laddr.sun_family = AF_UNIX;
  79. strncpy(laddr.sun_path, CONFD_SOCKET_PATH, UNIX_PATH_MAX);
  80. laddr.sun_path[UNIX_PATH_MAX - 1] = '\0';
  81. if (bind(sock, (struct sockaddr *)&laddr, sizeof(laddr))) {
  82. fprintf(stderr, "Can't bind()\n");
  83. /* syslog(LOG_ERR, "Can't bind()\n");
  84. */ return -1;
  85. }
  86. listen(sock, 5);
  87. /* Initialize the set of active sockets. */
  88. FD_ZERO(&active_fd_set);
  89. FD_SET(sock, &active_fd_set);
  90. /* Main loop */
  91. while (!sigterm) {
  92. int num;
  93. /* Block until input arrives on one or more active sockets. */
  94. read_fd_set = active_fd_set;
  95. num = select(FD_SETSIZE, &read_fd_set, NULL, NULL, NULL);
  96. if (num < 0) {
  97. if (EINTR == errno)
  98. continue;
  99. break;
  100. }
  101. if (0 == num)
  102. continue;
  103. /* Get string from stdout */
  104. /* if (!fgets(str, sizeof(str), stdin)) {
  105. enough = 1;
  106. continue;
  107. }
  108. */
  109. /* Service all the sockets with input pending. */
  110. for (i = 0; i < FD_SETSIZE; ++i) {
  111. if (FD_ISSET(i, &read_fd_set)) {
  112. if (i == sock) {
  113. /* Connection request on original socket. */
  114. int new;
  115. socklen_t size;
  116. size = sizeof(raddr);
  117. new = accept(sock, (struct sockaddr *)
  118. &raddr, &size);
  119. if (new < 0) {
  120. fprintf(stderr, "accept");
  121. continue;
  122. }
  123. fprintf(stderr, "Server: connect %u\n", new);
  124. conf_buftree_remove(&bufs, new);
  125. tbuf = conf_buf_new(new);
  126. /* insert it into the binary tree for this conf */
  127. lub_bintree_insert(&bufs, tbuf);
  128. FD_SET(new, &active_fd_set);
  129. } else {
  130. int nbytes;
  131. /* Data arriving on an already-connected socket. */
  132. if ((nbytes = conf_buftree_read(&bufs, i)) <= 0) {
  133. close(i);
  134. FD_CLR(i, &active_fd_set);
  135. conf_buftree_remove(&bufs, i);
  136. continue;
  137. }
  138. while ((str = conf_buftree_parse(&bufs, i))) {
  139. char *answer;
  140. if (!(answer = process_query(i, conf, str)))
  141. answer = lub_string_dup("-e");
  142. lub_string_free(str);
  143. answer_send(i, answer);
  144. lub_string_free(answer);
  145. }
  146. }
  147. }
  148. }
  149. }
  150. /* Free resources */
  151. cliconf_delete(conf);
  152. /* delete each buf */
  153. while ((tbuf = lub_bintree_findfirst(&bufs))) {
  154. /* remove the buf from the tree */
  155. lub_bintree_remove(&bufs, tbuf);
  156. /* release the instance */
  157. conf_buf_delete(tbuf);
  158. }
  159. return retval;
  160. }
  161. static char * process_query(int sock, cliconf_t * conf, char *str)
  162. {
  163. unsigned i;
  164. int res;
  165. cliconf_t *iconf;
  166. cliconf_t *tmpconf;
  167. query_t *query;
  168. char *retval = NULL;
  169. query_op_t ret;
  170. /* Parse query */
  171. query = query_new();
  172. res = query_parse_str(query, str);
  173. if (res < 0) {
  174. query_free(query);
  175. return NULL;
  176. }
  177. printf("----------------------\n");
  178. printf("REQUEST: %s\n", str);
  179. /* query_dump(query);
  180. */
  181. /* Go through the pwd */
  182. iconf = conf;
  183. for (i = 0; i < query__get_pwdc(query); i++) {
  184. if (!
  185. (iconf =
  186. cliconf_find_conf(iconf, query__get_pwd(query, i), 0))) {
  187. iconf = NULL;
  188. break;
  189. }
  190. }
  191. if (!iconf) {
  192. printf("Unknown path\n");
  193. query_free(query);
  194. return NULL;
  195. }
  196. switch (query__get_op(query)) {
  197. case QUERY_OP_SET:
  198. if (cliconf_find_conf(iconf, query->line, 0)) {
  199. ret = QUERY_OP_OK;
  200. break;
  201. }
  202. cliconf_del_pattern(iconf, query->pattern);
  203. tmpconf = cliconf_new_conf(iconf, query->line, query->priority);
  204. if (!tmpconf) {
  205. ret = QUERY_OP_ERROR;
  206. break;
  207. }
  208. cliconf__set_splitter(tmpconf, query->splitter);
  209. ret = QUERY_OP_OK;
  210. break;
  211. case QUERY_OP_UNSET:
  212. cliconf_del_pattern(iconf, query->pattern);
  213. ret = QUERY_OP_OK;
  214. break;
  215. case QUERY_OP_DUMP:
  216. if (dump_running_config(sock, iconf, query))
  217. ret = QUERY_OP_ERROR;
  218. else
  219. ret = QUERY_OP_OK;
  220. break;
  221. default:
  222. ret = QUERY_OP_ERROR;
  223. break;
  224. }
  225. #ifdef DEBUG
  226. /* Print whole tree */
  227. cliconf_fprintf(conf, stdout, NULL, -1, 0);
  228. #endif
  229. /* Free resources */
  230. query_free(query);
  231. switch (ret) {
  232. case QUERY_OP_OK:
  233. lub_string_cat(&retval, "-o");
  234. break;
  235. case QUERY_OP_ERROR:
  236. lub_string_cat(&retval, "-e");
  237. break;
  238. default:
  239. lub_string_cat(&retval, "-e");
  240. break;
  241. };
  242. return retval;
  243. }
  244. /*
  245. * Signal handler for temination signals (like SIGTERM, SIGINT, ...)
  246. */
  247. static void sighandler(int signo)
  248. {
  249. sigterm = 1;
  250. }
  251. int answer_send(int sock, char *command)
  252. {
  253. return send(sock, command, strlen(command) + 1, MSG_NOSIGNAL);
  254. }
  255. static int dump_running_config(int sock, cliconf_t *conf, query_t *query)
  256. {
  257. FILE *fd;
  258. char *filename;
  259. int dupsock = -1;
  260. if ((filename = query__get_path(query))) {
  261. if (!(fd = fopen(filename, "w")))
  262. return -1;
  263. } else {
  264. if ((dupsock = dup(sock)) < 0)
  265. return -1;
  266. fd = fdopen(dupsock, "w");
  267. }
  268. if (!filename) {
  269. fprintf(fd, "-t\n");
  270. #ifdef DEBUG
  271. printf("ANSWER: -t\n");
  272. #endif
  273. }
  274. cliconf_fprintf(conf, fd, query__get_pattern(query),
  275. query__get_pwdc(query) - 1, 0);
  276. if (!filename) {
  277. fprintf(fd, "\n");
  278. #ifdef DEBUG
  279. printf("SEND DATA: \n");
  280. #endif
  281. }
  282. fclose(fd);
  283. return 0;
  284. }