File: coordinatorapi.h

package info (click to toggle)
dmtcp 2.6.0-1
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 6,496 kB
  • sloc: cpp: 33,592; ansic: 28,099; sh: 6,735; makefile: 1,950; perl: 1,690; python: 1,241; asm: 138; java: 13
file content (127 lines) | stat: -rw-r--r-- 5,658 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
/****************************************************************************
 *   Copyright (C) 2006-2010 by Jason Ansel, Kapil Arya, and Gene Cooperman *
 *   jansel@csail.mit.edu, kapil@ccs.neu.edu, gene@ccs.neu.edu              *
 *                                                                          *
 *  This file is part of DMTCP.                                             *
 *                                                                          *
 *  DMTCP is free software: you can redistribute it and/or                  *
 *  modify it under the terms of the GNU Lesser General Public License as   *
 *  published by the Free Software Foundation, either version 3 of the      *
 *  License, or (at your option) any later version.                         *
 *                                                                          *
 *  DMTCP 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 Lesser General Public License for more details.                     *
 *                                                                          *
 *  You should have received a copy of the GNU Lesser General Public        *
 *  License along with DMTCP:dmtcp/src.  If not, see                        *
 *  <http://www.gnu.org/licenses/>.                                         *
 ****************************************************************************/

#ifndef COORDINATORAPI_H
#define COORDINATORAPI_H

#include "constants.h"
#include "protectedfds.h"
#include "dmtcpmessagetypes.h"
#include "shareddata.h"
#include "../jalib/jsocket.h"
#include "../jalib/jalloc.h"

namespace dmtcp
{
  enum CoordinatorMode {
    COORD_INVALID   = 0x0000,
    COORD_JOIN      = 0x0001,
    COORD_NEW       = 0x0002,
    COORD_NONE      = 0x0004,
    COORD_ANY       = 0x0010
  };

  class CoordinatorAPI
  {
    public:
#ifdef JALIB_ALLOCATOR
      static void* operator new(size_t nbytes, void* p) { return p; }
      static void* operator new(size_t nbytes) { JALLOC_HELPER_NEW(nbytes); }
      static void  operator delete(void* p) { JALLOC_HELPER_DELETE(p); }
#endif
      CoordinatorAPI (void) : _coordinatorSocket(-1), _nsSock(-1) {}
      // Use default destructor

      static CoordinatorAPI& instance();
      static void init();
      static void restart();
      static void resetOnFork(CoordinatorAPI& coordAPI);
      void resetCoordSocketFd();

      void setupVirtualCoordinator(CoordinatorInfo *coordInfo,
                                   struct in_addr  *localIP);
      void waitForCheckpointCommand();
      static bool noCoordinator();

      void connectToCoordOnStartup(CoordinatorMode  mode,
                                   string           progname,
                                   DmtcpUniqueProcessId *compId,
                                   CoordinatorInfo *coordInfo,
                                   struct in_addr  *localIP);
      void createNewConnectionBeforeFork(string& progname);
      void connectToCoordOnRestart(CoordinatorMode  mode,
                                   string progname,
                                   UniquePid compGroup,
                                   int np,
                                   CoordinatorInfo *coordInfo,
                                   const char *host,
                                   int port,
                                   struct in_addr  *localIP);
      void closeConnection() { _coordinatorSocket.close(); }

      //jalib::JSocket& coordinatorSocket() { return _coordinatorSocket; }
      bool isValid() { return _coordinatorSocket.isValid(); }

      void sendMsgToCoordinator(const DmtcpMessage &msg,
                                const void *extraData = NULL,
                                size_t len = 0);
      void recvMsgFromCoordinator(DmtcpMessage *msg,
                                  void **extraData = NULL);
      char* connectAndSendUserCommand(char c,
                                      int *coordCmdStatus = NULL,
                                      int *numPeers = NULL,
                                      int *isRunning = NULL,
                                      int *ckptInterval = NULL,
                                      uint32_t logMask = 0);

      void updateCoordCkptDir(const char *dir);
      string getCoordCkptDir(void);

      bool updateGlobalCkptDir(const char *dir);

      void sendCkptFilename();

      int sendKeyValPairToCoordinator(const char *id,
                                      const void *key, uint32_t key_len,
                                      const void *val, uint32_t val_len,
				      int sync = 0);
      int sendQueryToCoordinator(const char *id,
                                 const void *key, uint32_t key_len,
                                 void *val, uint32_t *val_len);
      int getUniqueIdFromCoordinator(const char *id,
                                     const void *key,
                                     uint32_t key_len,
                                     void *val,
                                     uint32_t *val_len,
                                     uint32_t offset = 1);

    private:
      void startNewCoordinator(CoordinatorMode mode);
      void createNewConnToCoord(CoordinatorMode mode);
      DmtcpMessage sendRecvHandshake(DmtcpMessage msg, string progname,
                                     UniquePid *compId = NULL);

      jalib::JSocket          _coordinatorSocket;
      jalib::JSocket          _nsSock;
  };
}

#endif