Annotation of libaitrpc/src/blob.c, revision 1.7.2.4
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.7.2.4 ! misho 6: * $Id: blob.c,v 1.7.2.3 2012/05/16 08:20:26 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:
1.5 misho 15: Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012
1.2 misho 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: /*
1.7.2.1 misho 50: * rpc_srv_blobCreate() - Create and map blob to memory region and return object
1.5 misho 51: *
1.2 misho 52: * @srv = RPC Server instance
53: * @len = BLOB length object
54: * return: NULL error or !=NULL allocated BLOB object
55: */
56: inline rpc_blob_t *
57: rpc_srv_blobCreate(rpc_srv_t * __restrict srv, int len)
58: {
59: rpc_blob_t *blob = NULL;
60: char szFName[MAXPATHLEN];
61: int f;
62: u_int rnd;
63:
64: again:
65: rnd = random() % UINT_MAX;
66:
1.3 misho 67: memset(szFName, 0, sizeof szFName);
1.5 misho 68: snprintf(szFName, sizeof szFName, BLOB_FILE, AIT_GET_STR(&srv->srv_blob.dir), rnd);
1.2 misho 69: f = open(szFName, O_CREAT | O_EXCL | O_RDWR, 0600);
70: if (f == -1) {
71: if (errno == EEXIST)
72: goto again;
73:
74: LOGERR;
75: return NULL;
76: }
1.7.2.1 misho 77: if (ftruncate(f, len) == -1) {
1.2 misho 78: LOGERR;
79: close(f);
80: unlink(szFName);
81: return NULL;
1.7.2.1 misho 82: }
1.2 misho 83:
84: blob = malloc(sizeof(rpc_blob_t));
85: if (!blob) {
86: LOGERR;
87: close(f);
88: unlink(szFName);
89: return NULL;
90: }
91:
92: blob->blob_data = mmap(NULL, len, PROT_READ | PROT_WRITE, MAP_SHARED, f, 0);
93: if (blob->blob_data == MAP_FAILED) {
94: LOGERR;
95: free(blob);
96: close(f);
97: unlink(szFName);
98: return NULL;
99: } else
100: close(f);
101:
102: blob->blob_len = len;
103: blob->blob_var = rnd;
104: return blob;
105: }
106:
107: /*
1.5 misho 108: * rpc_srv_blobMap() - Map blob to memory region
109: *
1.2 misho 110: * @srv = RPC Server instance
111: * @blob = Map to this BLOB element
112: * return: -1 error or 0 ok
113: */
114: inline int
115: rpc_srv_blobMap(rpc_srv_t * __restrict srv, rpc_blob_t * __restrict blob)
116: {
117: int f;
118: char szFName[MAXPATHLEN];
119:
120: if (!blob) {
1.5 misho 121: rpc_SetErr(EINVAL, "Invalid argument BLOB");
1.2 misho 122: return -1;
123: }
1.4 misho 124: if (blob->blob_data) {
1.5 misho 125: rpc_SetErr(EPERM, "Already mmapped object found!");
1.4 misho 126: return -1;
127: }
1.2 misho 128:
1.3 misho 129: memset(szFName, 0, sizeof szFName);
1.5 misho 130: snprintf(szFName, sizeof szFName, BLOB_FILE, AIT_GET_STR(&srv->srv_blob.dir), blob->blob_var);
1.2 misho 131: f = open(szFName, O_RDWR);
132: if (f == -1) {
133: LOGERR;
134: return -1;
135: }
136:
137: blob->blob_data = mmap(NULL, blob->blob_len, PROT_READ | PROT_WRITE, MAP_SHARED, f, 0);
138: if (blob->blob_data == MAP_FAILED) {
139: LOGERR;
140: close(f);
141: blob->blob_data = NULL;
142: return -1;
143: } else {
144: close(f);
145:
146: madvise(blob->blob_data, blob->blob_len, MADV_SEQUENTIAL);
147: }
148:
149: return 0;
150: }
151:
152: /*
1.5 misho 153: * rpc_srv_blobUnmap() - Unmap blob memory region
154: *
1.2 misho 155: * @blob = Mapped BLOB element
156: * return: none
157: */
158: inline void
159: rpc_srv_blobUnmap(rpc_blob_t * __restrict blob)
160: {
1.7.2.3 misho 161: if (blob && blob->blob_data) {
1.2 misho 162: munmap(blob->blob_data, blob->blob_len);
163: blob->blob_data = NULL;
164: }
165: }
166:
167: /*
1.5 misho 168: * rpc_srv_blobFree() - Free blob from disk & memory
169: *
1.2 misho 170: * @srv = RPC Server instance
171: * @blob = Mapped BLOB element
172: * return: -1 error or 0 ok
173: */
174: inline int
175: rpc_srv_blobFree(rpc_srv_t * __restrict srv, rpc_blob_t * __restrict blob)
176: {
177: char szFName[MAXPATHLEN];
178:
179: if (!blob) {
1.5 misho 180: rpc_SetErr(EINVAL, "Invalid argument BLOB");
1.2 misho 181: return -1;
1.7.2.1 misho 182: } else
1.2 misho 183: rpc_srv_blobUnmap(blob);
184:
1.3 misho 185: memset(szFName, 0, sizeof szFName);
1.5 misho 186: snprintf(szFName, sizeof szFName, BLOB_FILE, AIT_GET_STR(&srv->srv_blob.dir), blob->blob_var);
1.3 misho 187: if (unlink(szFName) == -1) {
1.2 misho 188: LOGERR;
189: return -1;
190: }
191:
192: return 0;
193: }
194:
1.7.2.1 misho 195: /* ------------------------------------------------------------ */
1.2 misho 196:
197: /*
1.5 misho 198: * rpc_srv_sendBLOB() - Send mapped BLOB to client
199: *
1.2 misho 200: * @cli = Client instance
201: * @blob = Mapped BLOB element
202: * return: -1 error, 0 ok
203: */
204: int
205: rpc_srv_sendBLOB(rpc_cli_t * __restrict cli, rpc_blob_t * __restrict blob)
206: {
207: int ret, len;
208: uint8_t *pos;
209:
210: if (!cli || !blob || !blob->blob_data) {
1.5 misho 211: rpc_SetErr(EINVAL, "Invalid arguments");
1.2 misho 212: return -1;
213: }
214:
1.3 misho 215: for (ret = blob->blob_len, pos = blob->blob_data; ret > 0; ret -= len, pos += len) {
1.7.2.2 misho 216: len = send(cli->cli_sock, pos, ret, MSG_NOSIGNAL);
1.3 misho 217: if (len == -1) {
1.2 misho 218: LOGERR;
219: return -1;
220: }
1.3 misho 221: }
1.2 misho 222:
223: return ret;
224: }
225:
226: /*
1.5 misho 227: * rpc_srv_recvBLOB() - Receive BLOB from client
228: *
1.2 misho 229: * @cli = Client instance
230: * @blob = Mapped BLOB element
231: * return: -1 error, 0 ok, >0 unreceived data from client, may be error?
232: */
233: int
234: rpc_srv_recvBLOB(rpc_cli_t * __restrict cli, rpc_blob_t * __restrict blob)
235: {
236: int ret, len;
237: uint8_t *pos;
1.7 misho 238: struct pollfd pfd;
1.2 misho 239:
240: if (!cli || !blob || !blob->blob_data) {
1.5 misho 241: rpc_SetErr(EINVAL, "Invalid arguments");
1.2 misho 242: return -1;
1.7 misho 243: }
1.2 misho 244:
1.7 misho 245: pfd.fd = cli->cli_sock;
246: pfd.events = POLLIN | POLLPRI;
1.2 misho 247: for (ret = blob->blob_len, pos = blob->blob_data; ret > 0; ret -= len, pos += len) {
1.7.2.2 misho 248: if ((len = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 ||
1.7 misho 249: pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
1.2 misho 250: LOGERR;
251: return -1;
252: }
253:
1.3 misho 254: len = recv(cli->cli_sock, pos, ret, 0);
255: if (len == -1) {
1.2 misho 256: LOGERR;
257: return -1;
258: }
259: }
260:
261: return ret;
262: }
263:
1.7.2.2 misho 264: /* ------------------------------------------------------------ */
1.2 misho 265:
266: /*
1.5 misho 267: * rpc_cli_sendBLOB() - Send BLOB to server
268: *
1.2 misho 269: * @cli = Client instance
270: * @var = BLOB variable
271: * @data = BLOB data
272: * return: -1 error, 0 ok, 1 remote error
273: */
274: int
1.4 misho 275: rpc_cli_sendBLOB(rpc_cli_t * __restrict cli, ait_val_t * __restrict var, void * __restrict data)
1.2 misho 276: {
277: int ret, len;
278: uint8_t *pos;
279: struct tagBLOBHdr hdr;
1.7 misho 280: struct pollfd pfd;
1.2 misho 281:
282: if (!cli || !var || !data) {
1.5 misho 283: rpc_SetErr(EINVAL, "Invalid arguments");
1.2 misho 284: return -1;
1.7 misho 285: }
1.2 misho 286:
1.5 misho 287: rpc_addPktSession(&hdr.hdr_session, cli->cli_parent);
1.2 misho 288: hdr.hdr_cmd = set;
289: hdr.hdr_var = 0;
290: hdr.hdr_ret = 0;
1.5 misho 291: hdr.hdr_len = htonl(AIT_LEN(var));
292: /* calculate CRC */
293: hdr.hdr_crc ^= hdr.hdr_crc;
1.6 misho 294: hdr.hdr_crc = htons(crcFletcher16((u_short*) &hdr, sizeof hdr / 2));
1.5 misho 295:
296: /* send SET request */
1.2 misho 297: if (send(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {
298: LOGERR;
299: return -1;
300: }
301:
1.4 misho 302: /* send BLOB to server */
303: for (ret = AIT_LEN(var), pos = data; ret > 0; ret -= len, pos += len)
1.3 misho 304: if ((len = send(cli->cli_sock, pos, ret, 0)) == -1) {
1.2 misho 305: LOGERR;
306: return -1;
307: }
308:
1.5 misho 309: /* wait for reply */
1.7 misho 310: pfd.fd = cli->cli_sock;
311: pfd.events = POLLIN | POLLPRI;
1.7.2.3 misho 312: if ((ret = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 ||
1.7 misho 313: pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
314: if (ret)
1.2 misho 315: LOGERR;
1.7 misho 316: else
317: rpc_SetErr(ETIMEDOUT, "Timeout reached! Server not respond");
318: return -1;
1.2 misho 319: }
320: if (recv(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {
321: LOGERR;
1.5 misho 322: return 1;
323: }
324: /* check CRC */
325: ret = ntohs(hdr.hdr_crc);
326: hdr.hdr_crc ^= hdr.hdr_crc;
1.6 misho 327: if (ret != crcFletcher16((u_short*) &hdr, sizeof hdr / 2)) {
1.5 misho 328: rpc_SetErr(ERPCMISMATCH, "Bad CRC BLOB packet");
329: return 1;
1.2 misho 330: }
1.5 misho 331:
1.2 misho 332: if (hdr.hdr_cmd != error) {
1.5 misho 333: if (ntohl(hdr.hdr_len) != AIT_LEN(var)) {
334: rpc_SetErr(ECANCELED, "Bad return length packet");
335: return 1;
1.2 misho 336: }
337:
1.5 misho 338: var->val.blob = ntohl(hdr.hdr_var);
1.2 misho 339: }
340:
341: return hdr.hdr_cmd == error;
342: }
343:
344: /*
1.5 misho 345: * rpc_cli_recvBLOB() - Receive BLOB from server
346: *
1.2 misho 347: * @cli = Client instance
348: * @var = BLOB variable
349: * @data = BLOB data, must be free after use!
350: * return: -1 error, 0 ok, 1 remote error
351: */
352: int
1.4 misho 353: rpc_cli_recvBLOB(rpc_cli_t * __restrict cli, ait_val_t * __restrict var, void ** __restrict data)
1.2 misho 354: {
355: int ret, len;
356: uint8_t *pos;
1.7 misho 357: struct pollfd pfd;
1.2 misho 358: struct tagBLOBHdr hdr;
359:
360: if (!cli || !var || !data) {
1.5 misho 361: rpc_SetErr(EINVAL, "Invalid arguments");
1.2 misho 362: return -1;
1.7 misho 363: }
1.2 misho 364:
1.4 misho 365: *data = malloc(AIT_LEN(var));
1.2 misho 366: if (!*data) {
367: LOGERR;
368: return -1;
369: } else
1.4 misho 370: memset(*data, 0, AIT_LEN(var));
1.2 misho 371:
1.5 misho 372: rpc_addPktSession(&hdr.hdr_session, cli->cli_parent);
1.2 misho 373: hdr.hdr_cmd = get;
1.5 misho 374: hdr.hdr_var = htonl((uint32_t) AIT_GET_BLOB(var));
1.2 misho 375: hdr.hdr_ret = 0;
376: hdr.hdr_len = 0;
1.5 misho 377: /* calculate CRC */
378: hdr.hdr_crc ^= hdr.hdr_crc;
1.6 misho 379: hdr.hdr_crc = htons(crcFletcher16((u_short*) &hdr, sizeof hdr / 2));
1.5 misho 380:
381: /* send GET request */
1.2 misho 382: if (send(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {
383: LOGERR;
384: free(*data);
385: *data = NULL;
386: return -1;
387: }
388:
1.4 misho 389: /* receive BLOB from server */
1.7 misho 390: pfd.fd = cli->cli_sock;
391: pfd.events = POLLIN | POLLPRI;
1.4 misho 392: for (ret = AIT_LEN(var), pos = *data; ret > 0; ret -= len, pos += len) {
1.7.2.3 misho 393: if ((len = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 ||
1.7 misho 394: pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
1.2 misho 395: LOGERR;
396: free(*data);
397: *data = NULL;
398: return -1;
399: }
400:
1.3 misho 401: if ((len = recv(cli->cli_sock, pos, ret, 0)) == -1) {
1.2 misho 402: LOGERR;
403: free(*data);
404: *data = NULL;
405: return -1;
406: }
407: }
408:
1.5 misho 409: /* wait for reply */
1.7.2.3 misho 410: if ((len = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 ||
1.7 misho 411: pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
412: if (len)
1.2 misho 413: LOGERR;
1.7 misho 414: else
415: rpc_SetErr(ETIMEDOUT, "Timeout reached! Server not respond");
416: free(*data);
417: *data = NULL;
418: return 1;
1.2 misho 419: }
420: if (recv(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {
421: LOGERR;
422: free(*data);
423: *data = NULL;
1.5 misho 424: return 1;
425: }
426: /* check CRC */
427: ret = ntohs(hdr.hdr_crc);
428: hdr.hdr_crc ^= hdr.hdr_crc;
1.6 misho 429: if (ret != crcFletcher16((u_short*) &hdr, sizeof hdr / 2)) {
1.5 misho 430: rpc_SetErr(ERPCMISMATCH, "Bad CRC BLOB packet");
431: free(*data);
432: *data = NULL;
433: return 1;
1.2 misho 434: }
435: if (hdr.hdr_cmd != error) {
1.5 misho 436: if (ntohl(hdr.hdr_len) != AIT_LEN(var)) {
437: rpc_SetErr(ECANCELED, "Bad return length packet");
1.2 misho 438: free(*data);
439: *data = NULL;
1.5 misho 440: return 1;
1.2 misho 441: }
442: }
443:
444: return hdr.hdr_cmd == error;
445: }
446:
447: /*
1.5 misho 448: * rpc_cli_delBLOB() - Delete BLOB from server
449: *
1.2 misho 450: * @cli = Client instance
451: * @var = BLOB variable
452: * return: -1 error, 0 ok, 1 remote error
453: */
454: int
1.4 misho 455: rpc_cli_delBLOB(rpc_cli_t * __restrict cli, ait_val_t * __restrict var)
1.2 misho 456: {
457: struct tagBLOBHdr hdr;
1.7 misho 458: struct pollfd pfd;
1.5 misho 459: int ret;
1.2 misho 460:
461: if (!cli || !var) {
1.5 misho 462: rpc_SetErr(EINVAL, "Invalid arguments");
1.2 misho 463: return -1;
1.7 misho 464: }
1.2 misho 465:
1.5 misho 466: rpc_addPktSession(&hdr.hdr_session, cli->cli_parent);
1.2 misho 467: hdr.hdr_cmd = unset;
1.5 misho 468: hdr.hdr_var = htonl((uint32_t) AIT_GET_BLOB(var));
1.2 misho 469: hdr.hdr_ret = 0;
470: hdr.hdr_len = 0;
1.5 misho 471: /* calculate CRC */
472: hdr.hdr_crc ^= hdr.hdr_crc;
1.6 misho 473: hdr.hdr_crc = htons(crcFletcher16((u_short*) &hdr, sizeof hdr / 2));
1.5 misho 474:
475: /* send UNSET request */
1.2 misho 476: if (send(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {
477: LOGERR;
478: return -1;
479: }
480:
1.5 misho 481: /* wait for reply */
1.7 misho 482: pfd.fd = cli->cli_sock;
483: pfd.events = POLLIN | POLLPRI;
1.7.2.3 misho 484: if ((ret = poll(&pfd, 1, DEF_RPC_TIMEOUT * 1000)) < 1 ||
1.7 misho 485: pfd.revents & (POLLERR | POLLHUP | POLLNVAL)) {
486: if (ret)
1.2 misho 487: LOGERR;
1.7 misho 488: else
489: rpc_SetErr(ETIMEDOUT, "Timeout reached! Server not respond");
490: return 1;
1.2 misho 491: }
492: if (recv(cli->cli_sock, &hdr, sizeof hdr, 0) == -1) {
493: LOGERR;
1.5 misho 494: return 1;
495: }
496: /* check CRC */
497: ret = ntohs(hdr.hdr_crc);
498: hdr.hdr_crc ^= hdr.hdr_crc;
1.6 misho 499: if (ret != crcFletcher16((u_short*) &hdr, sizeof hdr / 2)) {
1.5 misho 500: rpc_SetErr(ERPCMISMATCH, "Bad CRC BLOB packet");
501: return 1;
1.2 misho 502: }
503:
504: return hdr.hdr_cmd == error;
505: }
506:
507: /*
1.5 misho 508: * rpc_cli_getBLOB() - Receive BLOB from server and Delete after that
509: *
1.2 misho 510: * @cli = Client instance
511: * @var = BLOB variable
512: * @data = BLOB data, must be free after use!
513: * return: -1 error, 0 ok, 1 remote error
514: */
515: inline int
1.4 misho 516: rpc_cli_getBLOB(rpc_cli_t * __restrict cli, ait_val_t * __restrict var, void ** __restrict data)
1.2 misho 517: {
518: int ret;
519:
520: ret = rpc_cli_recvBLOB(cli, var, data);
521: ret |= rpc_cli_delBLOB(cli, var) > 0 ? 2 : 0;
522:
523: return ret;
524: }
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