sld.hh

#ifndef SLD_HH
# define SLD_HH

# include <cassert>
# include <ios>
# include <iostream>
# include <fstream>
# include <sstream>
# include <map>

# include "error.hh"

typedef std::string string;
typedef std::stringstream sstream;
typedef std::ostream ostream;
typedef std::ifstream ifstream;

enum {
  ERR_OK = 0,
  ERR_USAGE = 1,
  ERR_BADPARAM = 2,
  ERR_FILE = 3,
  ERR_NET = 4,
  ERR_PROTO = 5,
  ERR_AUTH = 6,
  ERR_NOMEM = 41,
  ERR_UNKNOWN = 42
  
};

// -----------------------------------------------------------------------------
// Gonstant
// -----------------------------------------------------------------------------

const string VERSION = "1";
const int MAX_LINE_SIZE = 512;

// -----------------------------------------------------------------------------
// SLDAEMON CLASS
// -----------------------------------------------------------------------------

class SLDaemon
{
public:

  static const string version;
  
  SLDaemon();

  struct options {
    options();
    string server;
    int port;
    string login;
    string pass;
    int verbose; //0=false, 1=true, 3=undef
    string scriptdir;
    string conffile;
  };

  inline bool verbose() const { return options_.verbose == 1; }

  options *getoptions(int argc, char *argv[]) const;
  options *getoptions(const string file) const;
  void applyoptions(const options *opt);
  void run();
protected:
  // parsed options
  options options_;

  // current socket
  int socket_fd_;
  FILE* socket_fs_;

  // pid list
  std::set<int> jobs;

  // Options functions
  void usage(const char *argv0) const;
  void check_options() const;

  // network functions
  void connect();
  void disconnect();  
  void send(const char *data, size_t len, bool buf = true);
  void send(const string str, bool buf = true);
  char *recv(size_t size);
  void recv(size_t size, const string filename);
  char *recvln();
  void flush();
  
  // protocol functions
  void auth();
  void tee(const string msg);
  void proto_ok();
  void proto_ko();
  void proto_violation();
  
  // commmand functions
  void cmd_exit();
  void cmd_version();
  void cmd_clean();
  void cmd_list();
  void cmd_reload();
  void cmd_update(const char *line);
  void cmd_exec(const char *line);
  void cmd_file(const char *line);

  // others functions
  string md5(const string file) const;
};

#endif