X-Git-Url: http://git.annexia.org/?a=blobdiff_plain;f=hostinfod%2Fhostinfod.h;h=8bb48e69c98c0b5bc12f876b7dfb615f258d2ee6;hb=HEAD;hp=f1da3e459c10d82ec79d54a6c4e821d2473b78ca;hpb=11bf652b21fb6e20dc072c7e250aeb26821b818e;p=virt-hostinfo.git

diff --git a/hostinfod/hostinfod.h b/hostinfod/hostinfod.h
index f1da3e4..8bb48e6 100644
--- a/hostinfod/hostinfod.h
+++ b/hostinfod/hostinfod.h
@@ -19,19 +19,81 @@
 #ifndef HOSTINFOD_H
 #define HOSTINFOD_H
 
+#include <time.h>
+
 #include <apr_general.h>
 #include <apr_errno.h>
+#include <apr_pools.h>
+#include <apr_network_io.h>
+#include <apr_poll.h>
+#include <apr_hash.h>
+
+#include <libvirt/libvirt.h>
+
+enum guest_state {
+  guest_state_connecting,	/* Connecting to socket. */
+  guest_state_request,		/* Waiting or reading the request. */
+  guest_state_reply,		/* Sending the reply. */
+  guest_state_dead		/* Connection is dead. */
+};
+
+struct guest_description {
+  apr_pool_t *pool;		/* Pool for lifetime of guest connection. */
+
+  apr_pool_t *rpool;		/* Pool for lifetime of single req/resp. */
+
+  const char *name;		/* "driver-name" */
+  const char *sock_path;	/* Full path to socket. */
+  int sock;			/* Real socket. */
+  apr_socket_t *aprsock;	/* APR socket. */
+  apr_pollfd_t pollfd;		/* APR poll descriptor. */
+  enum guest_state state;	/* State of the connection. */
+  int counter;			/* Used in reread_socket_dir. */
+
+  unsigned request_max;		/* Max. length of request buffer. */
+  unsigned request_posn;	/* Position in request buffer. */
+  char *request;		/* Request buffer. */
+
+  unsigned reply_size;		/* Size of reply buffer. */
+  unsigned reply_posn;		/* Position in reply buffer. */
+  char *reply;			/* Reply buffer. */
+
+  /* Increments every time guest does something bad, decremented once per min */
+  unsigned penalty;
+  struct timespec last_penalty_decr;
+
+  /* Hash records last time each command was run by this guest. */
+  apr_hash_t *lasttime;
+};
+
+enum arg_type {
+  arg_type_string,
+  arg_type_int,
+  arg_type_bool,
+};
+
+struct arg {
+  enum arg_type type;
+  union {
+    long i;			/* For int, bool types. */
+    char *str;			/* For string type. */
+  } u;
+};
 
 /* main.c */
 extern const char *conf_file;
 extern char *socket_dir;
 extern char *guests_file;
+extern char *libvirt_uri;
+extern int libvirt_uri_set_on_cmdline;
 extern int verbose;
 extern int verbose_set_on_cmdline;
 extern int foreground;
 extern int foreground_set_on_cmdline;
 extern int messages_to_stderr;
 extern apr_pool_t *pool;	/* pool for global memory allocation */
+extern void initialize (void);
+extern struct timespec *diff_timespec (struct timespec *r, const struct timespec *a, const struct timespec *b); /* r = a - b */
 
 /* error.c */
 extern void init_syslog (void);
@@ -52,9 +114,44 @@ extern void paprerror (apr_status_t r, const char *fs, ...)
 
 /* configuration.c */
 extern void read_main_conf_file (void);
+extern void check_guests_file (struct guest_description *hval, const char *cmd, double *interval, int *enabled);
 
 /* monitor_sockets.c */
 extern int sockets_inotify_fd;
 extern void monitor_socket_dir (void);
 
+/* commands.c */
+extern apr_hash_t *commands;
+extern void execute_command (const struct timespec *now, struct guest_description *hval, const char *command);
+extern int get_args (apr_array_header_t *args, const char *argfs, ...);
+extern void send_error (struct guest_description *hval, int code);
+extern void send_reply (struct guest_description *hval, int code, const char *fs, ...)
+  __attribute__((format (printf,3,4)));
+
+typedef void (*command_fn) (struct guest_description *hval, const char *cmd, apr_array_header_t *args);
+
+#define REGISTER_COMMAND(name)						\
+  static void register_##name (void) __attribute__((constructor));	\
+									\
+  static void								\
+  register_##name (void)						\
+  {									\
+    if (pool == NULL)							\
+      initialize ();							\
+									\
+    if (commands == NULL)						\
+      commands = apr_hash_make (pool);					\
+									\
+    /* The assignment below just causes a warning			\
+     * if the callback has the wrong type.				\
+     */									\
+    command_fn cb = name;						\
+    apr_hash_set (commands, #name, APR_HASH_KEY_STRING, cb);		\
+  }
+
+/* virt.c */
+extern void init_libvirt (void);
+extern virConnectPtr conn;
+extern virNodeInfo nodeinfo;
+
 #endif /* HOSTINFOD_H */