File: dmtcpworker.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 (76 lines) | stat: -rw-r--r-- 3,142 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
/****************************************************************************
 *   Copyright (C) 2006-2008 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 DMTCPDMTCPWORKER_H
#define DMTCPDMTCPWORKER_H

#include "jalloc.h"
#include "dmtcpalloc.h"
#include "dmtcpmessagetypes.h"

void restoreUserLDPRELOAD();

namespace dmtcp
{
  class DmtcpWorker
  {
    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
      DmtcpWorker();
      ~DmtcpWorker();
      static DmtcpWorker& instance();

      static void waitForCoordinatorMsg(string signalStr,
                                        DmtcpMessageType type);
      static void informCoordinatorOfRUNNINGState();
      static void waitForStage1Suspend();
      static void waitForStage2Checkpoint();
      static void waitForStage3Refill(bool isRestart);
      static void waitForStage4Resume(bool isRestart);
      static void restoreVirtualPidTable();
      static void postRestart();

      static void resetOnFork();
      static void cleanupWorker();

      static int determineCkptSignal();

      static void setExitInProgress() { _exitInProgress = true; };
      static bool exitInProgress() { return _exitInProgress; };
      static void interruptCkpthread();

      static void writeCheckpointPrefix(int fd);

      static void eventHook(DmtcpEvent_t id, DmtcpEventData_t *data);

    protected:
      static void sendUserCommand(char c, int* result = NULL);
    private:
      static DmtcpWorker theInstance;
      static bool _exitInProgress;
  };
}

#endif