1: /*************************************************************************
2: * (C) 2008 AITNET ltd - Sofia/Bulgaria - <misho@aitbg.com>
3: * by Michael Pounov <misho@openbsd-bg.org>
4: *
5: * $Author: misho $
6: * $Id: sess.c,v 1.5.4.1 2013/01/17 13:10:49 misho Exp $
7: *
8: **************************************************************************
9: The ELWIX and AITNET software is distributed under the following
10: terms:
11:
12: All of the documentation and software included in the ELWIX and AITNET
13: Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org>
14:
15: Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2013
16: by Michael Pounov <misho@elwix.org>. All rights reserved.
17:
18: Redistribution and use in source and binary forms, with or without
19: modification, are permitted provided that the following conditions
20: are met:
21: 1. Redistributions of source code must retain the above copyright
22: notice, this list of conditions and the following disclaimer.
23: 2. Redistributions in binary form must reproduce the above copyright
24: notice, this list of conditions and the following disclaimer in the
25: documentation and/or other materials provided with the distribution.
26: 3. All advertising materials mentioning features or use of this software
27: must display the following acknowledgement:
28: This product includes software developed by Michael Pounov <misho@elwix.org>
29: ELWIX - Embedded LightWeight unIX and its contributors.
30: 4. Neither the name of AITNET nor the names of its contributors
31: may be used to endorse or promote products derived from this software
32: without specific prior written permission.
33:
34: THIS SOFTWARE IS PROVIDED BY AITNET AND CONTRIBUTORS ``AS IS'' AND
35: ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
36: IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
37: ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
38: FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
39: DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
40: OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
41: HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
42: LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
43: OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
44: SUCH DAMAGE.
45: */
46: #include "global.h"
47:
48:
49: /*
50: * sess_FreeValues() Free all values from value array allocated from sess_GetValues()
51: *
52: * @ppsVals = Array strings
53: * return: none
54: */
55: inline void
56: sess_FreeValues(char *** __restrict ppsVals)
57: {
58: char **ptr;
59:
60: assert(ppsVals);
61: if (!ppsVals)
62: return;
63:
64: for (ptr = *ppsVals; *ptr; ptr++)
65: e_free(*ptr);
66: e_free(*ppsVals);
67: *ppsVals = NULL;
68: }
69:
70: /*
71: * sess_GetValues() Get all values from session shared memory
72: *
73: * @s = Session item
74: * @ppsVals = Return array strings
75: * return: -1 error: in parameter, !=-1 count of returned strings in ppsVals (must be free after use!)
76: */
77: int
78: sess_GetValues(ait_sess_t * __restrict s, char ***ppsVals)
79: {
80: register int i;
81: char **valz, *Shared = NULL;
82: char *peer, *p_brk;
83:
84: if (!s || !ppsVals)
85: return -1;
86: valz = e_malloc(sizeof(caddr_t));
87: if (!valz) {
88: LOGERR;
89: return -1;
90: } else
91: *valz = NULL;
92:
93: // allocated memory & mirrored shared memory into this
94: Shared = e_malloc(s->eom);
95: if (!Shared) {
96: LOGERR;
97: e_free(valz);
98: return -1;
99: } else
100: memcpy(Shared, s->addr, s->eom);
101:
102: for (i = 0, peer = strtok_r(Shared, MEM_DELIM"\r\n", &p_brk); peer;
103: peer = strtok_r(NULL, MEM_DELIM"\r\n", &p_brk)) {
104: if (!strchr(peer, '='))
105: continue;
106: else
107: i++;
108:
109: valz = e_realloc(valz, (i + 1) * sizeof(caddr_t));
110: if (!valz) {
111: LOGERR;
112: e_free(Shared);
113: return -1;
114: } else
115: valz[i] = NULL;
116:
117: valz[i - 1] = e_strdup(peer);
118: }
119:
120: e_free(Shared);
121: *ppsVals = valz;
122: return i;
123: }
124:
125: /*
126: * sess_GetValue() Get value from session shared memory from attribute
127: *
128: * @s = Session item
129: * @csAttr = Attribute for search
130: * @psVal = Return string buffer
131: * @pnLen = Length of return string buffer,
132: // *{pnLen} input is max_size of buffer & output is really taken bytes
133: * return: 0 not found, -1 error: in parameter, >0 get position, if define item merged with IS_DEF
134: */
135: int
136: sess_GetValue(ait_sess_t * __restrict s, const char *csAttr, char *psVal, int *pnLen)
137: {
138: register int i;
139: int def = IS_VAL;
140: char *Shared = NULL;
141: char *peer, *p_brk, *a_brk, *attr, *val;
142:
143: if (!s || !csAttr || !*csAttr)
144: return -1;
145: if (psVal) {
146: if (pnLen && *pnLen > 0)
147: memset(psVal, 0, *pnLen);
148: else
149: return -1;
150: }
151:
152: // allocated memory & mirrored shared memory into this
153: Shared = e_malloc(s->eom);
154: if (!Shared) {
155: LOGERR;
156: return -1;
157: } else
158: memcpy(Shared, s->addr, s->eom);
159:
160: for (i = 1, peer = strtok_r(Shared, MEM_DELIM"\r\n", &p_brk); peer;
161: i++, peer = strtok_r(NULL, MEM_DELIM"\r\n", &p_brk)) {
162: attr = strtok_r(peer, "=\r\n", &a_brk);
163: if (attr && !strncmp(attr, csAttr, MAX_ATTRIBUTE - 1)) {
164: val = strtok_r(NULL, "=\r\n", &a_brk);
165: if (val && strlen(val)) {
166: if (psVal)
167: strlcpy(psVal, val, *pnLen);
168: if (pnLen)
169: *pnLen = strlen(val);
170: } else
171: def = IS_DEF;
172:
173: e_free(Shared);
174: return i | def;
175: }
176: }
177:
178: e_free(Shared);
179: return 0;
180: }
181:
182: /*
183: * sess_DelValue() Delete item from session shared memory
184: *
185: * @s = Session item
186: * @csAttr = Attribute for erasing
187: * return: 0 Ok, -1 error: in parameter
188: */
189: int
190: sess_DelValue(ait_sess_t * __restrict s, const char *csAttr)
191: {
192: register int i;
193: int attrlen;
194: char *Buffer, *Shared, szAttr[MAX_ATTRIBUTE];
195: char *peer, *p_brk;
196:
197: if (!s || !csAttr || !*csAttr)
198: return -1;
199: else
200: attrlen = strlen(csAttr);
201: Buffer = Shared = NULL;
202: strlcpy(szAttr, csAttr, sizeof szAttr);
203: strlcat(szAttr, "=", sizeof szAttr);
204:
205: Buffer = e_malloc(s->eom);
206: if (!Buffer) {
207: LOGERR;
208: return -1;
209: } else
210: memset(Buffer, 0, s->eom);
211: Shared = e_malloc(s->eom);
212: if (!Shared) {
213: LOGERR;
214: e_free(Buffer);
215: return -1;
216: } else {
217: DEC_SEM(s);
218: memcpy(Shared, s->addr, s->eom);
219: }
220:
221: for (i = 1, peer = strtok_r(Shared, MEM_DELIM"\r\n", &p_brk); peer;
222: i++, peer = strtok_r(NULL, MEM_DELIM"\r\n", &p_brk)) {
223: if (!strncmp(peer, csAttr, attrlen))
224: if (peer[attrlen] == '=' || peer[attrlen] == *MEM_DELIM || !peer[attrlen] ||
225: peer[attrlen] == '\r' || peer[attrlen] == '\n')
226: continue;
227:
228: strlcat(Buffer, peer, s->eom);
229: strlcat(Buffer, MEM_DELIM, s->eom);
230: }
231:
232: memcpy(s->addr, Buffer, s->eom);
233:
234: if (s->type == SHARED_MAP)
235: msync(s->addr, 0, MS_SYNC | MS_INVALIDATE);
236:
237: INC_SEM(s);
238: e_free(Shared);
239: e_free(Buffer);
240: return 0;
241: }
242:
243: /*
244: * sess_SetValue() Set item into session shared memory or update if find it
245: *
246: * @s = Session item
247: * @csAttr = Attribute
248: * @psVal = Value
249: * return: 0 nothing, -1 error: in parameter,
250: >0 set position, if add item merged with IS_ADD and if define item merged with IS_DEF
251: */
252: int
253: sess_SetValue(ait_sess_t * __restrict s, const char *csAttr, const char *psVal)
254: {
255: register int i;
256: int upd, def = IS_VAL;
257: char *Buffer, *Shared, szAttr[MAX_ATTRIBUTE];
258: char *peer, *p_brk;
259:
260: if (!s || !csAttr || !*csAttr)
261: return -1;
262: else
263: Buffer = Shared = NULL;
264: strlcpy(szAttr, csAttr, sizeof szAttr);
265: strlcat(szAttr, "=", sizeof szAttr);
266:
267: Buffer = e_malloc(s->eom);
268: if (!Buffer) {
269: LOGERR;
270: return -1;
271: } else
272: memset(Buffer, 0, s->eom);
273: Shared = e_malloc(s->eom);
274: if (!Shared) {
275: LOGERR;
276: e_free(Buffer);
277: return -1;
278: } else {
279: DEC_SEM(s);
280: memcpy(Shared, s->addr, s->eom);
281: }
282:
283: for (i = 1, upd = 0, peer = strtok_r(Shared, MEM_DELIM"\r\n", &p_brk); peer;
284: i++, peer = strtok_r(NULL, MEM_DELIM"\r\n", &p_brk)) {
285: if (!strncmp(peer, szAttr, strlen(szAttr))) {
286: upd++;
287: if (psVal) {
288: strlcat(Buffer, szAttr, s->eom);
289: strlcat(Buffer, psVal, s->eom);
290: strlcat(Buffer, MEM_DELIM, s->eom);
291: } else {
292: strlcat(Buffer, csAttr, s->eom);
293: strlcat(Buffer, MEM_DELIM, s->eom);
294: def = IS_DEF;
295: }
296: continue;
297: }
298:
299: strlcat(Buffer, peer, s->eom);
300: strlcat(Buffer, MEM_DELIM, s->eom);
301: }
302:
303: if (!upd) {
304: if (psVal) {
305: strlcat(Buffer, szAttr, s->eom);
306: strlcat(Buffer, psVal, s->eom);
307: strlcat(Buffer, MEM_DELIM, s->eom);
308: } else {
309: strlcat(Buffer, csAttr, s->eom);
310: strlcat(Buffer, MEM_DELIM, s->eom);
311: def = IS_DEF;
312: }
313: def |= IS_ADD;
314: }
315:
316: memcpy(s->addr, Buffer, s->eom);
317:
318: if (s->type == SHARED_MAP)
319: msync(s->addr, 0, MS_SYNC | MS_INVALIDATE);
320:
321: INC_SEM(s);
322: e_free(Shared);
323: e_free(Buffer);
324: return upd | def;
325: }
326:
327:
328: /*
329: * sess_prepareSession() Attach to shared memory and de-marshaling data
330: *
331: * @s = Session
332: * @useDirect = Use direct shared memory if !=0 or snapshot of data to array
333: * return: NULL error or no data, !=NULL array with variables,
334: * after use must free resources with sess_doneSession()
335: */
336: array_t *
337: sess_prepareSession(ait_sess_t * __restrict s, char useDirect)
338: {
339: array_t *arr = NULL;
340: sess_hdr_t *hdr;
341:
342: assert(s);
343: if (!s) {
344: sess_SetErr(EINVAL, "Error:: invalid argument\n");
345: return NULL;
346: }
347: if (s->addr) {
348: sess_SetErr(EINVAL, "Error:: already attached memory\n");
349: return NULL;
350: }
351:
352: ATTACH_MEMORY(s);
353: if (!s->addr)
354: return NULL;
355: else
356: hdr = (sess_hdr_t*) s->addr;
357: if (hdr->hdr_magic != SESS_AIT_MAGIC) {
358: DETACH_MEMORY(s);
359:
360: sess_SetErr(EINVAL, "Error:: shared memory not contains values with proper format\n");
361: return NULL;
362: }
363:
364: DEC_SEM(s);
365: s->zcpy = useDirect;
366: arr = ait_map2vars(s->addr + sizeof(sess_hdr_t), s->eom - sizeof(sess_hdr_t),
367: hdr->hdr_argc, useDirect);
368: INC_SEM(s);
369:
370: if (!s->zcpy)
371: DETACH_MEMORY(s);
372: return arr;
373: }
374:
375: /*
376: * sess_doneSession() Free resources allocated with sess_prepareSession()
377: *
378: * @s = Session
379: * @arr = Array with variables for free
380: * return: none
381: */
382: void
383: sess_doneSession(ait_sess_t * __restrict s, array_t ** __restrict arr)
384: {
385: assert(s);
386: if (!s) {
387: sess_SetErr(EINVAL, "Error:: invalid argument\n");
388: return;
389: }
390:
391: if (!s->zcpy)
392: array_Free(*arr);
393: else
394: DETACH_MEMORY(s);
395: array_Destroy(arr);
396: }
397:
398: /*
399: * sess_commitSession() Commit data to shared memory
400: *
401: * @s = Session
402: * @arr = Array with variables for save
403: * return -1 error or !=-1 size of stored variables into shared memory
404: */
405: int
406: sess_commitSession(ait_sess_t * __restrict s, array_t * __restrict arr)
407: {
408: sess_hdr_t *hdr;
409: int ret = 0;
410:
411: assert(s && arr);
412: if (!s || !arr) {
413: sess_SetErr(EINVAL, "Error:: invalid argument\n");
414: return -1;
415: }
416:
417: ATTACH_MEMORY(s);
418: if (!s->addr) {
419: DETACH_MEMORY(s);
420: return -1;
421: } else
422: hdr = (sess_hdr_t*) s->addr;
423:
424: DEC_SEM(s);
425: if ((ret = ait_vars2map(s->addr + sizeof(sess_hdr_t),
426: s->eom - sizeof(sess_hdr_t), arr)) != -1) {
427: hdr->hdr_magic = SESS_AIT_MAGIC;
428: hdr->hdr_argc = array_Size(arr);
429: ret += sizeof(sess_hdr_t);
430: }
431: INC_SEM(s);
432:
433: DETACH_MEMORY(s);
434: return ret;
435: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>