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
|
#ifdef USE_PCH
#include "pch.h"
#else
#include <sys/types.h>
#include <unistd.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <cassert>
#include <string>
#include <konforka/exception.h>
using namespace std;
#include "sitecing/scoreboard.h"
#endif
namespace sitecing {
scoreboard::scoreboard()
: shmid(-1), slots(NULL) {
shmid = shmget(IPC_PRIVATE,MAX_SITECING_SCOREBOARD_SLOTS*sizeof(scoreboard_slot),IPC_CREAT|0600);
if(shmid<0)
throw konforka::exception(CODEPOINT,"failed to shmget()");
slots = (scoreboard_slot*)shmat(shmid,NULL,0);
if(shmctl(shmid,IPC_RMID,NULL))
throw konforka::exception(CODEPOINT,"failed to shmctl()");
if(!slots)
throw konforka::exception(CODEPOINT,"failed to shmat()");
for(int tmp=0;tmp<MAX_SITECING_SCOREBOARD_SLOTS;tmp++)
slots[tmp].state=scoreboard_slot::state_free;
}
scoreboard::~scoreboard() {
shmdt(slots);
}
int scoreboard::allocate_slot() {
for(int tmp=0;tmp<MAX_SITECING_SCOREBOARD_SLOTS;tmp++) {
if(slots[tmp].state==scoreboard_slot::state_free) {
slots[tmp].state=scoreboard_slot::state_allocated;
slots[tmp].pid=0;
return tmp;
}
}
throw konforka::exception(CODEPOINT,"out of scoreboard slots");
}
void scoreboard::free_slot(int slot) {
assert(slot>=0 && slot<MAX_SITECING_SCOREBOARD_SLOTS);
if(slots[slot].state==scoreboard_slot::state_free)
throw konforka::exception(CODEPOINT,"freeing unallocated slot");
slots[slot].state=scoreboard_slot::state_free;
}
scoreboard_slot *scoreboard::get_slot(int slot) {
assert(slot>=0 && slot<MAX_SITECING_SCOREBOARD_SLOTS);
return &slots[slot];
}
int scoreboard::get_slot_by_pid(pid_t pid) {
for(int rv=0;rv<MAX_SITECING_SCOREBOARD_SLOTS;rv++)
if( (slots[rv].state!=scoreboard_slot::state_free) && (slots[rv].pid == pid) )
return rv;
throw konforka::exception(CODEPOINT,"no such process");
}
int scoreboard::count_slots(enum scoreboard_slot::_state state) {
int rv = 0;
for(int tmp=0;tmp<MAX_SITECING_SCOREBOARD_SLOTS;tmp++) {
if(slots[tmp].state==state)
rv++;
}
return rv;
}
}
|
