CPCD.hpp

/*
   Copyright (C) 2003-2007 MySQL AB, 2009, 2010 Sun Microsystems, Inc.
    All rights reserved. Use is subject to license terms.

   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; version 2 of the License.

   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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301  USA
*/

#ifndef CPCD_HPP
#define CPCD_HPP

#include <Vector.hpp>
#include <Properties.hpp>
#include <NdbOut.hpp>
#include <NdbThread.h>
#include <NdbCondition.h>
#include <BaseString.hpp>

#ifdef _WIN32
typedef DWORD pid_t;
#endif
const pid_t bad_pid = -1;

inline bool is_bad_pid(pid_t pid)
{
#ifdef _WIN32
  return pid == bad_pid;
#else
  return pid <= 1;
#endif
}

#define CPCD_DEFAULT_PROC_FILE          "ndb_cpcd.conf"
#define CPCD_DEFAULT_TCP_PORT           1234
#define CPCD_DEFAULT_POLLING_INTERVAL   5 /* seconds */
#ifndef _WIN32
#define CPCD_DEFAULT_WORK_DIR           "/var/run/ndb_cpcd"
#define CPCD_DEFAULT_CONFIG_FILE        "/etc/ndb_cpcd.conf"

#else
#define CPCD_DEFAULT_WORK_DIR           "c:\\ndb_cpcd"
#define CPCD_DEFAULT_CONFIG_FILE        "c:\\ndb_cpcd\\ndb_cpcd.conf"
#endif
enum ProcessStatus {
  STOPPED  = 0,
  STARTING = 1,
  RUNNING  = 2,
  STOPPING = 3
};

enum ProcessType {
  PERMANENT = 0,
  TEMPORARY = 1
};

enum RequestStatusCode {
  OK = 0,            
  Error = 1,         
  AlreadyExists = 2, 
  NotExists = 3,     
  AlreadyStopped = 4
};

class CPCD {
public:
  class RequestStatus {
  public:
    RequestStatus() { m_status = OK; m_errorstring[0] = '\0'; };

    void err(enum RequestStatusCode, const char *);

    char *getErrMsg() { return m_errorstring; };

    enum RequestStatusCode getStatus() { return m_status; };
  private:
    enum RequestStatusCode m_status;
    char m_errorstring[256];
  };
  class Process {
    pid_t m_pid;
#ifdef _WIN32
    HANDLE m_job;
#endif
  public:
    Process(const Properties & props, class CPCD *cpcd);
    void monitor();

    bool isRunning();

    int start();

    void stop();      

    int readPid();

    int writePid(int pid);

    void print(FILE *);

    int m_id;

    BaseString m_name;  

    BaseString m_group;

    BaseString m_env;

    BaseString m_path;

    BaseString m_args;

    BaseString m_type;
    ProcessType m_processType;
    
    BaseString m_cwd;

    BaseString m_owner;

    BaseString m_runas;

    BaseString m_stdin;

    BaseString m_stdout;

    BaseString m_stderr;

    enum ProcessStatus m_status;

    BaseString m_ulimit;

    BaseString m_shutdown_options;

  private:
    class CPCD *m_cpcd;
    void do_exec();
  };

  class Monitor {
  public:
    Monitor(CPCD *cpcd, int poll = CPCD_DEFAULT_POLLING_INTERVAL);

    ~Monitor();

    void run();

    void signal();
  private:
    class CPCD *m_cpcd;
    struct NdbThread *m_monitorThread;
    bool m_monitorThreadQuitFlag;
    struct NdbCondition *m_changeCondition;
    NdbMutex *m_changeMutex;
    int m_pollingInterval; /* seconds */
  };

  CPCD();

  ~CPCD();

  bool defineProcess(RequestStatus *rs, Process * arg);

  bool undefineProcess(RequestStatus *rs, int id);

  bool startProcess(RequestStatus *rs, int id);

  bool stopProcess(RequestStatus *rs, int id);
  
  bool listProcesses(RequestStatus *rs, MutexVector<const char *> &);

  bool loadingProcessList;

  bool saveProcessList();

  bool loadProcessList();

  MutexVector<Process *> *getProcessList();

  MutexVector<Process *> m_processes;

private:
  friend class Process;  
  bool notifyChanges();
  int findUniqueId();
  BaseString m_procfile;
  Monitor *m_monitor;
};

#endif