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
|
/*
* $Header$
*
* Copyright 2008 Massachusetts Institute of Technology.
* All Rights Reserved.
*
* Export of this software from the United States of America may
* require a specific license from the United States Government.
* It is the responsibility of any person or organization contemplating
* export to obtain such a license before exporting.
*
* WITHIN THAT CONSTRAINT, permission to use, copy, modify, and
* distribute this software and its documentation for any purpose and
* without fee is hereby granted, provided that the above copyright
* notice appear in all copies and that both that copyright notice and
* this permission notice appear in supporting documentation, and that
* the name of M.I.T. not be used in advertising or publicity pertaining
* to distribution of the software without specific, written prior
* permission. Furthermore if you modify this software you must label
* your software as modified software and not distribute it in such a
* fashion that it might be confused with the original M.I.T. software.
* M.I.T. makes no representations about the suitability of
* this software for any purpose. It is provided "as is" without express
* or implied warranty.
*/
#include <string.h>
#include "assert.h"
#pragma warning (disable : 4996)
#include "win-utils.h"
#include "WorkItem.h"
extern "C" {
#include "cci_debugging.h"
}
// CountedBuffer makes a copy of the data. Each CountedBuffer must be deleted.
void deleteBuffer(char** buf) {
if (*buf) {
delete [](*buf);
*buf = NULL;
}
}
// WorkItem contains a CountedBuffer which must be deleted,
// so each WorkItem must be deleted.
WorkItem::WorkItem(k5_ipc_stream buf, WIN_PIPE* pipe, const long type, const long sst)
: _buf(buf), _rpcmsg(type), _pipe(pipe), _sst(sst) { }
WorkItem::WorkItem(const WorkItem& item) : _buf(NULL), _rpcmsg(0), _pipe(NULL), _sst(0) {
k5_ipc_stream _buf = NULL;
krb5int_ipc_stream_new(&_buf);
krb5int_ipc_stream_write(_buf,
krb5int_ipc_stream_data(item.payload()),
krb5int_ipc_stream_size(item.payload()) );
WorkItem(_buf, item._pipe, item._rpcmsg, item._sst);
}
WorkItem::WorkItem() : _buf(NULL), _rpcmsg(CCMSG_INVALID), _pipe(NULL), _sst(0) { }
WorkItem::~WorkItem() {
if (_buf) krb5int_ipc_stream_release(_buf);
if (_pipe) ccs_win_pipe_release(_pipe);
}
const k5_ipc_stream WorkItem::take_payload() {
k5_ipc_stream temp = payload();
_buf = NULL;
return temp;
}
WIN_PIPE* WorkItem::take_pipe() {
WIN_PIPE* temp = pipe();
_pipe = NULL;
return temp;
}
WorkList::WorkList() {
assert(InitializeCriticalSectionAndSpinCount(&cs, 0x80000400));
}
WorkList::~WorkList() {
// Delete any WorkItems in the queue:
WorkItem* item;
cci_debug_printf("%s", __FUNCTION__);
char buf[2048];
char* pbuf = (char*)buf;
while (remove(&item)) {
cci_debug_printf("WorkList::~WorkList() deleting %s", item->print(pbuf));
delete item;
}
DeleteCriticalSection(&cs);
}
char* WorkItem::print(char* buf) {
sprintf(buf, "WorkItem msg#:%d sst:%ld pipe:<%s>/0x%X", _rpcmsg, _sst,
ccs_win_pipe_getUuid(_pipe), ccs_win_pipe_getHandle(_pipe));
return buf;
}
int WorkList::add(WorkItem* item) {
EnterCriticalSection(&cs);
wl.push_front(item);
LeaveCriticalSection(&cs);
return 1;
}
int WorkList::remove(WorkItem** item) {
bool bEmpty;
bEmpty = wl.empty() & 1;
if (!bEmpty) {
EnterCriticalSection(&cs);
*item = wl.back();
wl.pop_back();
LeaveCriticalSection(&cs);
}
return !bEmpty;
}
|
