#include "daemon.h"
+#include "autosample.h"
+
int init_command_socket(const char* path)
{
int fd = socket(AF_UNIX, SOCK_STREAM, 0);
return fd;
}
+void clear_fd(int fd)
+{
+ fd_set fds;
+ struct timeval tv;
+ FD_ZERO(&fds);
+ FD_SET(fd, &fds);
+ tv.tv_sec = 0;
+ tv.tv_usec = 50000;
+ for(;;) {
+ int ret = select(fd+1, &fds, NULL, NULL, &tv);
+ if(ret > 0) {
+ char buffer[100];
+ ret = read(fd, buffer, sizeof(buffer));
+ }
+ else
+ break;
+ }
+}
+
int send_command(int tty_fd, cmd_t* cmd)
{
if(!cmd)
char c;
switch(cmd->cmd) {
- case POWER: {
- c = 'A'; // get command from powerids[cmd->param]
+ case POWER_ON: {
+ if(!cmd->param)
+ return 0;
+ c = toupper(cmd->param[0]);
+ break;
+ }
+ case POWER_OFF: {
+ if(!cmd->param)
+ return 0;
+ c = tolower(cmd->param[0]);
break;
}
case SAMPLE: {
- c = 'T'; // get command from sampledevs[cmd->param]
+ if(!cmd->param)
+ return 0;
+ c = cmd->param[0];
break;
}
default: return 0;
return ret;
}
-int process_cmd(const char* cmd, int fd, cmd_t **cmd_q, client_t* client_lst)
+#define SEND_TO_LISTENER(LISTENER_TYPE, TYPE_NAME, FD, STRING) \
+ client_t* client; \
+ int listener_cnt = 0; \
+ for(client = client_lst; client; client = client->next) \
+ if(client->LISTENER_TYPE && client->fd != FD) { \
+ send_response(client->fd, STRING); \
+ listener_cnt++; \
+ } \
+ log_printf(DEBUG, "sent %s to %d additional listeners", TYPE_NAME,listener_cnt);
+
+
+int process_cmd(char* cmd, int fd, cmd_t **cmd_q, client_t* client_lst, options_t* opt)
{
log_printf(DEBUG, "processing command from %d", fd);
return -1;
cmd_id_t cmd_id;
- if(!strncmp(cmd, "power", 5))
- cmd_id = POWER;
+ if(!strncmp(cmd, "power on", 8)) {
+ cmd_id = POWER_ON;
+ cmd[5] = '_';
+ }
+ else if(!strncmp(cmd, "power off", 9)) {
+ cmd_id = POWER_OFF;
+ cmd[5] = '_';
+ }
else if(!strncmp(cmd, "sample", 6))
cmd_id = SAMPLE;
else if(!strncmp(cmd, "log", 3))
return 0;
}
char* param = strchr(cmd, ' ');
- if(param)
+ if(param)
param++;
- if(cmd_id == POWER || cmd_id == SAMPLE) {
+ if(cmd_id == POWER_ON || cmd_id == POWER_OFF) {
+ char* orig_param = param;
+ param = key_value_storage_find(&opt->powerids_, param);
+ if(!param) {
+ send_response(fd, "Error: invalid power id");
+ log_printf(WARNING, "invalid power id '%s' in command from %d", orig_param, fd);
+ }
+ }
+
+ if(cmd_id == SAMPLE) {
+ char* orig_param = param;
+ param = key_value_storage_find(&opt->sampledevs_, param);
+ if(!param) {
+ send_response(fd, "Error: invalid sample device");
+ log_printf(WARNING, "invalid sample device '%s' in command from %d", orig_param, fd);
+ }
+ }
+
+ if(cmd_id == POWER_ON || cmd_id == POWER_OFF || cmd_id == SAMPLE) {
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++;
- }
+ SEND_TO_LISTENER(request_listener, "request", fd, resp);
free(resp);
- log_printf(DEBUG, "sent request to %d additional listeners", listener_cnt);
}
// else silently ignore memory alloc error
}
switch(cmd_id) {
- case POWER:
+ case POWER_ON:
+ case POWER_OFF:
case SAMPLE: {
int ret = cmd_push(cmd_q, fd, cmd_id, param);
if(ret)
return 0;
}
-int nonblock_recvline(read_buffer_t* buffer, int fd, cmd_t** cmd_q, client_t* client_lst)
+int nonblock_readline(read_buffer_t* buffer, int fd, cmd_t** cmd_q, client_t* client_lst, options_t* opt)
{
int ret = 0;
for(;;) {
- ret = recv(fd, &buffer->buf[buffer->offset], 1, 0);
- if(!ret)
+ ret = read(fd, &buffer->buf[buffer->offset], 1);
+ if(!ret || (ret == -1 && errno == EBADF))
return 2;
- if(ret == -1 && errno == EAGAIN)
+ if(ret == -1 && (errno == EAGAIN || errno == EWOULDBLOCK))
return 0;
else if(ret < 0)
break;
if(buffer->buf[buffer->offset] == '\n') {
buffer->buf[buffer->offset] = 0;
- ret = process_cmd(buffer->buf, fd, cmd_q, client_lst);
+ ret = process_cmd(buffer->buf, fd, cmd_q, client_lst, opt);
buffer->offset = 0;
break;
}
if(buffer->offset > 0 && buffer->buf[buffer->offset-1] == '\r')
buffer->buf[buffer->offset-1] = 0;
- log_printf(NOTICE, "tty-firmware: %s", buffer->buf);
+ log_printf(NOTICE, "firmware: %s", buffer->buf);
int cmd_fd = -1;
if(cmd_q && (*cmd_q)) {
}
if(!strncmp(buffer->buf, "Error:", 6)) {
- client_t* client;
- int listener_cnt = 0;
- for(client = client_lst; client; client = client->next)
- if(client->error_listener && client->fd != cmd_fd) {
- send_response(client->fd, buffer->buf);
- listener_cnt++;
- }
- log_printf(DEBUG, "sent error to %d additional listeners", listener_cnt);
+ SEND_TO_LISTENER(error_listener, "error", cmd_fd, buffer->buf);
}
+ if(!strncmp(buffer->buf, "movement", 8)) {
+ SEND_TO_LISTENER(movement_listener, "movement", cmd_fd, buffer->buf);
+ }
+
+ if(!strncmp(buffer->buf, "PanicButton", 11)) {
+ SEND_TO_LISTENER(button_listener, "panic buttont", cmd_fd, buffer->buf);
+ }
+
+ if(!strncmp(buffer->buf, "Temp ", 5)) {
+ SEND_TO_LISTENER(temp_listener, "", cmd_fd, buffer->buf);
+ }
+
cmd_pop(cmd_q);
buffer->offset = 0;
return 0;
return ret;
}
-int main_loop(int tty_fd, int cmd_listen_fd)
+int main_loop(int tty_fd, int cmd_listen_fd, int autosample_fd, options_t* opt)
{
log_printf(NOTICE, "entering main loop");
+ clear_fd(tty_fd);
+ clear_fd(autosample_fd);
+
fd_set readfds, tmpfds;
FD_ZERO(&readfds);
FD_SET(tty_fd, &readfds);
FD_SET(cmd_listen_fd, &readfds);
int max_fd = tty_fd > cmd_listen_fd ? tty_fd : cmd_listen_fd;
+ FD_SET(autosample_fd, &readfds);
+ max_fd = (max_fd < autosample_fd) ? autosample_fd : max_fd;
cmd_t* cmd_q = NULL;
client_t* client_lst = NULL;
read_buffer_t tty_buffer;
tty_buffer.offset = 0;
+ read_buffer_t autosample_buffer;
+ autosample_buffer.offset = 0;
int sig_fd = signal_init();
if(sig_fd < 0)
client_add(&client_lst, new_fd);
}
+ if(FD_ISSET(autosample_fd, &tmpfds)) {
+ return_value = nonblock_readline(&autosample_buffer, autosample_fd, &cmd_q, client_lst, opt);
+ if(return_value == 2) {
+ log_printf(WARNING, "autosample not running, removing pipe to it");
+ FD_CLR(autosample_fd, &readfds);
+ return_value = 0;
+ continue;
+ }
+ if(return_value)
+ break;
+ }
+
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);
+ return_value = nonblock_readline(&(lst->buffer), lst->fd, &cmd_q, client_lst, opt);
if(return_value == 2) {
log_printf(DEBUG, "removing closed command connection (fd=%d)", lst->fd);
client_t* deletee = lst;
return ret;
}
- fd_set fds;
- struct timeval tv;
- FD_ZERO(&fds);
- FD_SET(fd, &fds);
- tv.tv_sec = 0;
- tv.tv_usec = 50000;
- for(;;) {
- ret = select(fd+1, &fds, NULL, NULL, &tv);
- if(ret > 0) {
- char buffer[100];
- ret = read(fd, buffer, sizeof(buffer));
- }
- else
- break;
- }
+ clear_fd(fd);
return 0;
}
}
}
log_printf(NOTICE, "just started...");
- options_parse_post(&opt);
+ if(options_parse_post(&opt)) {
+ options_clear(&opt);
+ log_close();
+ exit(-1);
+ }
priv_info_t priv;
if(opt.username_)
fprintf(pid_file, "%d", pid);
fclose(pid_file);
}
+
+ int autosample_fd = -1;
+ if(key_value_storage_length(&opt.autosampledevs_) > 0) {
+ log_printf(NOTICE, "starting autosample process");
+ autosample_fd = start_autosample_process(&opt);
+ if(autosample_fd == -1) {
+ options_clear(&opt);
+ log_close();
+ exit(1);
+ }
+ else if(autosample_fd <= 0) {
+ if(!autosample_fd)
+ log_printf(NOTICE, "autosample process normal shutdown");
+ else if(autosample_fd == -2)
+ log_printf(NOTICE, "autosample shutdown after signal");
+ else
+ log_printf(NOTICE, "autosample shutdown after error");
+
+ options_clear(&opt);
+ log_close();
+ exit(1);
+ }
+ }
int cmd_listen_fd = init_command_socket(opt.command_sock_);
if(cmd_listen_fd < 0) {
if(ret)
ret = 2;
else
- ret = main_loop(tty_fd, cmd_listen_fd);
+ ret = main_loop(tty_fd, cmd_listen_fd, autosample_fd, &opt);
}
if(ret == 2) {