Annotation of libaitsess/src/aitsess.c, revision 1.3.2.1
1.2 misho 1: /*************************************************************************
2: * (C) 2008 AITNET ltd - Sofia/Bulgaria - <misho@aitbg.com>
3: * by Michael Pounov <misho@openbsd-bg.org>
4: *
5: * $Author: misho $
1.3.2.1 ! misho 6: * $Id: aitsess.c,v 1.3 2011/09/07 13:11:56 misho Exp $
1.2 misho 7: *
1.3 misho 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
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: */
1.1 misho 46: #include "global.h"
47: #include "aitsess.h"
48:
49:
50: #pragma GCC visibility push(hidden)
51:
1.3 misho 52: int sess_Errno;
53: char sess_Error[STRSIZ];
1.1 misho 54:
55: #pragma GCC visibility pop
56:
57: // -----------------------------------------------------------
58:
59: // Error maintenance functions ...
60:
61: // sess_GetErrno() Get error code of last operation
1.3 misho 62: inline int
63: sess_GetErrno()
1.1 misho 64: {
1.3 misho 65: return sess_Errno;
1.1 misho 66: }
67: // sess_GetError() Get error text of last operation
1.3 misho 68: inline const char *
69: sess_GetError()
1.1 misho 70: {
1.3 misho 71: return sess_Error;
1.1 misho 72: }
1.3 misho 73: // sess_SetErr() Set error to variables for internal use!!!
74: inline void
75: sess_SetErr(int eno, char *estr, ...)
1.1 misho 76: {
77: va_list lst;
78:
1.3 misho 79: sess_Errno = eno;
80: memset(sess_Error, 0, sizeof sess_Error);
81: va_start(lst, estr);
82: vsnprintf(sess_Error, sizeof sess_Error, estr, lst);
1.1 misho 83: va_end(lst);
84: }
85:
86: // -----------------------------------------------------------
87:
88: /*
89: * initSession() Initializing session structure, if session file not exists creating with specified tech
1.3 misho 90: * @pnID = Technology using in session.
91: SHARED_IPC IPC tech; SHARED_MAP BSD MemoryMap tech or if =NULL SHARED_IPC
1.1 misho 92: * @csFName = Session filename for build key and identified
1.3 misho 93: * @Sess = Session item, if =NULL allocate memory for session after use must be free!
1.1 misho 94: * return: 0 OK new key created, -1 error: no memory or file not created, 1 OK key finded
95: */
1.3 misho 96: inline int
97: initSession(int *pnID, const char *csFName, tagSess ** __restrict Sess)
1.1 misho 98: {
1.3 misho 99: int h, ret = 0, id = SHARED_IPC;
100: char szStr[STRSIZ];
1.1 misho 101:
102: if (!*Sess) {
103: *Sess = malloc(sizeof(tagSess));
104: if (!*Sess) {
105: LOGERR;
106: return -1;
107: }
1.3.2.1 ! misho 108: }
! 109: memset(*Sess, 0, sizeof(tagSess));
1.3 misho 110: if (pnID && *pnID)
111: id = *pnID;
1.1 misho 112:
1.3 misho 113: h = open(csFName, O_WRONLY | O_CREAT | O_EXCL, MEM_MODE);
1.1 misho 114: if (h == -1) {
1.3 misho 115: if (errno != EEXIST) {
116: LOGERR;
117: free(*Sess);
118: return -1;
119: }
120: /* If key file exist, session already connected */
121: h = open(csFName, O_RDONLY);
122: if (h == -1) {
123: LOGERR;
124: free(*Sess);
125: return -1;
126: }
127: ret = read(h, szStr, sizeof szStr);
128: if (ret == -1) {
1.1 misho 129: LOGERR;
130: close(h);
131: free(*Sess);
132: return -1;
1.3 misho 133: }
134: if (!strncmp(szStr, "IPC@", 4))
135: id = SHARED_IPC;
136: else if (!strncmp(szStr, "MAP@", 4))
137: id = SHARED_MAP;
138: else {
139: sess_SetErr(EPROTONOSUPPORT, "Error:: Session type not supported");
140: close(h);
141: free(*Sess);
142: return -1;
143: }
144:
145: ret = 1; /* key found */
146: } else {
147: /* Build new key & new session */
148: switch (id) {
149: case SHARED_IPC:
150: strlcpy(szStr, "IPC@", sizeof szStr);
151: break;
152: case SHARED_MAP:
153: strlcpy(szStr, "MAP@", sizeof szStr);
154: break;
155: default:
156: sess_SetErr(EPROTONOSUPPORT, "Error:: Session type not supported");
157: close(h);
158: unlink(csFName);
159: free(*Sess);
160: return -1;
161: }
162: strlcat(szStr, "ELWIX_Session ("PACKAGE_STRING")\n", sizeof szStr);
163: write(h, szStr, strlen(szStr));
164:
165: ret = 0; /* new key created */
1.1 misho 166: }
167: close(h);
168:
1.3 misho 169: (*Sess)->type = id;
1.1 misho 170: return ret;
171: }
172:
173: /*
174: * freeSession() Free allocated memory for session item and delete session file if present name
175: * @csFName = Session filename for delete, if NULL nothing delete
176: * @Sess = Session item
177: */
1.3 misho 178: inline void
179: freeSession(const char *csFName, tagSess ** __restrict Sess)
1.1 misho 180: {
1.3 misho 181: assert(Sess);
182: if (!Sess)
183: return;
184:
185: (*Sess)->type = SHARED_UNKNOWN;
1.1 misho 186: if (csFName)
187: unlink(csFName);
188: if (*Sess)
189: free(*Sess);
190: *Sess = NULL;
191: }
192:
193:
194: /*
195: * map_createSession() MMAP Created session and allocated resources
196: * @csFName = Session name for identified
197: * @cnSeed = Seed for securing key
198: * @cnSize = Allocated shared memory size in bytes
199: * @Sess = Session item
200: * return: 0 Ok successful, -1 error: not allocated resources
201: */
1.3 misho 202: int
203: map_createSession(const char *csFName, const int cnSeed, const u_int cnSize, tagSess ** __restrict Sess)
1.1 misho 204: {
1.3 misho 205: int ret = 0, id = SHARED_MAP;
206: char szSName[2][FILENAME_MAX];
1.1 misho 207:
1.3 misho 208: ret = initSession(&id, csFName, Sess);
1.1 misho 209: if (ret == -1 || !*Sess)
210: return -1;
211:
1.3 misho 212: /* genkey */
1.1 misho 213: (*Sess)->key = ftok(csFName, cnSeed);
214: if ((*Sess)->key == -1) {
215: LOGERR;
216: freeSession(csFName, Sess);
217: return -1;
218: }
219:
1.3 misho 220: /* build semaphore & shared memory name */
221: memset(szSName, 0, sizeof szSName);
222: snprintf(szSName[0], MAX_SEMNAME, "/%X.ANS", (u_int) (*Sess)->key);
223: snprintf(szSName[1], FILENAME_MAX, "%s-%x.ANM", csFName, (u_int) (*Sess)->key);
1.1 misho 224:
1.3 misho 225: /* create semaphore & add 1 */
1.2 misho 226: (*Sess)->id.sid = sem_open(szSName[0], O_CREAT, MEM_MODE);
1.1 misho 227: if ((*Sess)->id.sid == SEM_FAILED) {
228: LOGERR;
229: map_destroySession(csFName, Sess);
230: return -1;
231: } else
232: sem_post((*Sess)->id.sid);
233:
1.3 misho 234: /* create file for shared memory storage */
1.2 misho 235: (*Sess)->mem.fd = open(szSName[1], O_RDWR | O_CREAT, MEM_MODE);
1.1 misho 236: if ((*Sess)->mem.fd == -1) {
237: LOGERR;
238: map_destroySession(csFName, Sess);
239: return -1;
240: }
1.3 misho 241: /* if is new shared memory session, fill file with zeros */
1.1 misho 242: if (!ret) {
1.3 misho 243: if (lseek((*Sess)->mem.fd, cnSize - 1, SEEK_SET) == -1) {
1.1 misho 244: LOGERR;
245: map_destroySession(csFName, Sess);
246: return -1;
1.3 misho 247: } else
248: write((*Sess)->mem.fd, "", 1);
249: lseek((*Sess)->mem.fd, 0, SEEK_SET);
1.1 misho 250: }
251: (*Sess)->eom = cnSize;
252:
253: return ret;
254: }
255:
256: /*
257: * map_destroySession() MMAP free shared resources
258: * @csFName = Session name for delete
259: * @Sess = Session item
260: */
1.3 misho 261: void
262: map_destroySession(const char *csFName, tagSess ** __restrict Sess)
1.1 misho 263: {
264: int flg = 1;
1.3 misho 265: char szSName[2][FILENAME_MAX];
1.1 misho 266:
1.3 misho 267: assert(Sess);
268: if (!Sess || !*Sess)
1.1 misho 269: return;
270:
1.3 misho 271: memset(szSName, 0, sizeof szSName);
272: snprintf(szSName[0], MAX_SEMNAME, "/%X.ANS", (u_int) (*Sess)->key);
273: snprintf(szSName[1], FILENAME_MAX, "%s-%x.ANM", csFName, (u_int) (*Sess)->key);
1.1 misho 274:
275: if ((*Sess)->id.sid != SEM_FAILED) {
276: if (sem_close((*Sess)->id.sid) == -1)
277: flg = 0;
278: if (sem_unlink(szSName[0]) == -1)
279: /*flg = 0*/;
280: }
281: if ((*Sess)->mem.fd != -1) {
282: if (close((*Sess)->mem.fd) == -1)
283: flg = 0;
284: if (unlink(szSName[1]) == -1)
285: /*flg = 0*/;
286: }
287: (*Sess)->eom ^= (*Sess)->eom;
288:
289: freeSession(flg ? csFName : NULL, Sess);
290: }
291:
292: /*
293: * ipc_createSession() IPC Created session and allocated resources
294: * @csFName = Session name for identified
295: * @cnSeed = Seed for securing key
296: * @cnSize = Allocated shared memory size in bytes
297: * @Sess = Session item
298: * return: 0 Ok successful, -1 error: not allocated resources
299: */
1.3 misho 300: int
301: ipc_createSession(const char *csFName, const int cnSeed, const u_int cnSize, tagSess ** __restrict Sess)
1.1 misho 302: {
1.3 misho 303: int ret = 0, id = SHARED_IPC;
1.1 misho 304: union semun sems;
305:
1.3 misho 306: ret = initSession(&id, csFName, Sess);
1.1 misho 307: if (ret == -1 || !*Sess)
308: return -1;
309:
1.3 misho 310: /* genkey */
1.1 misho 311: (*Sess)->key = ftok(csFName, cnSeed);
312: if ((*Sess)->key == -1) {
313: LOGERR;
314: freeSession(csFName, Sess);
315: return -1;
316: }
317:
1.3 misho 318: /* create semaphore */
1.2 misho 319: (*Sess)->id.semid = semget((*Sess)->key, 1, MEM_MODE | IPC_CREAT);
1.1 misho 320: if ((*Sess)->id.semid == -1) {
321: LOGERR;
322: ipc_destroySession(csFName, Sess);
323: return -1;
324: }
1.3 misho 325: /* if is new shared memory session, init sempahore with 1 */
1.1 misho 326: if (!ret) {
327: sems.val = 1;
328: if (semctl((*Sess)->id.semid, 0, SETVAL, sems) == -1) {
329: LOGERR;
330: ipc_destroySession(csFName, Sess);
331: return -1;
332: }
333: }
334:
1.3 misho 335: /* create shared memory object */
1.2 misho 336: (*Sess)->mem.shmid = shmget((*Sess)->key, cnSize, MEM_MODE | IPC_CREAT);
1.1 misho 337: if ((*Sess)->mem.shmid == -1) {
338: LOGERR;
339: ipc_destroySession(csFName, Sess);
340: return -1;
341: }
342: (*Sess)->eom = cnSize;
343:
344: return ret;
345: }
346:
347: /*
348: * ipc_destroySession() IPC free shared resources
349: * @csFName = Session name for delete
350: * @Sess = Session item
351: */
1.3 misho 352: void
353: ipc_destroySession(const char *csFName, tagSess ** __restrict Sess)
1.1 misho 354: {
355: int flg = 1;
356: union semun sems;
357: struct shmid_ds ds;
358:
1.3 misho 359: assert(Sess);
360: if (!Sess || !*Sess)
1.1 misho 361: return;
362:
363: if ((*Sess)->id.semid != -1)
364: if (semctl((*Sess)->id.semid, 0, IPC_RMID, &sems) == -1)
365: flg = 0;
366: if ((*Sess)->mem.shmid != -1)
367: if (shmctl((*Sess)->mem.shmid, IPC_RMID, &ds) == -1)
368: flg = 0;
369: (*Sess)->eom ^= (*Sess)->eom;
370:
371: freeSession(flg ? csFName : NULL, Sess);
372: }
373:
374:
375: /*
376: * map_attachSession() MMAP Attach to shared memory & return begin address
377: * @s = Session item
378: * @procMem = Custom start address (optionl) *default must be 0*
379: * return: NULL failed attach, !=NULL begin address of memory
380: */
1.3 misho 381: inline void *
382: map_attachSession(tagSess * __restrict s, void *procMem)
1.1 misho 383: {
384: struct stat sb;
385:
386: if (!s)
387: return NULL;
388:
1.3 misho 389: /* Learn size of shared memory block */
1.1 misho 390: sync();
391: if (fstat(s->mem.fd, &sb) == -1) {
392: LOGERR;
393: return NULL;
394: } else
395: s->eom = sb.st_size;
396:
1.3 misho 397: /* attach to memory */
1.1 misho 398: s->addr = mmap(procMem, s->eom, PROT_READ | PROT_WRITE, MAP_SHARED, s->mem.fd, 0);
399: if (s->addr == MAP_FAILED) {
400: LOGERR;
401: s->addr = NULL;
402: }
403:
404: return s->addr;
405: }
406:
407: /*
408: * map_detachSession() MMAP Detach from shared memory
409: * @s = Session item
410: */
1.3 misho 411: inline void
412: map_detachSession(tagSess * __restrict s)
1.1 misho 413: {
414: if (!s)
415: return;
416:
417: msync(s->addr, 0, MS_SYNC | MS_INVALIDATE);
418:
419: if (s->addr && s->eom) {
420: munmap(s->addr, s->eom);
421: s->addr = NULL;
422: }
423: }
424:
425: /*
426: * ipc_attachSession() IPC Attach to shared memory & return begin address
427: * @s = Session item
428: * @procMem = Custom start address (optionl) *default must be 0*
429: * return: NULL failed attach, !=NULL begin address of memory
430: */
1.3 misho 431: inline void *
432: ipc_attachSession(tagSess * __restrict s, void *procMem)
1.1 misho 433: {
434: if (!s)
435: return NULL;
436:
437: s->addr = shmat(s->mem.shmid, procMem, 0);
438: if (s->addr == (void*) -1) {
439: LOGERR;
440: s->addr = NULL;
441: }
442:
443: return s->addr;
444: }
445:
446: /*
447: * ipc_detachSession() IPC Detach from shared memory
448: * @s = Session item
449: */
1.3 misho 450: inline void
451: ipc_detachSession(tagSess * __restrict s)
1.1 misho 452: {
453: if (!s)
454: return;
455:
456: if (s->addr) {
457: shmdt(s->addr);
458: s->addr = NULL;
459: }
460: }
461:
1.2 misho 462: /*
463: * isAttached() Check for mapped/(attached) shared memory
464: * @s = Session item
465: * return: -1 null session item, 0 not attached, 1 attached memory
466: */
1.3 misho 467: inline int
468: isAttached(tagSess * __restrict s)
1.2 misho 469: {
470: if (!s)
471: return -1;
472:
473: return (s->addr ? 1 : 0);
474: }
475:
1.1 misho 476:
477: /*
478: * map_notSemaphore() MMAP negative block if semaphore isn`t signaled
479: * @s = Session item
480: */
1.3 misho 481: inline void
482: map_notSemaphore(tagSess * __restrict s)
1.1 misho 483: {
484: int i = -1;
485:
486: if (!s)
487: return;
488:
489: sem_getvalue(s->id.sid, &i);
490: for (;i > 0; i--)
491: sem_wait(s->id.sid);
492: }
493:
494: /*
495: * map_isSemaphored() MMAP Check semaphore
496: * @s = Session item
497: * return: -1 error: can`t return semaphore, 0 = false, 1 = true
498: */
1.3 misho 499: inline int
500: map_isSemaphored(tagSess * __restrict s)
1.1 misho 501: {
502: int val = -1;
503:
504: if (!s)
505: return -1;
506:
507: sem_getvalue(s->id.sid, &val);
508: return val ? 0 : 1;
509: }
510:
511: /*
512: * map_addSemaphore() MMAP unblock semaphore, increment semaphore
513: * @s = Session item
514: * return: 0 Ok, -1 error: can`t increment
515: */
1.3 misho 516: inline int
517: map_addSemaphore(tagSess * __restrict s)
1.1 misho 518: {
519: if (!s)
520: return -1;
521:
522: return sem_post(s->id.sid);
523: }
524:
525: /*
526: * map_decSemaphore() MMAP block semaphore, decrement semaphore
527: * @s = Session item
528: * return: 0 Ok, -1 error: can`t decrement
529: */
1.3 misho 530: inline int
531: map_decSemaphore(tagSess * __restrict s)
1.1 misho 532: {
533: if (!s)
534: return -1;
535:
536: return sem_wait(s->id.sid);
537: }
538:
539: /*
540: * ipc_notSemaphore() IPC negative block if semaphore isn`t signaled
541: * @s = Session item
542: */
1.3 misho 543: inline void
544: ipc_notSemaphore(tagSess * __restrict s)
1.1 misho 545: {
546: struct sembuf sb = { 0, 0, 0 };
547:
548: if (s)
549: semop(s->id.semid, &sb, 1);
550: }
551:
552: /*
553: * ipc_isSemaphored() IPC Check semaphore
554: * @s = Session item
555: * return: -1 error: can`t return semaphore, 0 = false, 1 = true
556: */
1.3 misho 557: inline int
558: ipc_isSemaphored(tagSess * __restrict s)
1.1 misho 559: {
560: struct sembuf sb = { 0, 0, IPC_NOWAIT };
561:
562: if (!s)
563: return -1;
564:
565: return semop(s->id.semid, &sb, 1) + 1;
566: }
567:
568: /*
569: * ipc_addSemaphore() IPC unblock semaphore, increment semaphore
570: * @s = Session item
571: * return: 0 Ok, -1 error: can`t increment
572: */
1.3 misho 573: inline int
574: ipc_addSemaphore(tagSess * __restrict s)
1.1 misho 575: {
576: struct sembuf sb = { 0, 1, 0 };
577:
578: if (!s)
579: return -1;
580:
581: return semop(s->id.semid, &sb, 1);
582: }
583:
584: /*
585: * ipc_decSemaphore() IPC block semaphore, decrement semaphore
586: * @s = Session item
587: * return: 0 Ok, -1 error: can`t decrement
588: */
1.3 misho 589: inline int
590: ipc_decSemaphore(tagSess * __restrict s)
1.1 misho 591: {
592: struct sembuf sb = { 0, -1, 0 };
593:
594: if (!s)
595: return -1;
596:
597: return semop(s->id.semid, &sb, 1);
598: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>
![[BACK]](/icons/cvsweb/back.gif){kind=icon}
Return to aitsess.c CVS log ![[TXT]](/icons/cvsweb/text.gif){kind=icon}
![[DIR]](/icons/cvsweb/dir.gif){kind=icon}
Up to [ELWIX - Embedded LightWeight unIX -] / libaitsess / src