X-Git-Url: https://git.octo.it/?p=rrdtool.git;a=blobdiff_plain;f=src%2Frrd_daemon.c;h=28912c5358f8d17214b34d7c681fe8bcf714b69b;hp=8c4f04226eebf648bda21f45c6051dbb0c940563;hb=3c3effad5f54f8c73260068e3ec32c56684243bb;hpb=5a614bba830fb6e5c458fe3f5c044214acff7ded diff --git a/src/rrd_daemon.c b/src/rrd_daemon.c index 8c4f042..28912c5 100644 --- a/src/rrd_daemon.c +++ b/src/rrd_daemon.c @@ -107,12 +107,26 @@ typedef enum PRIV_HIGH } socket_privilege; +typedef enum { RESP_ERR = -1, RESP_OK = 0 } response_code; + struct listen_socket_s { int fd; char addr[PATH_MAX + 1]; int family; socket_privilege privilege; + + /* state for BATCH processing */ + time_t batch_start; + int batch_cmd; + + /* buffered IO */ + char *rbuf; + off_t next_cmd; + off_t next_read; + + char *wbuf; + ssize_t wbuf_len; }; typedef struct listen_socket_s listen_socket_t; @@ -124,6 +138,7 @@ struct cache_item_s char **values; int values_num; time_t last_flush_time; + time_t last_update_stamp; #define CI_FLAGS_IN_TREE (1<<0) #define CI_FLAGS_IN_QUEUE (1<<1) int flags; @@ -150,11 +165,13 @@ typedef enum queue_side_e queue_side_t; /* max length of socket command or response */ #define CMD_MAX 4096 +#define RBUF_SIZE (CMD_MAX*2) /* * Variables */ static int stay_foreground = 0; +static uid_t daemon_uid; static listen_socket_t *listen_fds = NULL; static size_t listen_fds_num = 0; @@ -322,86 +339,175 @@ static int remove_pidfile (void) /* {{{ */ return (errno); } /* }}} int remove_pidfile */ -static ssize_t sread (int fd, void *buffer_void, size_t buffer_size) /* {{{ */ +static char *next_cmd (listen_socket_t *sock, ssize_t *len) /* {{{ */ { - char *buffer; - size_t buffer_used; - size_t buffer_free; - ssize_t status; + char *eol; - buffer = (char *) buffer_void; - buffer_used = 0; - buffer_free = buffer_size; + eol = memchr(sock->rbuf + sock->next_cmd, '\n', + sock->next_read - sock->next_cmd); - while (buffer_free > 0) + if (eol == NULL) { - status = read (fd, buffer + buffer_used, buffer_free); - if ((status < 0) && ((errno == EAGAIN) || (errno == EINTR))) - continue; - - if (status < 0) - return (-1); + /* no commands left, move remainder back to front of rbuf */ + memmove(sock->rbuf, sock->rbuf + sock->next_cmd, + sock->next_read - sock->next_cmd); + sock->next_read -= sock->next_cmd; + sock->next_cmd = 0; + *len = 0; + return NULL; + } + else + { + char *cmd = sock->rbuf + sock->next_cmd; + *eol = '\0'; - if (status == 0) - return (0); + sock->next_cmd = eol - sock->rbuf + 1; - assert ((0 > status) || (buffer_free >= (size_t) status)); + if (eol > sock->rbuf && *(eol-1) == '\r') + *(--eol) = '\0'; /* handle "\r\n" EOL */ - buffer_free = buffer_free - status; - buffer_used = buffer_used + status; + *len = eol - cmd; - if (buffer[buffer_used - 1] == '\n') - break; + return cmd; } - assert (buffer_used > 0); + /* NOTREACHED */ + assert(1==0); +} + +/* add the characters directly to the write buffer */ +static int add_to_wbuf(listen_socket_t *sock, char *str, size_t len) /* {{{ */ +{ + char *new_buf; + + assert(sock != NULL); - if (buffer[buffer_used - 1] != '\n') + new_buf = realloc(sock->wbuf, sock->wbuf_len + len + 1); + if (new_buf == NULL) { - errno = ENOBUFS; - return (-1); + RRDD_LOG(LOG_ERR, "add_to_wbuf: realloc failed"); + return -1; } - buffer[buffer_used - 1] = 0; + strncpy(new_buf + sock->wbuf_len, str, len + 1); + + sock->wbuf = new_buf; + sock->wbuf_len += len; - /* Fix network line endings. */ - if ((buffer_used > 1) && (buffer[buffer_used - 2] == '\r')) + return 0; +} /* }}} static int add_to_wbuf */ + +/* add the text to the "extra" info that's sent after the status line */ +static int add_response_info(listen_socket_t *sock, char *fmt, ...) /* {{{ */ +{ + va_list argp; + char buffer[CMD_MAX]; + int len; + + if (sock == NULL) return 0; /* journal replay mode */ + if (sock->batch_start) return 0; /* no extra info returned when in BATCH */ + + va_start(argp, fmt); +#ifdef HAVE_VSNPRINTF + len = vsnprintf(buffer, sizeof(buffer)-1, fmt, argp); +#else + len = vsprintf(buffer, fmt, argp); +#endif + va_end(argp); + if (len < 0) { - buffer_used--; - buffer[buffer_used - 1] = 0; + RRDD_LOG(LOG_ERR, "add_response_info: vnsprintf failed"); + return -1; } - return (buffer_used); -} /* }}} ssize_t sread */ + return add_to_wbuf(sock, buffer, len); +} /* }}} static int add_response_info */ -static ssize_t swrite (int fd, const void *buf, size_t count) /* {{{ */ +static int count_lines(char *str) /* {{{ */ { - const char *ptr; - size_t nleft; - ssize_t status; + int lines = 0; - /* special case for journal replay */ - if (fd < 0) return 0; + if (str != NULL) + { + while ((str = strchr(str, '\n')) != NULL) + { + ++lines; + ++str; + } + } - ptr = (const char *) buf; - nleft = count; + return lines; +} /* }}} static int count_lines */ - while (nleft > 0) +/* send the response back to the user. + * returns 0 on success, -1 on error + * write buffer is always zeroed after this call */ +static int send_response (listen_socket_t *sock, response_code rc, + char *fmt, ...) /* {{{ */ +{ + va_list argp; + char buffer[CMD_MAX]; + int lines; + ssize_t wrote; + int rclen, len; + + if (sock == NULL) return rc; /* journal replay mode */ + + if (sock->batch_start) { - status = write (fd, (const void *) ptr, nleft); + if (rc == RESP_OK) + return rc; /* no response on success during BATCH */ + lines = sock->batch_cmd; + } + else if (rc == RESP_OK) + lines = count_lines(sock->wbuf); + else + lines = -1; + + rclen = sprintf(buffer, "%d ", lines); + va_start(argp, fmt); +#ifdef HAVE_VSNPRINTF + len = vsnprintf(buffer+rclen, sizeof(buffer)-rclen-1, fmt, argp); +#else + len = vsprintf(buffer+rclen, fmt, argp); +#endif + va_end(argp); + if (len < 0) + return -1; - if ((status < 0) && ((errno == EAGAIN) || (errno == EINTR))) - continue; + len += rclen; - if (status < 0) - return (status); + /* append the result to the wbuf, don't write to the user */ + if (sock->batch_start) + return add_to_wbuf(sock, buffer, len); + + /* first write must be complete */ + if (len != write(sock->fd, buffer, len)) + { + RRDD_LOG(LOG_INFO, "send_response: could not write status message"); + return -1; + } - nleft -= status; - ptr += status; + if (sock->wbuf != NULL && rc == RESP_OK) + { + wrote = 0; + while (wrote < sock->wbuf_len) + { + ssize_t wb = write(sock->fd, sock->wbuf + wrote, sock->wbuf_len - wrote); + if (wb <= 0) + { + RRDD_LOG(LOG_INFO, "send_response: could not write results"); + return -1; + } + wrote += wb; + } } - return (0); -} /* }}} ssize_t swrite */ + free(sock->wbuf); sock->wbuf = NULL; + sock->wbuf_len = 0; + + return 0; +} /* }}} */ static void wipe_ci_values(cache_item_t *ci, time_t when) { @@ -435,6 +541,34 @@ static void remove_from_queue(cache_item_t *ci) /* {{{ */ ci->flags &= ~CI_FLAGS_IN_QUEUE; } /* }}} static void remove_from_queue */ +/* remove an entry from the tree and free all its resources. + * must hold 'cache lock' while calling this. + * returns 0 on success, otherwise errno */ +static int forget_file(const char *file) +{ + cache_item_t *ci; + + ci = g_tree_lookup(cache_tree, file); + if (ci == NULL) + return ENOENT; + + g_tree_remove (cache_tree, file); + remove_from_queue(ci); + + for (int i=0; i < ci->values_num; i++) + free(ci->values[i]); + + free (ci->values); + free (ci->file); + + /* in case anyone is waiting */ + pthread_cond_broadcast(&ci->flushed); + + free (ci); + + return 0; +} /* }}} static int forget_file */ + /* * enqueue_cache_item: * `cache_lock' must be acquired before calling this function! @@ -569,26 +703,10 @@ static int flush_old_values (int max_age) for (k = 0; k < cfd.keys_num; k++) { - cache_item_t *ci; - - /* This must not fail. */ - ci = (cache_item_t *) g_tree_lookup (cache_tree, cfd.keys[k]); - assert (ci != NULL); - - /* If we end up here with values available, something's seriously - * messed up. */ - assert (ci->values_num == 0); - - /* Remove the node from the tree */ - g_tree_remove (cache_tree, cfd.keys[k]); - cfd.keys[k] = NULL; - - /* Now free and clean up `ci'. */ - free (ci->file); - ci->file = NULL; - free (ci); - ci = NULL; - } /* for (k = 0; k < cfd.keys_num; k++) */ + /* should never fail, since we have held the cache_lock + * the entire time */ + assert( forget_file(cfd.keys[k]) == 0 ); + } if (cfd.keys != NULL) { @@ -810,13 +928,12 @@ static int buffer_get_field (char **buffer_ret, /* {{{ */ * check whether the file falls within the dir * returns 1 if OK, otherwise 0 */ -static int check_file_access (const char *file, int fd) /* {{{ */ +static int check_file_access (const char *file, listen_socket_t *sock) /* {{{ */ { - char error[CMD_MAX]; assert(file != NULL); if (!config_write_base_only - || fd < 0 /* journal replay */ + || sock == NULL /* journal replay */ || config_base_dir == NULL) return 1; @@ -833,11 +950,26 @@ static int check_file_access (const char *file, int fd) /* {{{ */ return 1; err: - snprintf(error, sizeof(error)-1, "-1 %s\n", rrd_strerror(EACCES)); - swrite(fd, error, strlen(error)); + if (sock != NULL && sock->fd >= 0) + send_response(sock, RESP_ERR, "%s\n", rrd_strerror(EACCES)); + return 0; } /* }}} static int check_file_access */ +/* returns 1 if we have the required privilege level, + * otherwise issue an error to the user on sock */ +static int has_privilege (listen_socket_t *sock, /* {{{ */ + socket_privilege priv) +{ + if (sock == NULL) /* journal replay */ + return 1; + + if (sock->privilege >= priv) + return 1; + + return send_response(sock, RESP_ERR, "%s\n", rrd_strerror(EACCES)); +} /* }}} static int has_privilege */ + static int flush_file (const char *filename) /* {{{ */ { cache_item_t *ci; @@ -858,131 +990,150 @@ static int flush_file (const char *filename) /* {{{ */ pthread_cond_wait(&ci->flushed, &cache_lock); } + /* DO NOT DO ANYTHING WITH ci HERE!! The entry + * may have been purged during our cond_wait() */ + pthread_mutex_unlock(&cache_lock); return (0); } /* }}} int flush_file */ -static int handle_request_help (int fd, /* {{{ */ +static int handle_request_help (listen_socket_t *sock, /* {{{ */ char *buffer, size_t buffer_size) { int status; char **help_text; - size_t help_text_len; char *command; - size_t i; - char *help_help[] = - { - "5 Command overview\n", - "FLUSH \n", - "FLUSHALL\n", - "HELP []\n", - "UPDATE [ ...]\n", + char *help_help[2] = + { + "Command overview\n" + , + "HELP []\n" + "FLUSH \n" + "FLUSHALL\n" + "PENDING \n" + "FORGET \n" + "UPDATE [ ...]\n" + "BATCH\n" "STATS\n" }; - size_t help_help_len = sizeof (help_help) / sizeof (help_help[0]); - char *help_flush[] = + char *help_flush[2] = { - "4 Help for FLUSH\n", - "Usage: FLUSH \n", - "\n", - "Adds the given filename to the head of the update queue and returns\n", + "Help for FLUSH\n" + , + "Usage: FLUSH \n" + "\n" + "Adds the given filename to the head of the update queue and returns\n" "after is has been dequeued.\n" }; - size_t help_flush_len = sizeof (help_flush) / sizeof (help_flush[0]); - char *help_flushall[] = + char *help_flushall[2] = { - "3 Help for FLUSHALL\n", - "Usage: FLUSHALL\n", - "\n", + "Help for FLUSHALL\n" + , + "Usage: FLUSHALL\n" + "\n" "Triggers writing of all pending updates. Returns immediately.\n" }; - size_t help_flushall_len = sizeof(help_flushall) / sizeof(help_flushall[0]); - char *help_update[] = + char *help_pending[2] = + { + "Help for PENDING\n" + , + "Usage: PENDING \n" + "\n" + "Shows any 'pending' updates for a file, in order.\n" + "The updates shown have not yet been written to the underlying RRD file.\n" + }; + + char *help_forget[2] = { - "9 Help for UPDATE\n", + "Help for FORGET\n" + , + "Usage: FORGET \n" + "\n" + "Removes the file completely from the cache.\n" + "Any pending updates for the file will be lost.\n" + }; + + char *help_update[2] = + { + "Help for UPDATE\n" + , "Usage: UPDATE [ ...]\n" - "\n", - "Adds the given file to the internal cache if it is not yet known and\n", - "appends the given value(s) to the entry. See the rrdcached(1) manpage\n", - "for details.\n", - "\n", - "Each has the following form:\n", - " =