Annotation of libaitio/src/sess.c, revision 1.2
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 $
! 6: * $Id: sess.c,v 1.1.2.4 2013/03/13 14:42:28 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_initSession() - Initializing session structure,
! 51: * if session file not exists creating with specified tech
! 52: *
! 53: * @id = Technology using in session. SHARED_IPC IPC tech or SHARED_MAP BSD MemoryMap tech
! 54: * @csFName = Session filename for build key and identified
! 55: * @Sess = Session item, if =NULL allocate memory for session after use must be free!
! 56: * return: 0 OK new key created, -1 error: no memory or file not created, 1 OK key finded
! 57: */
! 58: int
! 59: sess_initSession(int id, const char *csFName, ait_sess_t ** __restrict Sess)
! 60: {
! 61: int h, ret = 0;
! 62: char szStr[STRSIZ];
! 63:
! 64: if (!csFName) {
! 65: io_SetErr(EINVAL, "Filename is NULL");
! 66: return -1;
! 67: }
! 68: if (id < SHARED_UNKNOWN || id > SHARED_MAP) {
! 69: io_SetErr(EPROTONOSUPPORT, "Session type not supported");
! 70: return -1;
! 71: }
! 72:
! 73: if (!*Sess) {
! 74: *Sess = e_malloc(sizeof(ait_sess_t));
! 75: if (!*Sess) {
! 76: LOGERR;
! 77: return -1;
! 78: }
! 79: }
! 80: memset(*Sess, 0, sizeof(ait_sess_t));
! 81: strlcpy((*Sess)->name, csFName, sizeof (*Sess)->name);
! 82:
! 83: h = open((*Sess)->name, O_WRONLY | O_CREAT | O_EXCL, SESS_MEM_MODE);
! 84: if (h == -1) {
! 85: if (errno != EEXIST) {
! 86: LOGERR;
! 87: e_free(*Sess);
! 88: return -1;
! 89: }
! 90: /* If key file exist, session already connected */
! 91: h = open((*Sess)->name, O_RDONLY);
! 92: if (h == -1) {
! 93: LOGERR;
! 94: e_free(*Sess);
! 95: return -1;
! 96: }
! 97: ret = read(h, szStr, sizeof szStr);
! 98: if (ret == -1) {
! 99: LOGERR;
! 100: close(h);
! 101: e_free(*Sess);
! 102: return -1;
! 103: }
! 104: if (!strncmp(szStr, "IPC@", 4) && id == SHARED_IPC) {
! 105: ret = 1;
! 106:
! 107: (*Sess)->sess.create = (int (*)(int, long, void*, ...)) ipc_createSession;
! 108: (*Sess)->sess.destroy = (void (*)(void*)) ipc_destroySession;
! 109: (*Sess)->sess.attach = (void* (*)(void*, void*)) ipc_attachSession;
! 110: (*Sess)->sess.detach = (void (*)(void*)) ipc_detachSession;
! 111: (*Sess)->sess.notSem = (void (*)(void*)) ipc_notSemaphore;
! 112: (*Sess)->sess.isSemOK = (int (*)(void*)) ipc_isSemaphoreOK;
! 113: (*Sess)->sess.incSem = (int (*)(void*)) ipc_incSemaphore;
! 114: (*Sess)->sess.decSem = (int (*)(void*)) ipc_decSemaphore;
! 115: } else if (!strncmp(szStr, "MAP@", 4) && id == SHARED_MAP) {
! 116: ret = 1;
! 117:
! 118: (*Sess)->sess.create = (int (*)(int, long, void*, ...)) map_createSession;
! 119: (*Sess)->sess.destroy = (void (*)(void*)) map_destroySession;
! 120: (*Sess)->sess.attach = (void* (*)(void*, void*)) map_attachSession;
! 121: (*Sess)->sess.detach = (void (*)(void*)) map_detachSession;
! 122: (*Sess)->sess.notSem = (void (*)(void*)) map_notSemaphore;
! 123: (*Sess)->sess.isSemOK = (int (*)(void*)) map_isSemaphoreOK;
! 124: (*Sess)->sess.incSem = (int (*)(void*)) map_incSemaphore;
! 125: (*Sess)->sess.decSem = (int (*)(void*)) map_decSemaphore;
! 126: } else {
! 127: io_SetErr(EPROTONOSUPPORT,
! 128: "Session type not supported or wrong session type");
! 129: close(h);
! 130: e_free(*Sess);
! 131: return -1;
! 132: }
! 133: /* key found */
! 134: } else {
! 135: /* Build new key & new session */
! 136: if (id == SHARED_IPC) {
! 137: strlcpy(szStr, "IPC@", sizeof szStr);
! 138:
! 139: (*Sess)->sess.create = (int (*)(int, long, void*, ...)) ipc_createSession;
! 140: (*Sess)->sess.destroy = (void (*)(void*)) ipc_destroySession;
! 141: (*Sess)->sess.attach = (void* (*)(void*, void*)) ipc_attachSession;
! 142: (*Sess)->sess.detach = (void (*)(void*)) ipc_detachSession;
! 143: (*Sess)->sess.notSem = (void (*)(void*)) ipc_notSemaphore;
! 144: (*Sess)->sess.isSemOK = (int (*)(void*)) ipc_isSemaphoreOK;
! 145: (*Sess)->sess.incSem = (int (*)(void*)) ipc_incSemaphore;
! 146: (*Sess)->sess.decSem = (int (*)(void*)) ipc_decSemaphore;
! 147: } else if (id == SHARED_MAP) {
! 148: strlcpy(szStr, "MAP@", sizeof szStr);
! 149:
! 150: (*Sess)->sess.create = (int (*)(int, long, void*, ...)) map_createSession;
! 151: (*Sess)->sess.destroy = (void (*)(void*)) map_destroySession;
! 152: (*Sess)->sess.attach = (void* (*)(void*, void*)) map_attachSession;
! 153: (*Sess)->sess.detach = (void (*)(void*)) map_detachSession;
! 154: (*Sess)->sess.notSem = (void (*)(void*)) map_notSemaphore;
! 155: (*Sess)->sess.isSemOK = (int (*)(void*)) map_isSemaphoreOK;
! 156: (*Sess)->sess.incSem = (int (*)(void*)) map_incSemaphore;
! 157: (*Sess)->sess.decSem = (int (*)(void*)) map_decSemaphore;
! 158: } else {
! 159: io_SetErr(EINVAL, "Session type must be specified");
! 160: close(h);
! 161: unlink(csFName);
! 162: e_free(*Sess);
! 163: return -1;
! 164: }
! 165: strlcat(szStr, "ELWIX_Session ("PACKAGE_STRING")\n", sizeof szStr);
! 166: write(h, szStr, strlen(szStr));
! 167:
! 168: ret = 0;
! 169: /* new key created */
! 170: }
! 171: close(h);
! 172:
! 173: (*Sess)->type = id;
! 174: (*Sess)->zcpy = (char) ret;
! 175: return ret;
! 176: }
! 177:
! 178: /*
! 179: * sess_freeSession() - Free allocated memory for session item and delete session file if present name
! 180: *
! 181: * @Sess = Session item
! 182: * return: none
! 183: */
! 184: void
! 185: sess_freeSession(ait_sess_t ** __restrict Sess)
! 186: {
! 187: if (!Sess || !(*Sess))
! 188: return;
! 189:
! 190: if ((*Sess)->addr)
! 191: DETACH_MEMORY(*Sess);
! 192:
! 193: /*
! 194: memset(&(*Sess)->sess, 0, sizeof (*Sess)->sess);
! 195:
! 196: (*Sess)->type = SHARED_UNKNOWN;
! 197: */
! 198:
! 199: e_free(*Sess);
! 200: *Sess = NULL;
! 201: }
! 202:
! 203:
! 204: /*
! 205: * map_createSession() - MMAP Created session and allocated resources
! 206: *
! 207: * @nSeed = Seed for securing key, if =-1 must add ready for use key
! 208: * @nSize = Allocated shared memory size in bytes
! 209: * @Sess = Session item
! 210: * @... = If nSeed == -1 add ready for use key value
! 211: * return: 0 Ok successful, -1 error: not allocated resources
! 212: */
! 213: int
! 214: map_createSession(int nSeed, long nSize, ait_sess_t * __restrict Sess, ...)
! 215: {
! 216: char szSName[2][FILENAME_MAX];
! 217: va_list lst;
! 218:
! 219: if (!Sess || !*Sess->name)
! 220: return -1;
! 221:
! 222: if (nSeed != -1) {
! 223: /* genkey */
! 224: Sess->key = ftok(Sess->name, nSeed);
! 225: if (Sess->key == -1) {
! 226: LOGERR;
! 227: return -1;
! 228: }
! 229: } else {
! 230: /* get key from va_args */
! 231: va_start(lst, Sess);
! 232: Sess->key = va_arg(lst, key_t);
! 233: va_end(lst);
! 234: }
! 235:
! 236: /* build semaphore & shared memory name */
! 237: memset(szSName, 0, sizeof szSName);
! 238: snprintf(szSName[0], MAX_SEMNAME, "/%X.ANS", (u_int) Sess->key);
! 239: #ifdef HAVE_SHM_OPEN
! 240: snprintf(szSName[1], FILENAME_MAX, "/%s-%x.ANM", Sess->name, (u_int) Sess->key);
! 241: #else
! 242: snprintf(szSName[1], FILENAME_MAX, "%s-%x.ANM", Sess->name, (u_int) Sess->key);
! 243: #endif
! 244:
! 245: /* create semaphore & add 1 */
! 246: Sess->id.sid = sem_open(szSName[0], O_CREAT, SESS_MEM_MODE);
! 247: if (Sess->id.sid == SEM_FAILED) {
! 248: LOGERR;
! 249: map_destroySession(Sess);
! 250: return -1;
! 251: }
! 252: /* if is new shared memory session, init sempahore with 1 */
! 253: if (!Sess->zcpy)
! 254: sem_post(Sess->id.sid);
! 255:
! 256: /* create file for shared memory storage */
! 257: #ifdef HAVE_SHM_OPEN
! 258: Sess->mem.fd = shm_open(szSName[1], O_RDWR | O_CREAT, SESS_MEM_MODE);
! 259: #else
! 260: Sess->mem.fd = open(szSName[1], O_RDWR | O_CREAT, SESS_MEM_MODE);
! 261: #endif
! 262: if (Sess->mem.fd == -1) {
! 263: LOGERR;
! 264: map_destroySession(Sess);
! 265: return -1;
! 266: }
! 267: if (!Sess->zcpy) {
! 268: #ifdef HAVE_SHM_OPEN
! 269: if (ftruncate(Sess->mem.fd, nSize) == -1) {
! 270: LOGERR;
! 271: map_destroySession(Sess);
! 272: return -1;
! 273: }
! 274: #else
! 275: /* if is new shared memory session, fill file with zeros */
! 276: if (lseek(Sess->mem.fd, nSize - 1, SEEK_SET) == -1) {
! 277: LOGERR;
! 278: map_destroySession(Sess);
! 279: return -1;
! 280: } else
! 281: write(Sess->mem.fd, "", 1);
! 282: lseek(Sess->mem.fd, 0, SEEK_SET);
! 283: #endif
! 284: }
! 285: Sess->eom = nSize;
! 286:
! 287: return (int) Sess->zcpy;
! 288: }
! 289:
! 290: /*
! 291: * map_destroySession() - MMAP free shared resources
! 292: *
! 293: * @Sess = Session item
! 294: * return: none
! 295: */
! 296: void
! 297: map_destroySession(ait_sess_t * __restrict Sess)
! 298: {
! 299: char szSName[2][FILENAME_MAX];
! 300:
! 301: if (!Sess || sess_isAttached(Sess) || !*Sess->name)
! 302: return;
! 303:
! 304: memset(szSName, 0, sizeof szSName);
! 305: snprintf(szSName[0], MAX_SEMNAME, "/%X.ANS", (u_int) Sess->key);
! 306: #ifdef HAVE_SHM_UNLINK
! 307: snprintf(szSName[1], FILENAME_MAX, "/%s-%x.ANM", Sess->name, (u_int) Sess->key);
! 308: #else
! 309: snprintf(szSName[1], FILENAME_MAX, "%s-%x.ANM", Sess->name, (u_int) Sess->key);
! 310: #endif
! 311:
! 312: if (Sess->id.sid != SEM_FAILED) {
! 313: sem_close(Sess->id.sid);
! 314: sem_unlink(szSName[0]);
! 315: }
! 316: if (Sess->mem.fd > 2) {
! 317: close(Sess->mem.fd);
! 318: #ifdef HAVE_SHM_UNLINK
! 319: shm_unlink(szSName[1]);
! 320: #else
! 321: unlink(szSName[1]);
! 322: #endif
! 323: }
! 324: unlink(Sess->name);
! 325: memset(Sess->name, 0, sizeof Sess->name);
! 326: Sess->eom ^= Sess->eom;
! 327: Sess->key ^= Sess->key;
! 328: }
! 329:
! 330: /*
! 331: * ipc_createSession() - IPC Created session and allocated resources
! 332: *
! 333: * @nSeed = Seed for securing key, if =-1 must add ready for use key
! 334: * @nSize = Allocated shared memory size in bytes
! 335: * @Sess = Session item
! 336: * @... = If nSeed == -1 add ready for use key value
! 337: * return: 0 Ok successful, -1 error: not allocated resources
! 338: */
! 339: int
! 340: ipc_createSession(int nSeed, long nSize, ait_sess_t * __restrict Sess, ...)
! 341: {
! 342: union semun sems;
! 343: va_list lst;
! 344:
! 345: if (!Sess || !*Sess->name)
! 346: return -1;
! 347:
! 348: if (nSeed != -1) {
! 349: /* genkey */
! 350: Sess->key = ftok(Sess->name, nSeed);
! 351: if (Sess->key == -1) {
! 352: LOGERR;
! 353: return -1;
! 354: }
! 355: } else {
! 356: /* get key from va_args */
! 357: va_start(lst, Sess);
! 358: Sess->key = va_arg(lst, key_t);
! 359: va_end(lst);
! 360: }
! 361:
! 362: /* create semaphore */
! 363: Sess->id.semid = semget(Sess->key, 1, SESS_MEM_MODE | IPC_CREAT);
! 364: if (Sess->id.semid == -1) {
! 365: LOGERR;
! 366: ipc_destroySession(Sess);
! 367: return -1;
! 368: }
! 369: /* if is new shared memory session, init sempahore with 1 */
! 370: if (!Sess->zcpy) {
! 371: sems.val = 1;
! 372: if (semctl(Sess->id.semid, 0, SETVAL, sems) == -1) {
! 373: LOGERR;
! 374: ipc_destroySession(Sess);
! 375: return -1;
! 376: }
! 377: }
! 378:
! 379: /* create shared memory object */
! 380: Sess->mem.shmid = shmget(Sess->key, nSize, SESS_MEM_MODE | IPC_CREAT);
! 381: if (Sess->mem.shmid == -1) {
! 382: LOGERR;
! 383: ipc_destroySession(Sess);
! 384: return -1;
! 385: } else
! 386: Sess->eom = nSize;
! 387:
! 388: return (int) Sess->zcpy;
! 389: }
! 390:
! 391: /*
! 392: * ipc_destroySession() - IPC free shared resources
! 393: *
! 394: * @Sess = Session item
! 395: * return: none
! 396: */
! 397: void
! 398: ipc_destroySession(ait_sess_t * __restrict Sess)
! 399: {
! 400: union semun sems;
! 401: struct shmid_ds ds;
! 402:
! 403: if (!Sess || sess_isAttached(Sess))
! 404: return;
! 405:
! 406: if (Sess->id.semid != -1)
! 407: semctl(Sess->id.semid, 0, IPC_RMID, &sems);
! 408: if (Sess->mem.shmid != -1)
! 409: shmctl(Sess->mem.shmid, IPC_RMID, &ds);
! 410: unlink(Sess->name);
! 411: memset(Sess->name, 0, sizeof Sess->name);
! 412: Sess->eom ^= Sess->eom;
! 413: Sess->key ^= Sess->key;
! 414: }
! 415:
! 416:
! 417: /*
! 418: * map_attachSession() - MMAP Attach to shared memory & return begin address
! 419: *
! 420: * @s = Session item
! 421: * @procMem = Custom start address (optionl) *default must be 0*
! 422: * return: NULL failed attach, !=NULL begin address of memory
! 423: */
! 424: void *
! 425: map_attachSession(ait_sess_t * __restrict s, void *procMem)
! 426: {
! 427: if (!s)
! 428: return NULL;
! 429:
! 430: sync();
! 431:
! 432: /* attach to memory */
! 433: s->addr = mmap(procMem, s->eom, PROT_READ | PROT_WRITE, MAP_SHARED, s->mem.fd, 0);
! 434: if (s->addr == MAP_FAILED) {
! 435: LOGERR;
! 436: s->addr = NULL;
! 437: }
! 438:
! 439: return s->addr;
! 440: }
! 441:
! 442: /*
! 443: * map_detachSession() - MMAP Detach from shared memory
! 444: *
! 445: * @s = Session item
! 446: * return: none
! 447: */
! 448: void
! 449: map_detachSession(ait_sess_t * __restrict s)
! 450: {
! 451: if (!s)
! 452: return;
! 453:
! 454: msync(s->addr, 0, MS_SYNC | MS_INVALIDATE);
! 455:
! 456: if (s->addr && s->eom) {
! 457: munmap(s->addr, s->eom);
! 458: s->addr = NULL;
! 459: }
! 460: }
! 461:
! 462: /*
! 463: * ipc_attachSession() - IPC Attach to shared memory & return begin address
! 464: *
! 465: * @s = Session item
! 466: * @procMem = Custom start address (optionl) *default must be 0*
! 467: * return: NULL failed attach, !=NULL begin address of memory
! 468: */
! 469: void *
! 470: ipc_attachSession(ait_sess_t * __restrict s, void *procMem)
! 471: {
! 472: if (!s)
! 473: return NULL;
! 474:
! 475: s->addr = shmat(s->mem.shmid, procMem, 0);
! 476: if (s->addr == (void*) -1) {
! 477: LOGERR;
! 478: s->addr = NULL;
! 479: }
! 480:
! 481: return s->addr;
! 482: }
! 483:
! 484: /*
! 485: * ipc_detachSession() - IPC Detach from shared memory
! 486: *
! 487: * @s = Session item
! 488: * return: none
! 489: */
! 490: void
! 491: ipc_detachSession(ait_sess_t * __restrict s)
! 492: {
! 493: if (!s)
! 494: return;
! 495:
! 496: if (s->addr) {
! 497: shmdt(s->addr);
! 498: s->addr = NULL;
! 499: }
! 500: }
! 501:
! 502: /*
! 503: * sess_isAttached() - Check for attached shared memory
! 504: *
! 505: * @s = Session item
! 506: * return: -1 null session item, 0 not attached, 1 attached memory
! 507: */
! 508: inline int
! 509: sess_isAttached(ait_sess_t * __restrict s)
! 510: {
! 511: if (!s)
! 512: return -1;
! 513:
! 514: return (s->addr ? 1 : 0);
! 515: }
! 516:
! 517:
! 518: /*
! 519: * map_notSemaphore() - MMAP negative block if semaphore isn`t signaled
! 520: *
! 521: * @s = Session item
! 522: * return: none
! 523: */
! 524: void
! 525: map_notSemaphore(ait_sess_t * __restrict s)
! 526: {
! 527: int i = -1;
! 528:
! 529: if (!s)
! 530: return;
! 531:
! 532: sem_getvalue(s->id.sid, &i);
! 533: while (i > 0) {
! 534: sem_wait(s->id.sid);
! 535: i--;
! 536: }
! 537: }
! 538:
! 539: /*
! 540: * map_isSemaphoreOK() - MMAP Check semaphore
! 541: *
! 542: * @s = Session item
! 543: * return: -1 error: can`t return semaphore, 0 = false, 1 = true
! 544: */
! 545: int
! 546: map_isSemaphoreOK(ait_sess_t * __restrict s)
! 547: {
! 548: int val = -1;
! 549:
! 550: if (!s)
! 551: return -1;
! 552:
! 553: sem_getvalue(s->id.sid, &val);
! 554: return (val ? 0 : 1);
! 555: }
! 556:
! 557: /*
! 558: * map_incSemaphore() - MMAP unblock semaphore, increment semaphore
! 559: *
! 560: * @s = Session item
! 561: * return: 0 Ok, -1 error: can`t increment
! 562: */
! 563: int
! 564: map_incSemaphore(ait_sess_t * __restrict s)
! 565: {
! 566: if (!s)
! 567: return -1;
! 568:
! 569: return sem_post(s->id.sid);
! 570: }
! 571:
! 572: /*
! 573: * map_decSemaphore() - MMAP block semaphore, decrement semaphore
! 574: *
! 575: * @s = Session item
! 576: * return: 0 Ok, -1 error: can`t decrement
! 577: */
! 578: int
! 579: map_decSemaphore(ait_sess_t * __restrict s)
! 580: {
! 581: if (!s)
! 582: return -1;
! 583:
! 584: return sem_wait(s->id.sid);
! 585: }
! 586:
! 587: /*
! 588: * ipc_notSemaphore() - IPC negative block if semaphore isn`t signaled
! 589: *
! 590: * @s = Session item
! 591: * return: none
! 592: */
! 593: void
! 594: ipc_notSemaphore(ait_sess_t * __restrict s)
! 595: {
! 596: struct sembuf sb = { 0, 0, 0 };
! 597:
! 598: if (s)
! 599: semop(s->id.semid, &sb, 1);
! 600: }
! 601:
! 602: /*
! 603: * ipc_isSemaphoreOK() - IPC Check semaphore
! 604: *
! 605: * @s = Session item
! 606: * return: -1 error: can`t return semaphore, 0 = false, 1 = true
! 607: */
! 608: int
! 609: ipc_isSemaphoreOK(ait_sess_t * __restrict s)
! 610: {
! 611: struct sembuf sb = { 0, 0, IPC_NOWAIT };
! 612:
! 613: if (!s)
! 614: return -1;
! 615:
! 616: return semop(s->id.semid, &sb, 1) + 1;
! 617: }
! 618:
! 619: /*
! 620: * ipc_incSemaphore() - IPC unblock semaphore, increment semaphore
! 621: *
! 622: * @s = Session item
! 623: * return: 0 Ok, -1 error: can`t increment
! 624: */
! 625: int
! 626: ipc_incSemaphore(ait_sess_t * __restrict s)
! 627: {
! 628: struct sembuf sb = { 0, 1, 0 };
! 629:
! 630: if (!s)
! 631: return -1;
! 632:
! 633: return semop(s->id.semid, &sb, 1);
! 634: }
! 635:
! 636: /*
! 637: * ipc_decSemaphore() - IPC block semaphore, decrement semaphore
! 638: *
! 639: * @s = Session item
! 640: * return: 0 Ok, -1 error: can`t decrement
! 641: */
! 642: int
! 643: ipc_decSemaphore(ait_sess_t * __restrict s)
! 644: {
! 645: struct sembuf sb = { 0, -1, 0 };
! 646:
! 647: if (!s)
! 648: return -1;
! 649:
! 650: return semop(s->id.semid, &sb, 1);
! 651: }
! 652:
! 653: /* --- A/V management --- */
! 654:
! 655: /*
! 656: * sess_FreeValues() - Free all values from value array allocated from sess_GetValues()
! 657: *
! 658: * @ppsVals = Array strings
! 659: * return: none
! 660: */
! 661: inline void
! 662: sess_FreeValues(char *** __restrict ppsVals)
! 663: {
! 664: str_FreeNullTerm(ppsVals);
! 665: }
! 666:
! 667: /*
! 668: * sess_GetValues() - Get all values from session shared memory
! 669: *
! 670: * @s = Session item
! 671: * @ppsVals = Return array strings
! 672: * return: -1 error: in parameter, !=-1 count of returned strings in ppsVals
! 673: * (must be sess_FreeValues after use!)
! 674: */
! 675: int
! 676: sess_GetValues(ait_sess_t * __restrict s, char ***ppsVals)
! 677: {
! 678: register int i;
! 679: char **valz, *Shared = NULL;
! 680: char *peer, *p_brk;
! 681:
! 682: if (!s || !ppsVals)
! 683: return -1;
! 684: valz = e_malloc(sizeof(caddr_t));
! 685: if (!valz) {
! 686: LOGERR;
! 687: return -1;
! 688: } else
! 689: *valz = NULL;
! 690:
! 691: /* allocated memory & mirrored shared memory into this */
! 692: Shared = e_malloc(s->eom);
! 693: if (!Shared) {
! 694: LOGERR;
! 695: e_free(valz);
! 696: return -1;
! 697: } else
! 698: memcpy(Shared, s->addr, s->eom);
! 699:
! 700: for (i = 0, peer = strtok_r(Shared, SESS_MEM_DELIM"\r\n", &p_brk); peer;
! 701: peer = strtok_r(NULL, SESS_MEM_DELIM"\r\n", &p_brk)) {
! 702: if (!strchr(peer, '='))
! 703: continue;
! 704: else
! 705: i++;
! 706:
! 707: valz = e_realloc(valz, (i + 1) * sizeof(caddr_t));
! 708: if (!valz) {
! 709: LOGERR;
! 710: e_free(Shared);
! 711: return -1;
! 712: } else
! 713: valz[i] = NULL;
! 714:
! 715: valz[i - 1] = e_strdup(peer);
! 716: }
! 717:
! 718: e_free(Shared);
! 719: *ppsVals = valz;
! 720: return i;
! 721: }
! 722:
! 723: /*
! 724: * sess_GetValue() - Get value from session shared memory from attribute
! 725: *
! 726: * @s = Session item
! 727: * @csAttr = Attribute for search
! 728: * @psVal = Return string buffer
! 729: * @pnLen = Length of return string buffer,
! 730: // *{pnLen} input is max_size of buffer & output is really taken bytes
! 731: * return: 0 not found, -1 error: in parameter, >0 get position,
! 732: * if define item merged with IS_DEF
! 733: */
! 734: int
! 735: sess_GetValue(ait_sess_t * __restrict s, const char *csAttr, char *psVal, int *pnLen)
! 736: {
! 737: register int i;
! 738: int def = IS_VAL;
! 739: char *Shared = NULL;
! 740: char *peer, *p_brk, *a_brk, *attr, *val;
! 741:
! 742: if (!s || !csAttr || !*csAttr)
! 743: return -1;
! 744: if (psVal) {
! 745: if (pnLen && *pnLen > 0)
! 746: memset(psVal, 0, *pnLen);
! 747: else
! 748: return -1;
! 749: }
! 750:
! 751: /* allocated memory & mirrored shared memory into this */
! 752: Shared = e_malloc(s->eom);
! 753: if (!Shared) {
! 754: LOGERR;
! 755: return -1;
! 756: } else
! 757: memcpy(Shared, s->addr, s->eom);
! 758:
! 759: for (i = 1, peer = strtok_r(Shared, SESS_MEM_DELIM"\r\n", &p_brk); peer;
! 760: i++, peer = strtok_r(NULL, SESS_MEM_DELIM"\r\n", &p_brk)) {
! 761: attr = strtok_r(peer, "=\r\n", &a_brk);
! 762: if (attr && !strncmp(attr, csAttr, MAX_ATTRIBUTE - 1)) {
! 763: val = strtok_r(NULL, "=\r\n", &a_brk);
! 764: if (val && strlen(val)) {
! 765: if (psVal)
! 766: strlcpy(psVal, val, *pnLen);
! 767: if (pnLen)
! 768: *pnLen = strlen(val);
! 769: } else
! 770: def = IS_DEF;
! 771:
! 772: e_free(Shared);
! 773: return i | def;
! 774: }
! 775: }
! 776:
! 777: e_free(Shared);
! 778: return 0;
! 779: }
! 780:
! 781: /*
! 782: * sess_DelValue() - Delete item from session shared memory
! 783: *
! 784: * @s = Session item
! 785: * @csAttr = Attribute for erasing
! 786: * return: 0 Ok, -1 error: in parameter
! 787: */
! 788: int
! 789: sess_DelValue(ait_sess_t * __restrict s, const char *csAttr)
! 790: {
! 791: register int i;
! 792: int attrlen;
! 793: char *Buffer, *Shared, szAttr[MAX_ATTRIBUTE];
! 794: char *peer, *p_brk;
! 795:
! 796: if (!s || !csAttr || !*csAttr)
! 797: return -1;
! 798: else
! 799: attrlen = strlen(csAttr);
! 800: Buffer = Shared = NULL;
! 801: strlcpy(szAttr, csAttr, sizeof szAttr);
! 802: strlcat(szAttr, "=", sizeof szAttr);
! 803:
! 804: Buffer = e_malloc(s->eom);
! 805: if (!Buffer) {
! 806: LOGERR;
! 807: return -1;
! 808: } else
! 809: memset(Buffer, 0, s->eom);
! 810: Shared = e_malloc(s->eom);
! 811: if (!Shared) {
! 812: LOGERR;
! 813: e_free(Buffer);
! 814: return -1;
! 815: } else {
! 816: DEC_SEM(s);
! 817: memcpy(Shared, s->addr, s->eom);
! 818: }
! 819:
! 820: for (i = 1, peer = strtok_r(Shared, SESS_MEM_DELIM"\r\n", &p_brk); peer;
! 821: i++, peer = strtok_r(NULL, SESS_MEM_DELIM"\r\n", &p_brk)) {
! 822: if (!strncmp(peer, csAttr, attrlen))
! 823: if (peer[attrlen] == '=' || peer[attrlen] == *SESS_MEM_DELIM || !peer[attrlen] ||
! 824: peer[attrlen] == '\r' || peer[attrlen] == '\n')
! 825: continue;
! 826:
! 827: strlcat(Buffer, peer, s->eom);
! 828: strlcat(Buffer, SESS_MEM_DELIM, s->eom);
! 829: }
! 830:
! 831: memcpy(s->addr, Buffer, s->eom);
! 832:
! 833: if (s->type == SHARED_MAP)
! 834: msync(s->addr, 0, MS_SYNC | MS_INVALIDATE);
! 835:
! 836: INC_SEM(s);
! 837: e_free(Shared);
! 838: e_free(Buffer);
! 839: return 0;
! 840: }
! 841:
! 842: /*
! 843: * sess_SetValue() - Set item into session shared memory or update if exists
! 844: *
! 845: * @s = Session item
! 846: * @csAttr = Attribute
! 847: * @psVal = Value
! 848: * return: 0 nothing, -1 error: in parameter,
! 849: >0 set position, if add item merged with IS_ADD and if define item merged with IS_DEF
! 850: */
! 851: int
! 852: sess_SetValue(ait_sess_t * __restrict s, const char *csAttr, const char *psVal)
! 853: {
! 854: register int i;
! 855: int upd, def = IS_VAL;
! 856: char *Buffer, *Shared, szAttr[MAX_ATTRIBUTE];
! 857: char *peer, *p_brk;
! 858:
! 859: if (!s || !csAttr || !*csAttr)
! 860: return -1;
! 861: else
! 862: Buffer = Shared = NULL;
! 863: strlcpy(szAttr, csAttr, sizeof szAttr);
! 864: strlcat(szAttr, "=", sizeof szAttr);
! 865:
! 866: Buffer = e_malloc(s->eom);
! 867: if (!Buffer) {
! 868: LOGERR;
! 869: return -1;
! 870: } else
! 871: memset(Buffer, 0, s->eom);
! 872: Shared = e_malloc(s->eom);
! 873: if (!Shared) {
! 874: LOGERR;
! 875: e_free(Buffer);
! 876: return -1;
! 877: } else {
! 878: DEC_SEM(s);
! 879: memcpy(Shared, s->addr, s->eom);
! 880: }
! 881:
! 882: for (i = 1, upd = 0, peer = strtok_r(Shared, SESS_MEM_DELIM"\r\n", &p_brk); peer;
! 883: i++, peer = strtok_r(NULL, SESS_MEM_DELIM"\r\n", &p_brk)) {
! 884: if (!strncmp(peer, szAttr, strlen(szAttr))) {
! 885: upd++;
! 886: if (psVal) {
! 887: strlcat(Buffer, szAttr, s->eom);
! 888: strlcat(Buffer, psVal, s->eom);
! 889: strlcat(Buffer, SESS_MEM_DELIM, s->eom);
! 890: } else {
! 891: strlcat(Buffer, csAttr, s->eom);
! 892: strlcat(Buffer, SESS_MEM_DELIM, s->eom);
! 893: def = IS_DEF;
! 894: }
! 895: continue;
! 896: }
! 897:
! 898: strlcat(Buffer, peer, s->eom);
! 899: strlcat(Buffer, SESS_MEM_DELIM, s->eom);
! 900: }
! 901:
! 902: if (!upd) {
! 903: if (psVal) {
! 904: strlcat(Buffer, szAttr, s->eom);
! 905: strlcat(Buffer, psVal, s->eom);
! 906: strlcat(Buffer, SESS_MEM_DELIM, s->eom);
! 907: } else {
! 908: strlcat(Buffer, csAttr, s->eom);
! 909: strlcat(Buffer, SESS_MEM_DELIM, s->eom);
! 910: def = IS_DEF;
! 911: }
! 912: def |= IS_ADD;
! 913: }
! 914:
! 915: memcpy(s->addr, Buffer, s->eom);
! 916:
! 917: if (s->type == SHARED_MAP)
! 918: msync(s->addr, 0, MS_SYNC | MS_INVALIDATE);
! 919:
! 920: INC_SEM(s);
! 921: e_free(Shared);
! 922: e_free(Buffer);
! 923: return upd | def;
! 924: }
! 925:
! 926:
! 927: /*
! 928: * sess_prepareSession() - Attach to shared memory and de-marshaling data
! 929: *
! 930: * @s = Session
! 931: * @useDirect = Use direct shared memory if !=0 or snapshot of data to array
! 932: * return: NULL error or no data, !=NULL array with variables,
! 933: * after use must free resources with sess_doneSession()
! 934: */
! 935: array_t *
! 936: sess_prepareSession(ait_sess_t * __restrict s, char useDirect)
! 937: {
! 938: array_t *arr = NULL;
! 939: sess_hdr_t *hdr;
! 940:
! 941: assert(s);
! 942: if (!s) {
! 943: io_SetErr(EINVAL, "Invalid argument\n");
! 944: return NULL;
! 945: }
! 946: if (s->addr) {
! 947: io_SetErr(EINVAL, "Already attached memory\n");
! 948: return NULL;
! 949: }
! 950:
! 951: ATTACH_MEMORY(s);
! 952: if (!s->addr)
! 953: return NULL;
! 954: else
! 955: hdr = (sess_hdr_t*) s->addr;
! 956: if (hdr->hdr_magic != SESS_AIT_MAGIC) {
! 957: DETACH_MEMORY(s);
! 958:
! 959: io_SetErr(EINVAL, "Shared memory not contains values with proper format\n");
! 960: return NULL;
! 961: }
! 962:
! 963: DEC_SEM(s);
! 964: s->zcpy = useDirect;
! 965: arr = ait_map2vars(s->addr + sizeof(sess_hdr_t), s->eom - sizeof(sess_hdr_t),
! 966: hdr->hdr_argc, useDirect);
! 967: INC_SEM(s);
! 968:
! 969: if (!s->zcpy)
! 970: DETACH_MEMORY(s);
! 971: return arr;
! 972: }
! 973:
! 974: /*
! 975: * sess_doneSession() - Free resources allocated with sess_prepareSession()
! 976: *
! 977: * @s = Session
! 978: * @arr = Array with variables for free
! 979: * return: none
! 980: */
! 981: void
! 982: sess_doneSession(ait_sess_t * __restrict s, array_t ** __restrict arr)
! 983: {
! 984: assert(s);
! 985: if (!s) {
! 986: io_SetErr(EINVAL, "Invalid argument\n");
! 987: return;
! 988: }
! 989:
! 990: if (!s->zcpy)
! 991: array_Free(*arr);
! 992: else
! 993: DETACH_MEMORY(s);
! 994: array_Destroy(arr);
! 995: }
! 996:
! 997: /*
! 998: * sess_commitSession() - Commit data to shared memory
! 999: *
! 1000: * @s = Session
! 1001: * @arr = Array with variables for save
! 1002: * return -1 error or !=-1 size of stored variables into shared memory
! 1003: */
! 1004: int
! 1005: sess_commitSession(ait_sess_t * __restrict s, array_t * __restrict arr)
! 1006: {
! 1007: sess_hdr_t *hdr;
! 1008: int ret = 0;
! 1009:
! 1010: assert(s && arr);
! 1011: if (!s || !arr) {
! 1012: io_SetErr(EINVAL, "Invalid argument\n");
! 1013: return -1;
! 1014: }
! 1015:
! 1016: ATTACH_MEMORY(s);
! 1017: if (!s->addr) {
! 1018: DETACH_MEMORY(s);
! 1019: return -1;
! 1020: } else
! 1021: hdr = (sess_hdr_t*) s->addr;
! 1022:
! 1023: DEC_SEM(s);
! 1024: if ((ret = ait_vars2map(s->addr + sizeof(sess_hdr_t),
! 1025: s->eom - sizeof(sess_hdr_t), arr)) != -1) {
! 1026: hdr->hdr_magic = SESS_AIT_MAGIC;
! 1027: hdr->hdr_argc = array_Size(arr);
! 1028: ret += sizeof(sess_hdr_t);
! 1029: }
! 1030: INC_SEM(s);
! 1031:
! 1032: DETACH_MEMORY(s);
! 1033: return ret;
! 1034: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>