Annotation of libaitrpc/src/blob.c, revision 1.3
1.2 misho 1: /*************************************************************************
2: * (C) 2010 AITNET ltd - Sofia/Bulgaria - <misho@aitbg.com>
3: * by Michael Pounov <misho@openbsd-bg.org>
4: *
5: * $Author: misho $
1.3 ! misho 6: * $Id: blob.c,v 1.2.4.5 2011/08/19 14:10:53 misho Exp $
1.2 misho 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
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: * rpc_srv_blobCreate() Create map blob to memory region and return object
51: * @srv = RPC Server instance
52: * @len = BLOB length object
53: * return: NULL error or !=NULL allocated BLOB object
54: */
55: inline rpc_blob_t *
56: rpc_srv_blobCreate(rpc_srv_t * __restrict srv, int len)
57: {
58: rpc_blob_t *blob = NULL;
59: char szFName[MAXPATHLEN];
60: int f;
61: u_int rnd;
62:
63: #ifdef HAVE_SRANDOMDEV
64: srandomdev();
65: #else
66: time_t tim;
67:
68: srandom((time(&tim) ^ getpid()));
69: #endif
70: again:
71: rnd = random() % UINT_MAX;
72:
1.3 ! misho 73: memset(szFName, 0, sizeof szFName);
! 74: snprintf(szFName, sizeof szFName, BLOB_FILE, srv->srv_blob.dir, rnd);
1.2 misho 75: f = open(szFName, O_CREAT | O_EXCL | O_RDWR, 0600);
76: if (f == -1) {
77: if (errno == EEXIST)
78: goto again;
79:
80: LOGERR;
81: return NULL;
82: }
83: if (lseek(f, len - 1, SEEK_SET) == -1) {
84: LOGERR;
85: close(f);
86: unlink(szFName);
87: return NULL;
88: } else
89: write(f, "", 1);
90:
91: blob = malloc(sizeof(rpc_blob_t));
92: if (!blob) {
93: LOGERR;
94: close(f);
95: unlink(szFName);
96: return NULL;
97: }
98:
99: blob->blob_data = mmap(NULL, len, PROT_READ | PROT_WRITE, MAP_SHARED, f, 0);
100: if (blob->blob_data == MAP_FAILED) {
101: LOGERR;
102: free(blob);
103: close(f);
104: unlink(szFName);
105: return NULL;
106: } else
107: close(f);
108:
109: blob->blob_len = len;
110: blob->blob_var = rnd;
111: return blob;
112: }
113:
114: /*
115: * rpc_srv_blobMap() Map blob to memory region
116: * @srv = RPC Server instance
117: * @blob = Map to this BLOB element
118: * return: -1 error or 0 ok
119: */
120: inline int
121: rpc_srv_blobMap(rpc_srv_t * __restrict srv, rpc_blob_t * __restrict blob)
122: {
123: int f;
124: char szFName[MAXPATHLEN];
125:
126: if (!blob) {
127: rpc_SetErr(EINVAL, "Error:: invalid arguments ...\n");
128: return -1;
129: }
130:
1.3 ! misho 131: memset(szFName, 0, sizeof szFName);
! 132: snprintf(szFName, sizeof szFName, BLOB_FILE, srv->srv_blob.dir, blob->blob_var);
1.2 misho 133: f = open(szFName, O_RDWR);
134: if (f == -1) {
135: LOGERR;
136: return -1;
137: }
138:
139: blob->blob_data = mmap(NULL, blob->blob_len, PROT_READ | PROT_WRITE, MAP_SHARED, f, 0);
140: if (blob->blob_data == MAP_FAILED) {
141: LOGERR;
142: close(f);
143: blob->blob_data = NULL;
144: return -1;
145: } else {
146: close(f);
147:
148: madvise(blob->blob_data, blob->blob_len, MADV_SEQUENTIAL);
149: }
150:
151: return 0;
152: }
153:
154: /*
155: * rpc_srv_blobUnmap() Unmap blob memory region
156: * @blob = Mapped BLOB element
157: * return: none
158: */
159: inline void
160: rpc_srv_blobUnmap(rpc_blob_t * __restrict blob)
161: {
162: if (!blob || !blob->blob_data)
163: rpc_SetErr(EINVAL, "Error:: invalid arguments ...\n");
164: else {
165: munmap(blob->blob_data, blob->blob_len);
166: blob->blob_data = NULL;
167: }
168: }
169:
170: /*
171: * rpc_srv_blobFree() Free blob from disk & memory
172: * @srv = RPC Server instance
173: * @blob = Mapped BLOB element
174: * return: -1 error or 0 ok
175: */
176: inline int
177: rpc_srv_blobFree(rpc_srv_t * __restrict srv, rpc_blob_t * __restrict blob)
178: {
179: char szFName[MAXPATHLEN];
180:
181: if (!blob) {
182: rpc_SetErr(EINVAL, "Error:: invalid arguments ...\n");
183: return -1;
184: }
185:
186: if (blob->blob_data)
187: rpc_srv_blobUnmap(blob);
188:
1.3 ! misho 189: memset(szFName, 0, sizeof szFName);
! 190: snprintf(szFName, sizeof szFName, BLOB_FILE, srv->srv_blob.dir, blob->blob_var);
! 191: if (unlink(szFName) == -1) {
1.2 misho 192: LOGERR;
193: return -1;
194: }
195:
196: return 0;
197: }
198:
199: // ------------------------------------------------------------
200:
201: /*
202: * rpc_srv_sendBLOB() Send mapped BLOB to client
203: * @cli = Client instance
204: * @blob = Mapped BLOB element
205: * return: -1 error, 0 ok
206: */
207: int
208: rpc_srv_sendBLOB(rpc_cli_t * __restrict cli, rpc_blob_t * __restrict blob)
209: {
210: int ret, len;
211: uint8_t *pos;
212:
213: if (!cli || !blob || !blob->blob_data) {
214: rpc_SetErr(EINVAL, "Error:: invalid arguments ...\n");
215: return -1;
216: }
217:
1.3 ! misho 218: for (ret = blob->blob_len, pos = blob->blob_data; ret > 0; ret -= len, pos += len) {
! 219: len = send(cli->cli_sock, pos, ret, 0);
! 220: if (len == -1) {
1.2 misho 221: LOGERR;
222: return -1;
223: }
1.3 ! misho 224: }
1.2 misho 225:
226: return ret;
227: }
228:
229: /*
230: * rpc_srv_recvBLOB() Receive BLOB from client
231: * @cli = Client instance
232: * @blob = Mapped BLOB element
233: * return: -1 error, 0 ok, >0 unreceived data from client, may be error?
234: */
235: int
236: rpc_srv_recvBLOB(rpc_cli_t * __restrict cli, rpc_blob_t * __restrict blob)
237: {
238: int ret, len;
239: uint8_t *pos;
240: fd_set fds;
241: struct timeval tv = { DEF_RPC_TIMEOUT, 0 };
242:
243: if (!cli || !blob || !blob->blob_data) {
244: rpc_SetErr(EINVAL, "Error:: invalid arguments ...\n");
245: return -1;
246: }
247:
248: for (ret = blob->blob_len, pos = blob->blob_data; ret > 0; ret -= len, pos += len) {
249: FD_ZERO(&fds);
250: FD_SET(cli->cli_sock, &fds);
251: len = select(cli->cli_sock + 1, &fds, NULL, NULL, &tv);
252: if (len < 1) {
253: LOGERR;
254: return -1;
255: }
256:
1.3 ! misho 257: len = recv(cli->cli_sock, pos, ret, 0);
! 258: if (len == -1) {
1.2 misho 259: LOGERR;
260: return -1;
261: }
262: }
263:
264: return ret;
265: }
266:
267: // ------------------------------------------------------------
268:
269: /*
270: * rpc_cli_sendBLOB() Send BLOB to server
271: * @cli = Client instance
272: * @var = BLOB variable
273: * @data = BLOB data
274: * return: -1 error, 0 ok, 1 remote error
275: */
276: int
277: rpc_cli_sendBLOB(rpc_cli_t * __restrict cli, rpc_val_t * __restrict var, void * __restrict data)
278: {
279: int ret, len;
280: uint8_t *pos;
281: struct tagBLOBHdr hdr;
282: fd_set fds;
283: struct timeval tv = { DEF_RPC_TIMEOUT, 0 };
284:
285: if (!cli || !var || !data) {
286: rpc_SetErr(EINVAL, "Error:: invalid arguments ...\n");
287: return -1;
288: }
289:
290: memcpy(&hdr.hdr_session, cli->cli_parent, sizeof(rpc_sess_t));
291: hdr.hdr_cmd = set;
292: hdr.hdr_var = 0;
293: hdr.hdr_ret = 0;
294: hdr.hdr_len = var->val_len;
295: if (send(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {
296: LOGERR;
297: return -1;
298: }
299:
300: for (ret = var->val_len, pos = data; ret > 0; ret -= len, pos += len)
1.3 ! misho 301: if ((len = send(cli->cli_sock, pos, ret, 0)) == -1) {
1.2 misho 302: LOGERR;
303: return -1;
304: }
305:
306: FD_ZERO(&fds);
307: FD_SET(cli->cli_sock, &fds);
308: switch (select(cli->cli_sock + 1, &fds, NULL, NULL, &tv)) {
309: case -1:
310: LOGERR;
311: return -1;
312: case 0:
313: rpc_SetErr(ETIMEDOUT, "Error:: Timeout reached! Server not responde ...\n");
314: return -1;
315: }
316: if (recv(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {
317: LOGERR;
318: return -1;
319: }
320: if (hdr.hdr_cmd != error) {
321: if (hdr.hdr_len != var->val_len) {
322: rpc_SetErr(ECANCELED, "Error:: Bad return length packet ...\n");
323: return -1;
324: }
325:
326: var->val.blob = hdr.hdr_var;
327: }
328:
329: return hdr.hdr_cmd == error;
330: }
331:
332: /*
333: * rpc_cli_recvBLOB() Receive BLOB from server
334: * @cli = Client instance
335: * @var = BLOB variable
336: * @data = BLOB data, must be free after use!
337: * return: -1 error, 0 ok, 1 remote error
338: */
339: int
340: rpc_cli_recvBLOB(rpc_cli_t * __restrict cli, rpc_val_t * __restrict var, void ** data)
341: {
342: int ret, len;
343: uint8_t *pos;
344: fd_set fds;
345: struct timeval tv = { DEF_RPC_TIMEOUT, 0 };
346: struct tagBLOBHdr hdr;
347:
348: if (!cli || !var || !data) {
349: rpc_SetErr(EINVAL, "Error:: invalid arguments ...\n");
350: return -1;
351: }
352:
353: *data = malloc(var->val_len);
354: if (!*data) {
355: LOGERR;
356: return -1;
357: } else
358: memset(*data, 0, var->val_len);
359:
360: memcpy(&hdr.hdr_session, cli->cli_parent, sizeof(rpc_sess_t));
361: hdr.hdr_cmd = get;
362: hdr.hdr_var = (uint32_t) RPC_GET_BLOB(var);
363: hdr.hdr_ret = 0;
364: hdr.hdr_len = 0;
365: if (send(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {
366: LOGERR;
367: free(*data);
368: *data = NULL;
369: return -1;
370: }
371:
372: for (ret = var->val_len, pos = *data; ret > 0; ret -= len, pos += len) {
373: FD_ZERO(&fds);
374: FD_SET(cli->cli_sock, &fds);
375: len = select(cli->cli_sock + 1, &fds, NULL, NULL, &tv);
376: if (len < 1) {
377: LOGERR;
378: free(*data);
379: *data = NULL;
380: return -1;
381: }
382:
1.3 ! misho 383: if ((len = recv(cli->cli_sock, pos, ret, 0)) == -1) {
1.2 misho 384: LOGERR;
385: free(*data);
386: *data = NULL;
387: return -1;
388: }
389: }
390:
391: FD_ZERO(&fds);
392: FD_SET(cli->cli_sock, &fds);
393: switch (select(cli->cli_sock + 1, &fds, NULL, NULL, &tv)) {
394: case -1:
395: LOGERR;
396: free(*data);
397: *data = NULL;
398: return -1;
399: case 0:
400: rpc_SetErr(ETIMEDOUT, "Error:: Timeout reached! Server not responde ...\n");
401: free(*data);
402: *data = NULL;
403: return -1;
404: }
405: if (recv(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {
406: LOGERR;
407: free(*data);
408: *data = NULL;
409: return -1;
410: }
411: if (hdr.hdr_cmd != error) {
412: if (hdr.hdr_len != var->val_len) {
413: rpc_SetErr(ECANCELED, "Error:: Bad return length packet ...\n");
414: free(*data);
415: *data = NULL;
416: return -1;
417: }
418: }
419:
420: return hdr.hdr_cmd == error;
421: }
422:
423: /*
424: * rpc_cli_delBLOB() Delete BLOB from server
425: * @cli = Client instance
426: * @var = BLOB variable
427: * return: -1 error, 0 ok, 1 remote error
428: */
429: int
430: rpc_cli_delBLOB(rpc_cli_t * __restrict cli, rpc_val_t * __restrict var)
431: {
432: struct tagBLOBHdr hdr;
433: fd_set fds;
434: struct timeval tv = { DEF_RPC_TIMEOUT, 0 };
435:
436: if (!cli || !var) {
437: rpc_SetErr(EINVAL, "Error:: invalid arguments ...\n");
438: return -1;
439: }
440:
441: memcpy(&hdr.hdr_session, cli->cli_parent, sizeof(rpc_sess_t));
442: hdr.hdr_cmd = unset;
443: hdr.hdr_var = (uint32_t) RPC_GET_BLOB(var);
444: hdr.hdr_ret = 0;
445: hdr.hdr_len = 0;
446: if (send(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {
447: LOGERR;
448: return -1;
449: }
450:
451: FD_ZERO(&fds);
452: FD_SET(cli->cli_sock, &fds);
453: switch (select(cli->cli_sock + 1, &fds, NULL, NULL, &tv)) {
454: case -1:
455: LOGERR;
456: return -1;
457: case 0:
458: rpc_SetErr(ETIMEDOUT, "Error:: Timeout reached! Server not responde ...\n");
459: return -1;
460: }
461: if (recv(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {
462: LOGERR;
463: return -1;
464: }
465:
466: return hdr.hdr_cmd == error;
467: }
468:
469: /*
470: * rpc_cli_getBLOB() Receive BLOB from server and Delete after that
471: * @cli = Client instance
472: * @var = BLOB variable
473: * @data = BLOB data, must be free after use!
474: * return: -1 error, 0 ok, 1 remote error
475: */
476: inline int
477: rpc_cli_getBLOB(rpc_cli_t * __restrict cli, rpc_val_t * __restrict var, void ** data)
478: {
479: int ret;
480:
481: ret = rpc_cli_recvBLOB(cli, var, data);
482: ret |= rpc_cli_delBLOB(cli, var) > 0 ? 2 : 0;
483:
484: return ret;
485: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>
![[BACK]](/icons/cvsweb/back.gif){kind=icon}
Return to blob.c CVS log ![[TXT]](/icons/cvsweb/text.gif){kind=icon}
![[DIR]](/icons/cvsweb/dir.gif){kind=icon}
Up to [ELWIX - Embedded LightWeight unIX -] / libaitrpc / src