-
Notifications
You must be signed in to change notification settings - Fork 44
/
lock_server.cc
80 lines (71 loc) · 1.82 KB
/
lock_server.cc
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
// the lock server implementation
#include "lock_server.h"
#include <sstream>
#include <stdio.h>
#include <unistd.h>
#include <arpa/inet.h>
lock::lock(lock_protocol::lockid_t lid):
status(lock::FREE)
{
lid = lid;
pthread_cond_init(&lcond, NULL);
}
lock::lock(lock_protocol::lockid_t lid, int stat):
status(stat)
{
lid = lid;
pthread_cond_init(&lcond, NULL);
}
lock_server::lock_server():
nacquire (0)
{
pthread_mutex_init(&mutex, NULL);
}
lock_protocol::status
lock_server::stat(int clt, lock_protocol::lockid_t lid, int &r)
{
lock_protocol::status ret = lock_protocol::OK;
printf("stat request from clt %d\n", clt);
r = nacquire;
return ret;
}
lock_protocol::status
lock_server::acquire(int clt, lock_protocol::lockid_t lid, int &r)
{
lock_protocol::status ret = lock_protocol::OK;
std::map<lock_protocol::lockid_t, lock* >::iterator iter;
pthread_mutex_lock(&mutex);
iter = lockmap.find(lid);
if(iter != lockmap.end()) {
while(iter->second->status != lock::FREE) {
pthread_cond_wait(&(iter->second->lcond), &mutex);
}
iter->second->status = lock::LOCKED;
pthread_mutex_unlock(&mutex);
return ret;
} else {
lock *new_lock = new lock(lid, lock::LOCKED);
// lockmap[lid] = new_lock;
lockmap.insert(std::make_pair(lid, new_lock));
pthread_mutex_unlock(&mutex);
return ret;
}
}
lock_protocol::status
lock_server::release(int clt, lock_protocol::lockid_t lid, int &r)
{
lock_protocol::status ret = lock_protocol::OK;
std::map<lock_protocol::lockid_t, lock*>::iterator iter;
pthread_mutex_lock(&mutex);
iter = lockmap.find(lid);
if (iter != lockmap.end()) {
iter->second->status = lock::FREE;
pthread_cond_signal(&(iter->second->lcond));
pthread_mutex_unlock(&mutex);
return ret;
} else {
ret = lock_protocol::IOERR;
pthread_mutex_unlock(&mutex);
return ret;
}
}