/* virt-hostinfo
 * Copyright (C) 2009 Red Hat Inc.
 *
 * This program 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 2 of the License, or
 * (at your option) any later version.
 *
 * This program 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 this program; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */

#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);
extern void error (const char *fs, ...)
  __attribute__((format (printf,1,2)));
extern void perrorf (const char *fs, ...)
  __attribute__((format (printf,1,2)));
extern void warning (const char *fs, ...)
  __attribute__((format (printf,1,2)));
extern void pwarningf (const char *fs, ...)
  __attribute__((format (printf,1,2)));
extern void debug (const char *fs, ...)
  __attribute__((format (printf,1,2)));
extern void message (const char *fs, ...)
  __attribute__((format (printf,1,2)));
extern void paprerror (apr_status_t r, const char *fs, ...)
  __attribute__((format (printf,2,3)));

/* 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 */