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: aitsess.h,v 1.4.2.6 2012/02/28 10:34:43 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, 2012
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: #ifndef __AITSESS_H
47: #define __AITSESS_H
48:
49:
50: #include <pthread.h>
51: #include <assert.h>
52: #include <semaphore.h>
53: #include <aitio.h>
54:
55: #define SHARED_UNKNOWN -1
56: #define SHARED_IPC 0
57: #define SHARED_MAP 1
58:
59: #define IS_VAL 0x0
60: #define IS_ADD 0x40000000
61: #define IS_DEF 0x80000000
62:
63: #define ISNEW(x) (((x) & IS_ADD) == IS_ADD)
64: #define ISDEF(x) (((x) & IS_DEF) == IS_DEF)
65:
66: #define MAX_ATTRIBUTE 64
67: #define MAX_SEMNAME 14
68:
69: /* Memory pool */
70:
71: #define MEM_BUCKETS 28 /* 32 bits - 4 bits = 28 items in bucket array */
72:
73: struct tagAlloc {
74: char alloc_name[MAX_ATTRIBUTE];
75: volatile unsigned int alloc_flags;
76:
77: unsigned int *alloc_mem;
78:
79: TAILQ_ENTRY(tagAlloc) alloc_node;
80: };
81: typedef TAILQ_HEAD(, tagAlloc) mpool_bucket_t;
82:
83: typedef struct _tagMPool {
84: pthread_mutex_t pool_mtx;
85:
86: struct {
87: unsigned long alloc;
88: unsigned long free;
89: unsigned long cache;
90: } pool_calls;
91: struct {
92: unsigned long alloc;
93: unsigned long free;
94: unsigned long cache;
95: } pool_bytes;
96: struct {
97: unsigned long max;
98: unsigned long curr;
99: } pool_quota;
100:
101: /* pool buckets */
102: mpool_bucket_t pool_active[MEM_BUCKETS];
103: mpool_bucket_t pool_inactive[MEM_BUCKETS];
104: } mpool_t;
105: #define mpool_lock(x) (assert((x)), pthread_mutex_lock(&(x)->pool_mtx))
106: #define mpool_unlock(x) (assert((x)), pthread_mutex_unlock(&(x)->pool_mtx))
107:
108: typedef void (*mpool_stat_cb)(unsigned int, unsigned int, unsigned int);
109:
110:
111: /* Shared memory session */
112:
113: typedef struct tagSess {
114: key_t key;
115: char type;
116: char zcpy;
117:
118: char name[BUFSIZ];
119: off_t eom;
120: void *addr;
121: off_t offset;
122: union {
123: int shmid;
124: int fd;
125: } mem;
126: union {
127: int semid;
128: sem_t *sid;
129: } id;
130:
131: struct {
132: int (*create)(int, long, void *, ...);
133: void (*destroy)(void *);
134: void *(*attach)(void *, void *);
135: void (*detach)(void *);
136: void (*notSem)(void *);
137: int (*isSemOK)(void *);
138: int (*incSem)(void *);
139: int (*decSem)(void *);
140: } sess;
141: } ait_sess_t;
142:
143:
144: // -------------------------------------------------------
145: // sess_GetErrno() Get error code of last operation
146: inline int sess_GetErrno();
147: // sess_GetError() Get error text of last operation
148: inline const char *sess_GetError();
149: // sess_SetErr() Set error to variables for internal use!!!
150: inline void sess_SetErr(int eno, char *estr, ...);
151: // -------------------------------------------------------
152:
153:
154: /*
155: * mpool_init() - Init memory pool
156: *
157: * @maxmem = If !=0 set maximum memory quota
158: * return: =NULL error or !=NULL new allocated pool
159: */
160: mpool_t *mpool_init(unsigned long maxmem);
161: /*
162: * mpool_destroy() - Destroy memory pool
163: *
164: * @mp = Memory pool
165: * return: none
166: */
167: void mpool_destroy(mpool_t ** __restrict mp);
168: /*
169: * mpool_purge() - Purge memory block cache and release resources
170: *
171: * @mp = Memory pool
172: * @atmost = Free at most in buckets
173: * return: -1 error or 0 ok
174: */
175: int mpool_purge(mpool_t * __restrict mp, unsigned int atmost);
176: /*
177: * mpool_malloc() - Memory allocation
178: *
179: * @mp = Memory pool
180: * @size = Size
181: * @memname = Optional memory block name
182: * return: NULL error or !=NULL ok allocated memory
183: */
184: void *mpool_malloc(mpool_t * __restrict mp, unsigned int size, const char *memname);
185: /*
186: * mpool_free() Free allocated memory with mpool_alloc()
187: *
188: * @mp = Memory pool
189: * @data = Allocated memory data
190: * @purge = if !=0 force release memory block
191: * return: <0 error or 0 ok released memory block
192: */
193: int mpool_free(mpool_t * __restrict mp, void * __restrict data, int purge);
194: /*
195: * mpool_free2() Free allocated memory with mpool_alloc() by size and memory name
196: *
197: * @mp = Memory pool
198: * @size = Allocated memory data size
199: * @memname = Memory name
200: * @purge = if !=0 force release memory block
201: * return: <0 error or 0 ok released memory block
202: */
203: int mpool_free2(mpool_t * __restrict mp, unsigned int size, const char *memname, int purge);
204: /*
205: * mpool_realloc() Reallocate memory block with new size
206: *
207: * @mp = Memory pool
208: * @data = Allocated memory data
209: * @newsize = New size of memory block
210: * @memname = Optional new memory block name
211: * return: NULL error or !=NULL new reallocated memory block
212: */
213: void *mpool_realloc(mpool_t * __restrict mp, void * __restrict data,
214: unsigned int newsize, const char *memname);
215: /*
216: * mpool_getmembynam() Find allocated memory block by size and memory name
217: *
218: * @mp = Memory pool
219: * @size = Memory size
220: * @memname = Memory name
221: * return: NULL error or not found and !=NULL allocated memory
222: */
223: inline struct tagAlloc *mpool_getmembynam(mpool_t * __restrict mp, unsigned int size, const char *memname);
224: /*
225: * mpool_getsizebyaddr() - Get size of allocated memory block by address
226: *
227: * @data = allocated memory from mpool_malloc()
228: * return: usable size of allocated memory block
229: */
230: inline unsigned int mpool_getsizebyaddr(void * __restrict data);
231: /*
232: * mpool_chkaddr() - Check validity of given address
233: *
234: * @data = allocated memory from mpool_malloc()
235: * return: -1 bad address, 1 corrupted address or 0 ok
236: */
237: inline int mpool_chkaddr(void * __restrict data);
238: /*
239: * mpool_setquota() - Change maximum memory quota
240: *
241: * @mp = Memory pool
242: * @maxmem = New max quota size
243: * return: old maximum memory quota size
244: */
245: inline unsigned long mpool_setquota(mpool_t * __restrict mp, unsigned long maxmem);
246: /*
247: * mpool_getquota() - Get memory quota
248: *
249: * @mp = Memory pool
250: * @currmem = Return current memory
251: * @maxmem = Return max quota size
252: * return: none
253: */
254: inline void mpool_getquota(mpool_t * __restrict mp, unsigned long *currmem,
255: unsigned long *maxmem);
256: /*
257: * mpool_statistics() - Dump statistics from memory pool buckets
258: *
259: * @mp = Memory pool
260: * @cb = Export statistics to callback
261: * return: none
262: */
263: void mpool_statistics(mpool_t * __restrict mp, mpool_stat_cb cb);
264:
265:
266: /*
267: * sess_initSession() Initializing session structure, if session file not exists creating with specified tech
268: *
269: * @id = Technology using in session. SHARED_IPC IPC tech orSHARED_MAP BSD MemoryMap tech
270: * @csFName = Session filename for build key and identified
271: * @Sess = Session item, if =NULL allocate memory for session after use must be free!
272: * return: 0 OK new key created, -1 error: no memory or file not created, 1 OK key finded
273: */
274: int sess_initSession(int id, const char *csFName, ait_sess_t ** __restrict Sess);
275: /*
276: * sess_freeSession() Free allocated memory for session item and delete session file if present name
277: *
278: * @Sess = Session item
279: * return: none
280: */
281: void sess_freeSession(ait_sess_t ** __restrict Sess);
282:
283:
284: /*
285: * map_createSession() MMAP Created session and allocated resources
286: *
287: * @nSeed = Seed for securing key, if =-1 must add ready for use key
288: * @nSize = Allocated shared memory size in bytes
289: * @Sess = Session item
290: * @... = If nSeed == -1 add ready for use key value
291: * return: 0 Ok successful, -1 error: not allocated resources
292: */
293: int map_createSession(int nSeed, long nSize, ait_sess_t * __restrict Sess, ...);
294: /*
295: * map_destroySession() MMAP free shared resources
296: *
297: * @Sess = Session item
298: * return: none
299: */
300: void map_destroySession(ait_sess_t * __restrict Sess);
301:
302: /*
303: * ipc_createSession() IPC Created session and allocated resources
304: *
305: * @nSeed = Seed for securing key, if =-1 must add ready for use key
306: * @nSize = Allocated shared memory size in bytes
307: * @Sess = Session item
308: * @... = If nSeed == -1 add ready for use key value
309: * return: 0 Ok successful, -1 error: not allocated resources
310: */
311: int ipc_createSession(int nSeed, long nSize, ait_sess_t * __restrict Sess, ...);
312: /*
313: * ipc_destroySession() IPC free shared resources
314: *
315: * @Sess = Session item
316: * return: none
317: */
318: void ipc_destroySession(ait_sess_t * __restrict Sess);
319:
320: /*
321: * map_attachSession() MMAP Attach to shared memory & return begin address
322: *
323: * @s = Session item
324: * @procMem = Custom start address (optionl) *default must be 0*
325: * return: NULL failed attach, !=NULL begin address of memory
326: */
327: void *map_attachSession(ait_sess_t * __restrict s, void *procMem);
328: /*
329: * map_detachSession() MMAP Detach from shared memory
330: *
331: * @s = Session item
332: * return: none
333: */
334: void map_detachSession(ait_sess_t * __restrict s);
335:
336: /*
337: * ipc_attachSession() IPC Attach to shared memory & return begin address
338: *
339: * @s = Session item
340: * @procMem = Custom start address (optionl) *default must be 0*
341: * return: NULL failed attach, !=NULL begin address of memory
342: */
343: void *ipc_attachSession(ait_sess_t * __restrict s, void *procMem);
344: /*
345: * ipc_detachSession() IPC Detach from shared memory
346: *
347: * @s = Session item
348: * return: none
349: */
350: void ipc_detachSession(ait_sess_t * __restrict s);
351:
352: /*
353: * sess_isAttached() Check for attached shared memory
354: *
355: * @s = Session item
356: * return: -1 null session item, 0 not attached, 1 attached memory
357: */
358: inline int sess_isAttached(ait_sess_t * __restrict s);
359:
360:
361: /*
362: * map_notSemaphore() MMAP negative block if semaphore isn`t signaled
363: *
364: * @s = Session item
365: * return: none
366: */
367: void map_notSemaphore(ait_sess_t * __restrict s);
368: /*
369: * map_isSemaphoreOK() MMAP Check semaphore
370: *
371: * @s = Session item
372: * return: -1 error: can`t return semaphore, 0 = false, 1 = true
373: */
374: int map_isSemaphoreOK(ait_sess_t * __restrict s);
375: /*
376: * map_incSemaphore() MMAP unblock semaphore, increment semaphore
377: *
378: * @s = Session item
379: * return: 0 Ok, -1 error: can`t increment
380: */
381: int map_incSemaphore(ait_sess_t * __restrict s);
382: /*
383: * map_decSemaphore() MMAP block semaphore, decrement semaphore
384: *
385: * @s = Session item
386: * return: 0 Ok, -1 error: can`t decrement
387: */
388: int map_decSemaphore(ait_sess_t * __restrict s);
389:
390: /*
391: * ipc_notSemaphore() IPC negative block if semaphore isn`t signaled
392: *
393: * @s = Session item
394: * return: none
395: */
396: void ipc_notSemaphore(ait_sess_t * __restrict s);
397: /*
398: * ipc_isSemaphoreOK() IPC Check semaphore
399: *
400: * @s = Session item
401: * return: -1 error: can`t return semaphore, 0 = false, 1 = true
402: */
403: int ipc_isSemaphoreOK(ait_sess_t * __restrict s);
404: /*
405: * ipc_incSemaphore() IPC unblock semaphore, increment semaphore
406: *
407: * @s = Session item
408: * return: 0 Ok, -1 error: can`t increment
409: */
410: int ipc_incSemaphore(ait_sess_t * __restrict s);
411: /*
412: * ipc_decSemaphore() IPC block semaphore, decrement semaphore
413: *
414: * @s = Session item
415: * return: 0 Ok, -1 error: can`t decrement
416: */
417: int ipc_decSemaphore(ait_sess_t * __restrict s);
418:
419:
420: /* --------------------------------------------------------- */
421:
422: #define ALLOC_MEMORY(sd, siz, s, ...) (assert((s)), (s)->sess.create((sd), \
423: (siz), (s), ## __VA_ARGS__))
424: #define ATTACH_MEMORY(s) (assert((s)), (s)->sess.attach((s), NULL))
425: #define DETACH_MEMORY(s) do { assert((s)); (s)->sess.detach((s)); } while(0)
426: #define FREE_MEMORY(s) do { assert((s)); (s)->sess.destroy((s)); } while(0)
427:
428: #define IS_SEMOK(s) (assert((s)), (s)->sess.isSemOK((s)))
429: #define INC_SEM(s) (assert((s)), (s)->sess.incSem((s)))
430: #define DEC_SEM(s) (assert((s)), (s)->sess.decSem((s)))
431: #define NOT_SEM(s) do { assert((s)); (s)->sess.notSem((s)); } while(0)
432:
433: /* --------------------------------------------------------- */
434:
435: /*
436: * sess_FreeValues() Free all values from value array allocated from sess_GetValues()
437: *
438: * @ppsVals = Array strings
439: * return: none
440: */
441: inline void sess_FreeValues(char *** __restrict ppsVals);
442: /*
443: * sess_GetValues() Get all values from session shared memory
444: *
445: * @s = Session item
446: * @ppsVals = Return array strings
447: * return: -1 error: in parameter, !=-1 count of returned strings in ppsVals (must be free after use!)
448: */
449: int sess_GetValues(ait_sess_t * __restrict s, char ***ppsVals);
450: /*
451: * sess_GetValue() Get value from session shared memory from attribute
452: *
453: * @s = Session item
454: * @csAttr = Attribute for search
455: * @psVal = Return string buffer
456: * @pnLen = Length of return string buffer,
457: // *{pnLen} input is max_size of buffer & output is really taken bytes
458: * return: 0 not found, -1 error: in parameter, >0 get position, if define item merged with IS_DEF
459: */
460: int sess_GetValue(ait_sess_t * __restrict s, const char *csAttr, char *psVal, int *pnLen);
461: /*
462: * sess_DelValue() Delete item from session shared memory
463: *
464: * @s = Session item
465: * @csAttr = Attribute for erasing
466: * return: 0 Ok, -1 error: in parameter
467: */
468: int sess_DelValue(ait_sess_t * __restrict s, const char *csAttr);
469: /*
470: * sess_SetValue() Set item into session shared memory or update if find it
471: *
472: * @s = Session item
473: * @csAttr = Attribute
474: * @psVal = Value
475: * return: 0 nothing, -1 error: in parameter,
476: >0 set position, if add item merged with IS_ADD and if define item merged with IS_DEF
477: */
478: int sess_SetValue(ait_sess_t * __restrict s, const char *csAttr, const char *psVal);
479:
480: /*
481: * sess_prepareSession() Attach to shared memory and de-marshaling data
482: *
483: * @s = Session
484: * @useDirect = Use direct shared memory if !=0 or snapshot of data to array
485: * return: NULL error or no data, !=NULL array with variables,
486: * after use must free resources with sess_doneSession()
487: */
488: array_t *sess_prepareSession(ait_sess_t * __restrict s, char useDirect);
489: /*
490: * sess_doneSession() Free resources allocated with sess_prepareSession()
491: *
492: * @s = Session
493: * @arr = Array with variables for free
494: * return: none
495: */
496: void sess_doneSession(ait_sess_t * __restrict s, array_t ** __restrict arr);
497: /*
498: * sess_commitSession() Commit data to shared memory
499: *
500: * @s = Session
501: * @arr = Array with variables for save
502: * return -1 error or !=-1 size of stored variables into shared memory
503: */
504: int sess_commitSession(ait_sess_t * __restrict s, array_t * __restrict arr);
505:
506:
507: #endif
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>