diff options
Diffstat (limited to 'switchctl.c')
| -rw-r--r-- | switchctl.c | 433 |
1 files changed, 433 insertions, 0 deletions
diff --git a/switchctl.c b/switchctl.c new file mode 100644 index 0000000..1f119bc --- /dev/null +++ b/switchctl.c @@ -0,0 +1,433 @@ +/* + * rhctl + * + * Copyright (C) 2009 Christian Pointner <equinox@spreadspace.org> + * + * This file is part of rhctl. + * + * rhctl is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * any later version. + * + * rhctl is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with rhctl. If not, see <http://www.gnu.org/licenses/>. + */ + +#include "datatypes.h" + +#include <stdlib.h> +#include <stdio.h> +#include <string.h> +#include <errno.h> + +#include "log.h" +#include "sig_handler.h" +#include "options.h" + +#include "command_queue.h" +#include "client_list.h" +#include "key_value_storage.h" + +#include "daemon.h" +#include "utils.h" + +int send_command(int switch_fd, cmd_t* cmd) +{ + if(!cmd) + return -1; + + char* c = NULL; + switch(cmd->cmd) { + case SWITCH: c = cmd->param; break; + case STATUS: c = "*0SL"; break; + } + + if(c == NULL) + return 0; + + int ret = send_string(switch_fd, c); + if(ret > 0) { + cmd_sent(cmd); + return 0; + } + + return ret; +} + +int send_response(int fd, const char* response) +{ + if(!response) + return -1; + + int ret = send_string(fd, response); + do { + ret = write(fd, "\n", 1); + } while(!ret || (ret == -1 && errno == EINTR)); + + if(ret > 0) + return 0; + + return ret; +} + +int process_cmd(const char* cmd, int fd, cmd_t **cmd_q, client_t* client_lst) +{ + log_printf(DEBUG, "processing command from %d", fd); + + if(!cmd_q || !cmd) + return -1; + + cmd_id_t cmd_id; + if(!strncmp(cmd, "switch", 6)) + cmd_id = SWITCH; + else if(!strncmp(cmd, "status", 6)) + cmd_id = STATUS; + else if(!strncmp(cmd, "log", 3)) + cmd_id = LOG; + else if(!strncmp(cmd, "listen", 6)) { + cmd_id = LISTEN; + } + else { + log_printf(WARNING, "unknown command '%s'", cmd); + return 0; + } + char* param = strchr(cmd, ' '); + if(param) + param++; + + if(cmd_id == SWITCH) { + char* resp; + asprintf(&resp, "Request: %s", cmd); + if(resp) { + char* linefeed = strchr(resp, '\n'); + if(linefeed) linefeed[0] = 0; + client_t* client; + int listener_cnt = 0; + for(client = client_lst; client; client = client->next) + if(client->request_listener && client->fd != fd) { + send_response(client->fd, resp); + listener_cnt++; + } + free(resp); + log_printf(DEBUG, "sent request to %d additional listeners", listener_cnt); + } +// else silently ignore memory alloc error + } + + switch(cmd_id) { + case SWITCH: + case STATUS: { + int ret = cmd_push(cmd_q, fd, cmd_id, param); + if(ret) + return ret; + + log_printf(NOTICE, "command: %s", cmd); + break; + } + case LOG: { + if(param && param[0]) + log_printf(NOTICE, "ext msg: %s", param); + else + log_printf(DEBUG, "ignoring empty ext log message"); + break; + } + case LISTEN: { + client_t* listener = client_find(client_lst, fd); + if(listener) { + if(!param) { + listener->status_listener = 1; + listener->error_listener = 1; + listener->request_listener = 1; + } + else { + if(!strncmp(param, "status", 6)) + listener->status_listener = 1; + else if(!strncmp(param, "error", 5)) + listener->error_listener = 1; + else if(!strncmp(param, "request", 7)) + listener->request_listener = 1; + else { + log_printf(DEBUG, "unkown listener type '%s'", param); + break; + } + } + log_printf(DEBUG, "listener %d requests %s messages", fd, param ? param:"all"); + } + else { + log_printf(ERROR, "unable to add listener %d", fd); + } + break; + } + } + + return 0; +} + +int process_switch(int switch_fd) +{ + int ret = 0; + struct timeval tv; + fd_set fds; + FD_ZERO(&fds); + FD_SET(switch_fd, &fds); + char buf[1]; + + for(;;) { + tv.tv_sec = 0; + tv.tv_usec = 0; + ret = select(switch_fd+1, &fds, NULL, NULL, &tv); + if(!ret) + return 0; + else if(ret < 0) + return ret; + + ret = read(switch_fd, buf, 1); + if(!ret) + return 2; + if(ret == -1 && errno == EAGAIN) + return 0; + else if(ret < 0) + break; + } + + return ret; +} + +int main_loop(int switch_fd, int cmd_listen_fd) +{ + log_printf(NOTICE, "entering main loop"); + + fd_set readfds, tmpfds; + FD_ZERO(&readfds); + FD_SET(switch_fd, &readfds); + FD_SET(cmd_listen_fd, &readfds); + int max_fd = switch_fd > cmd_listen_fd ? switch_fd : cmd_listen_fd; + cmd_t* cmd_q = NULL; + client_t* client_lst = NULL; + + int sig_fd = signal_init(); + if(sig_fd < 0) + return -1; + FD_SET(sig_fd, &readfds); + max_fd = (max_fd < sig_fd) ? sig_fd : max_fd; + + struct timeval timeout; + int return_value = 0; + while(!return_value) { + memcpy(&tmpfds, &readfds, sizeof(tmpfds)); + + timeout.tv_sec = 0; + timeout.tv_usec = 200000; + int ret = select(max_fd+1, &tmpfds, NULL, NULL, &timeout); + if(ret == -1 && errno != EINTR) { + log_printf(ERROR, "select returned with error: %s", strerror(errno)); + return_value = -1; + break; + } + if(ret == -1) + continue; + if(!ret) { + if(cmd_q && cmd_has_expired(*cmd_q)) { + log_printf(ERROR, "last command expired"); + cmd_pop(&cmd_q); + } + else + continue; + } + + if(FD_ISSET(sig_fd, &tmpfds)) { + if(signal_handle()) { + return_value = 1; + break; + } + } + + if(FD_ISSET(switch_fd, &tmpfds)) { + return_value = process_switch(switch_fd); + if(return_value) + break; + } + + if(FD_ISSET(cmd_listen_fd, &tmpfds)) { + int new_fd = accept(cmd_listen_fd, NULL, NULL); + if(new_fd < 0) { + log_printf(ERROR, "accept returned with error: %s", strerror(errno)); + return_value = -1; + break; + } + log_printf(DEBUG, "new command connection (fd=%d)", new_fd); + FD_SET(new_fd, &readfds); + max_fd = (max_fd < new_fd) ? new_fd : max_fd; + fcntl(new_fd, F_SETFL, O_NONBLOCK); + client_add(&client_lst, new_fd); + } + + client_t* lst = client_lst; + while(lst) { + if(FD_ISSET(lst->fd, &tmpfds)) { + return_value = nonblock_recvline(&(lst->buffer), lst->fd, &cmd_q, client_lst); + if(return_value == 2) { + log_printf(DEBUG, "removing closed command connection (fd=%d)", lst->fd); + client_t* deletee = lst; + lst = lst->next; + FD_CLR(deletee->fd, &readfds); + client_remove(&client_lst, deletee->fd); + return_value = 0; + continue; + } + if(return_value) + break; + + } + lst = lst->next; + } + + if(cmd_q && !cmd_q->sent) + send_command(switch_fd, cmd_q); + } + + cmd_clear(&cmd_q); + client_clear(&client_lst); + signal_stop(); + return return_value; +} + +int main(int argc, char* argv[]) +{ + log_init(); + + options_t opt; + int ret = options_parse(&opt, argc, argv); + if(ret) { + if(ret > 0) { + fprintf(stderr, "syntax error near: %s\n\n", argv[ret]); + } + if(ret == -2) { + fprintf(stderr, "memory error on options_parse, exiting\n"); + } + + if(ret != -2) + options_print_usage(); + + options_clear(&opt); + log_close(); + exit(ret); + } + string_list_element_t* tmp = opt.log_targets_.first_; + if(!tmp) { + log_add_target("syslog:3,rhctl,daemon"); + } + else { + while(tmp) { + ret = log_add_target(tmp->string_); + if(ret) { + switch(ret) { + case -2: fprintf(stderr, "memory error on log_add_target, exitting\n"); break; + case -3: fprintf(stderr, "unknown log target: '%s', exitting\n", tmp->string_); break; + case -4: fprintf(stderr, "this log target is only allowed once: '%s', exitting\n", tmp->string_); break; + default: fprintf(stderr, "syntax error near: '%s', exitting\n", tmp->string_); break; + } + + options_clear(&opt); + log_close(); + exit(ret); + } + tmp = tmp->next_; + } + } + log_printf(NOTICE, "just started..."); + options_parse_post(&opt); + + priv_info_t priv; + if(opt.username_) + if(priv_init(&priv, opt.username_, opt.groupname_)) { + options_clear(&opt); + log_close(); + exit(-1); + } + + FILE* pid_file = NULL; + if(opt.pid_file_) { + pid_file = fopen(opt.pid_file_, "w"); + if(!pid_file) { + log_printf(WARNING, "unable to open pid file: %s", strerror(errno)); + } + } + + if(opt.chroot_dir_) + if(do_chroot(opt.chroot_dir_)) { + options_clear(&opt); + log_close(); + exit(-1); + } + if(opt.username_) + if(priv_drop(&priv)) { + options_clear(&opt); + log_close(); + exit(-1); + } + + if(opt.daemonize_) { + pid_t oldpid = getpid(); + daemonize(); + log_printf(INFO, "running in background now (old pid: %d)", oldpid); + } + + if(pid_file) { + pid_t pid = getpid(); + fprintf(pid_file, "%d", pid); + fclose(pid_file); + } + + int cmd_listen_fd = init_command_socket(opt.command_sock_); + if(cmd_listen_fd < 0) { + options_clear(&opt); + log_close(); + exit(-1); + } + + int switch_fd = 0; + for(;;) { + switch_fd = open(opt.switch_dev_, O_RDWR | O_NOCTTY); + if(switch_fd < 0) + ret = 2; + else { + ret = setup_tty(switch_fd); + if(ret) + ret = 2; + else + ret = main_loop(switch_fd, cmd_listen_fd); + } + + if(ret == 2) { + log_printf(ERROR, "%s error, trying to reopen in 5 seconds..", opt.switch_dev_); + if(switch_fd > 0) + close(switch_fd); + sleep(5); + } + else + break; + } + + close(cmd_listen_fd); + if(switch_fd > 0) + close(switch_fd); + + if(!ret) + log_printf(NOTICE, "normal shutdown"); + else if(ret < 0) + log_printf(NOTICE, "shutdown after error"); + else + log_printf(NOTICE, "shutdown after signal"); + + options_clear(&opt); + log_close(); + + return ret; +} |