coroipcc.c 26 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174
  1. /*
  2. * vi: set autoindent tabstop=4 shiftwidth=4 :
  3. *
  4. * Copyright (c) 2002-2006 MontaVista Software, Inc.
  5. * Copyright (c) 2006-2009 Red Hat, Inc.
  6. *
  7. * All rights reserved.
  8. *
  9. * Author: Steven Dake (sdake@redhat.com)
  10. *
  11. * This software licensed under BSD license, the text of which follows:
  12. *
  13. * Redistribution and use in source and binary forms, with or without
  14. * modification, are permitted provided that the following conditions are met:
  15. *
  16. * - Redistributions of source code must retain the above copyright notice,
  17. * this list of conditions and the following disclaimer.
  18. * - Redistributions in binary form must reproduce the above copyright notice,
  19. * this list of conditions and the following disclaimer in the documentation
  20. * and/or other materials provided with the distribution.
  21. * - Neither the name of the MontaVista Software, Inc. nor the names of its
  22. * contributors may be used to endorse or promote products derived from this
  23. * software without specific prior written permission.
  24. *
  25. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  26. * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  27. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  28. * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
  29. * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  30. * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  31. * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  32. * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  33. * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  34. * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
  35. * THE POSSIBILITY OF SUCH DAMAGE.
  36. */
  37. #include <config.h>
  38. #include <stdlib.h>
  39. #include <stdio.h>
  40. #include <unistd.h>
  41. #include <limits.h>
  42. #include <errno.h>
  43. #include <string.h>
  44. #include <fcntl.h>
  45. #include <sys/ioctl.h>
  46. #include <sys/types.h>
  47. #include <sys/uio.h>
  48. #include <sys/socket.h>
  49. #include <sys/select.h>
  50. #include <sys/time.h>
  51. #include <sys/un.h>
  52. #include <net/if.h>
  53. #include <arpa/inet.h>
  54. #include <netinet/in.h>
  55. #include <assert.h>
  56. #include <sys/shm.h>
  57. #include <sys/mman.h>
  58. #include <corosync/corotypes.h>
  59. #include <corosync/coroipc_types.h>
  60. #include <corosync/coroipc_ipc.h>
  61. #include <corosync/coroipcc.h>
  62. #include <corosync/hdb.h>
  63. #if _POSIX_THREAD_PROCESS_SHARED > 0
  64. #include <semaphore.h>
  65. #else
  66. #include <sys/sem.h>
  67. #endif
  68. #include "util.h"
  69. struct ipc_instance {
  70. int fd;
  71. struct control_buffer *control_buffer;
  72. char *request_buffer;
  73. char *response_buffer;
  74. char *dispatch_buffer;
  75. size_t control_size;
  76. size_t request_size;
  77. size_t response_size;
  78. size_t dispatch_size;
  79. uid_t euid;
  80. pthread_mutex_t mutex;
  81. };
  82. void ipc_hdb_destructor (void *context);
  83. DECLARE_HDB_DATABASE(ipc_hdb,ipc_hdb_destructor);
  84. #if defined(COROSYNC_LINUX) || defined(COROSYNC_SOLARIS)
  85. #define COROSYNC_SUN_LEN(a) sizeof(*(a))
  86. #else
  87. #define COROSYNC_SUN_LEN(a) SUN_LEN(a)
  88. #endif
  89. #ifdef SO_NOSIGPIPE
  90. static void socket_nosigpipe(int s)
  91. {
  92. int on = 1;
  93. setsockopt(s, SOL_SOCKET, SO_NOSIGPIPE, (void *)&on, sizeof(on));
  94. }
  95. #endif
  96. #ifndef MSG_NOSIGNAL
  97. #define MSG_NOSIGNAL 0
  98. #endif
  99. static inline int shared_mem_dispatch_bytes_left (struct ipc_instance *context)
  100. {
  101. unsigned int n_read;
  102. unsigned int n_write;
  103. unsigned int bytes_left;
  104. n_read = context->control_buffer->read;
  105. n_write = context->control_buffer->write;
  106. if (n_read <= n_write) {
  107. bytes_left = context->dispatch_size - n_write + n_read;
  108. } else {
  109. bytes_left = n_read - n_write;
  110. }
  111. return (bytes_left);
  112. }
  113. static cs_error_t
  114. socket_send (
  115. int s,
  116. void *msg,
  117. size_t len)
  118. {
  119. cs_error_t res = CS_OK;
  120. int result;
  121. struct msghdr msg_send;
  122. struct iovec iov_send;
  123. char *rbuf = msg;
  124. int processed = 0;
  125. msg_send.msg_iov = &iov_send;
  126. msg_send.msg_iovlen = 1;
  127. msg_send.msg_name = 0;
  128. msg_send.msg_namelen = 0;
  129. #if !defined(COROSYNC_SOLARIS)
  130. msg_send.msg_control = 0;
  131. msg_send.msg_controllen = 0;
  132. msg_send.msg_flags = 0;
  133. #else
  134. msg_send.msg_accrights = NULL;
  135. msg_send.msg_accrightslen = 0;
  136. #endif
  137. retry_send:
  138. iov_send.iov_base = &rbuf[processed];
  139. iov_send.iov_len = len - processed;
  140. result = sendmsg (s, &msg_send, MSG_NOSIGNAL);
  141. if (result == -1) {
  142. switch (errno) {
  143. case EINTR:
  144. res = CS_ERR_TRY_AGAIN;
  145. goto res_exit;
  146. case EAGAIN:
  147. goto retry_send;
  148. break;
  149. default:
  150. res = CS_ERR_LIBRARY;
  151. goto res_exit;
  152. }
  153. }
  154. processed += result;
  155. if (processed != len) {
  156. goto retry_send;
  157. }
  158. return (CS_OK);
  159. res_exit:
  160. return (res);
  161. }
  162. static cs_error_t
  163. socket_recv (
  164. int s,
  165. void *msg,
  166. size_t len)
  167. {
  168. cs_error_t res = CS_OK;
  169. int result;
  170. struct msghdr msg_recv;
  171. struct iovec iov_recv;
  172. char *rbuf = msg;
  173. int processed = 0;
  174. msg_recv.msg_iov = &iov_recv;
  175. msg_recv.msg_iovlen = 1;
  176. msg_recv.msg_name = 0;
  177. msg_recv.msg_namelen = 0;
  178. #if !defined (COROSYNC_SOLARIS)
  179. msg_recv.msg_control = 0;
  180. msg_recv.msg_controllen = 0;
  181. msg_recv.msg_flags = 0;
  182. #else
  183. msg_recv.msg_accrights = NULL;
  184. msg_recv.msg_accrightslen = 0;
  185. #endif
  186. retry_recv:
  187. iov_recv.iov_base = (void *)&rbuf[processed];
  188. iov_recv.iov_len = len - processed;
  189. result = recvmsg (s, &msg_recv, MSG_NOSIGNAL|MSG_WAITALL);
  190. if (result == -1) {
  191. switch (errno) {
  192. case EINTR:
  193. res = CS_ERR_TRY_AGAIN;
  194. goto res_exit;
  195. case EAGAIN:
  196. goto retry_recv;
  197. break;
  198. default:
  199. res = CS_ERR_LIBRARY;
  200. goto res_exit;
  201. }
  202. }
  203. /*
  204. * EOF is also detected when recvmsg return 0.
  205. */
  206. if (result == 0) {
  207. res = CS_ERR_LIBRARY;
  208. goto res_exit;
  209. }
  210. processed += result;
  211. if (processed != len) {
  212. goto retry_recv;
  213. }
  214. assert (processed == len);
  215. res_exit:
  216. return (res);
  217. }
  218. static int
  219. priv_change_send (struct ipc_instance *ipc_instance)
  220. {
  221. #if _POSIX_THREAD_PROCESS_SHARED < 1
  222. char buf_req;
  223. mar_req_priv_change req_priv_change;
  224. unsigned int res;
  225. req_priv_change.euid = geteuid();
  226. /*
  227. * Don't resend request unless euid has changed
  228. */
  229. if (ipc_instance->euid == req_priv_change.euid) {
  230. return (0);
  231. }
  232. req_priv_change.egid = getegid();
  233. buf_req = MESSAGE_REQ_CHANGE_EUID;
  234. res = socket_send (ipc_instance->fd, &buf_req, 1);
  235. if (res == -1) {
  236. return (-1);
  237. }
  238. res = socket_send (ipc_instance->fd, &req_priv_change,
  239. sizeof (req_priv_change));
  240. if (res == -1) {
  241. return (-1);
  242. }
  243. ipc_instance->euid = req_priv_change.euid;
  244. #else
  245. ipc_instance = NULL;
  246. #endif
  247. return (0);
  248. }
  249. static int
  250. circular_memory_map (char *path, const char *file, void **buf, size_t bytes)
  251. {
  252. int32_t fd;
  253. void *addr_orig;
  254. void *addr;
  255. int32_t res;
  256. int32_t i;
  257. int32_t written;
  258. char *buffer;
  259. long page_size;
  260. snprintf (path, PATH_MAX, "/dev/shm/%s", file);
  261. fd = mkstemp (path);
  262. if (fd == -1) {
  263. snprintf (path, PATH_MAX, LOCALSTATEDIR "/run/%s", file);
  264. fd = mkstemp (path);
  265. if (fd == -1) {
  266. return (-1);
  267. }
  268. }
  269. res = ftruncate (fd, bytes);
  270. if (res == -1) {
  271. goto error_close_unlink;
  272. }
  273. page_size = sysconf(_SC_PAGESIZE);
  274. if (page_size == -1) {
  275. goto error_close_unlink;
  276. }
  277. buffer = malloc (page_size);
  278. if (buffer == NULL) {
  279. goto error_close_unlink;
  280. }
  281. memset (buffer, 0, page_size);
  282. for (i = 0; i < (bytes / page_size); i++) {
  283. retry_write:
  284. written = write (fd, buffer, page_size);
  285. if (written == -1 && errno == EINTR) {
  286. goto retry_write;
  287. }
  288. if (written != page_size) {
  289. free (buffer);
  290. goto error_close_unlink;
  291. }
  292. }
  293. free (buffer);
  294. addr_orig = mmap (NULL, bytes << 1, PROT_NONE,
  295. MAP_ANONYMOUS | MAP_PRIVATE, -1, 0);
  296. if (addr_orig == MAP_FAILED) {
  297. goto error_close_unlink;
  298. }
  299. addr = mmap (addr_orig, bytes, PROT_READ | PROT_WRITE,
  300. MAP_FIXED | MAP_SHARED, fd, 0);
  301. if (addr != addr_orig) {
  302. goto error_close_unlink;
  303. }
  304. #ifdef COROSYNC_BSD
  305. madvise(addr_orig, bytes, MADV_NOSYNC);
  306. #endif
  307. addr = mmap (((char *)addr_orig) + bytes,
  308. bytes, PROT_READ | PROT_WRITE,
  309. MAP_FIXED | MAP_SHARED, fd, 0);
  310. if (addr == MAP_FAILED) {
  311. goto error_close_unlink;
  312. }
  313. #ifdef COROSYNC_BSD
  314. madvise(((char *)addr_orig) + bytes, bytes, MADV_NOSYNC);
  315. #endif
  316. res = close (fd);
  317. if (res) {
  318. return (-1);
  319. }
  320. *buf = addr_orig;
  321. return (0);
  322. error_close_unlink:
  323. close (fd);
  324. unlink(path);
  325. return (-1);
  326. }
  327. static void
  328. memory_unmap (void *addr, size_t bytes)
  329. {
  330. (void)munmap (addr, bytes);
  331. }
  332. void ipc_hdb_destructor (void *context ) {
  333. struct ipc_instance *ipc_instance = (struct ipc_instance *)context;
  334. /*
  335. * << 1 (or multiplied by 2) because this is a wrapped memory buffer
  336. */
  337. memory_unmap (ipc_instance->control_buffer, ipc_instance->control_size);
  338. memory_unmap (ipc_instance->request_buffer, ipc_instance->request_size);
  339. memory_unmap (ipc_instance->response_buffer, ipc_instance->response_size);
  340. memory_unmap (ipc_instance->dispatch_buffer, (ipc_instance->dispatch_size) << 1);
  341. }
  342. static int
  343. memory_map (char *path, const char *file, void **buf, size_t bytes)
  344. {
  345. int32_t fd;
  346. void *addr_orig;
  347. void *addr;
  348. int32_t res;
  349. char *buffer;
  350. int32_t i;
  351. int32_t written;
  352. long page_size;
  353. snprintf (path, PATH_MAX, "/dev/shm/%s", file);
  354. fd = mkstemp (path);
  355. if (fd == -1) {
  356. snprintf (path, PATH_MAX, LOCALSTATEDIR "/run/%s", file);
  357. fd = mkstemp (path);
  358. if (fd == -1) {
  359. return (-1);
  360. }
  361. }
  362. res = ftruncate (fd, bytes);
  363. if (res == -1) {
  364. goto error_close_unlink;
  365. }
  366. page_size = sysconf(_SC_PAGESIZE);
  367. if (page_size == -1) {
  368. goto error_close_unlink;
  369. }
  370. buffer = malloc (page_size);
  371. if (buffer == NULL) {
  372. goto error_close_unlink;
  373. }
  374. memset (buffer, 0, page_size);
  375. for (i = 0; i < (bytes / page_size); i++) {
  376. retry_write:
  377. written = write (fd, buffer, page_size);
  378. if (written == -1 && errno == EINTR) {
  379. goto retry_write;
  380. }
  381. if (written != page_size) {
  382. free (buffer);
  383. goto error_close_unlink;
  384. }
  385. }
  386. free (buffer);
  387. addr_orig = mmap (NULL, bytes, PROT_NONE,
  388. MAP_ANONYMOUS | MAP_PRIVATE, -1, 0);
  389. if (addr_orig == MAP_FAILED) {
  390. goto error_close_unlink;
  391. }
  392. addr = mmap (addr_orig, bytes, PROT_READ | PROT_WRITE,
  393. MAP_FIXED | MAP_SHARED, fd, 0);
  394. if (addr != addr_orig) {
  395. goto error_close_unlink;
  396. }
  397. #ifdef COROSYNC_BSD
  398. madvise(addr_orig, bytes, MADV_NOSYNC);
  399. #endif
  400. res = close (fd);
  401. if (res) {
  402. return (-1);
  403. }
  404. *buf = addr_orig;
  405. return 0;
  406. error_close_unlink:
  407. close (fd);
  408. unlink(path);
  409. return -1;
  410. }
  411. static cs_error_t
  412. msg_send (
  413. struct ipc_instance *ipc_instance,
  414. const struct iovec *iov,
  415. unsigned int iov_len)
  416. {
  417. int i;
  418. int res;
  419. int req_buffer_idx = 0;
  420. for (i = 0; i < iov_len; i++) {
  421. if ((req_buffer_idx + iov[i].iov_len) >
  422. ipc_instance->request_size) {
  423. return (CS_ERR_INVALID_PARAM);
  424. }
  425. memcpy (&ipc_instance->request_buffer[req_buffer_idx],
  426. iov[i].iov_base,
  427. iov[i].iov_len);
  428. req_buffer_idx += iov[i].iov_len;
  429. }
  430. /*
  431. * Signal semaphore #3 and #0 indicting a new message from client
  432. * to server request queue
  433. */
  434. res = ipc_sem_post (ipc_instance->control_buffer, SEMAPHORE_REQUEST);
  435. if (res != CS_OK) {
  436. return (CS_ERR_LIBRARY);
  437. }
  438. res = ipc_sem_post (ipc_instance->control_buffer, SEMAPHORE_REQUEST_OR_FLUSH_OR_EXIT);
  439. if (res != CS_OK) {
  440. return (CS_ERR_LIBRARY);
  441. }
  442. return (CS_OK);
  443. }
  444. static cs_error_t
  445. reply_receive (
  446. struct ipc_instance *ipc_instance,
  447. void *res_msg,
  448. size_t res_len)
  449. {
  450. coroipc_response_header_t *response_header;
  451. cs_error_t res;
  452. retry_ipc_sem_wait:
  453. res = ipc_sem_wait (ipc_instance->control_buffer, SEMAPHORE_RESPONSE, ipc_instance->fd);
  454. if (res != CS_OK) {
  455. if (res == CS_ERR_TRY_AGAIN) {
  456. priv_change_send (ipc_instance);
  457. goto retry_ipc_sem_wait;
  458. } else {
  459. return (res);
  460. }
  461. }
  462. response_header = (coroipc_response_header_t *)ipc_instance->response_buffer;
  463. if (response_header->error == CS_ERR_TRY_AGAIN) {
  464. return (CS_ERR_TRY_AGAIN);
  465. }
  466. memcpy (res_msg, ipc_instance->response_buffer, res_len);
  467. return (CS_OK);
  468. }
  469. static cs_error_t
  470. reply_receive_in_buf (
  471. struct ipc_instance *ipc_instance,
  472. void **res_msg)
  473. {
  474. cs_error_t res;
  475. retry_ipc_sem_wait:
  476. res = ipc_sem_wait (ipc_instance->control_buffer, SEMAPHORE_RESPONSE, ipc_instance->fd);
  477. if (res != CS_OK) {
  478. if (res == CS_ERR_TRY_AGAIN) {
  479. priv_change_send (ipc_instance);
  480. goto retry_ipc_sem_wait;
  481. } else {
  482. return (res);
  483. }
  484. }
  485. *res_msg = (char *)ipc_instance->response_buffer;
  486. return (CS_OK);
  487. }
  488. /*
  489. * External API
  490. */
  491. cs_error_t
  492. coroipcc_service_connect (
  493. const char *socket_name,
  494. unsigned int service,
  495. size_t request_size,
  496. size_t response_size,
  497. size_t dispatch_size,
  498. hdb_handle_t *handle)
  499. {
  500. int request_fd;
  501. struct sockaddr_un address;
  502. cs_error_t res;
  503. struct ipc_instance *ipc_instance;
  504. #if _POSIX_THREAD_PROCESS_SHARED < 1
  505. key_t semkey = 0;
  506. union semun semun;
  507. #endif
  508. int sys_res;
  509. mar_req_setup_t req_setup;
  510. mar_res_setup_t res_setup;
  511. char control_map_path[PATH_MAX];
  512. char request_map_path[PATH_MAX];
  513. char response_map_path[PATH_MAX];
  514. char dispatch_map_path[PATH_MAX];
  515. res = hdb_error_to_cs (hdb_handle_create (&ipc_hdb,
  516. sizeof (struct ipc_instance), handle));
  517. if (res != CS_OK) {
  518. return (res);
  519. }
  520. res = hdb_error_to_cs (hdb_handle_get (&ipc_hdb, *handle, (void **)&ipc_instance));
  521. if (res != CS_OK) {
  522. return (res);
  523. }
  524. res_setup.error = CS_ERR_LIBRARY;
  525. #if defined(COROSYNC_SOLARIS)
  526. request_fd = socket (PF_UNIX, SOCK_STREAM, 0);
  527. #else
  528. request_fd = socket (PF_LOCAL, SOCK_STREAM, 0);
  529. #endif
  530. if (request_fd == -1) {
  531. return (CS_ERR_LIBRARY);
  532. }
  533. #ifdef SO_NOSIGPIPE
  534. socket_nosigpipe (request_fd);
  535. #endif
  536. memset (&address, 0, sizeof (struct sockaddr_un));
  537. address.sun_family = AF_UNIX;
  538. #if defined(COROSYNC_BSD) || defined(COROSYNC_DARWIN)
  539. address.sun_len = SUN_LEN(&address);
  540. #endif
  541. #if defined(COROSYNC_LINUX)
  542. sprintf (address.sun_path + 1, "%s", socket_name);
  543. #else
  544. sprintf (address.sun_path, "%s/%s", SOCKETDIR, socket_name);
  545. #endif
  546. sys_res = connect (request_fd, (struct sockaddr *)&address,
  547. COROSYNC_SUN_LEN(&address));
  548. if (sys_res == -1) {
  549. res = CS_ERR_TRY_AGAIN;
  550. goto error_connect;
  551. }
  552. sys_res = memory_map (
  553. control_map_path,
  554. "control_buffer-XXXXXX",
  555. (void *)&ipc_instance->control_buffer,
  556. 8192);
  557. if (sys_res == -1) {
  558. res = CS_ERR_LIBRARY;
  559. goto error_connect;
  560. }
  561. sys_res = memory_map (
  562. request_map_path,
  563. "request_buffer-XXXXXX",
  564. (void *)&ipc_instance->request_buffer,
  565. request_size);
  566. if (sys_res == -1) {
  567. res = CS_ERR_LIBRARY;
  568. goto error_request_buffer;
  569. }
  570. sys_res = memory_map (
  571. response_map_path,
  572. "response_buffer-XXXXXX",
  573. (void *)&ipc_instance->response_buffer,
  574. response_size);
  575. if (sys_res == -1) {
  576. res = CS_ERR_LIBRARY;
  577. goto error_response_buffer;
  578. }
  579. sys_res = circular_memory_map (
  580. dispatch_map_path,
  581. "dispatch_buffer-XXXXXX",
  582. (void *)&ipc_instance->dispatch_buffer,
  583. dispatch_size);
  584. if (sys_res == -1) {
  585. res = CS_ERR_LIBRARY;
  586. goto error_dispatch_buffer;
  587. }
  588. #if _POSIX_THREAD_PROCESS_SHARED > 0
  589. sem_init (&ipc_instance->control_buffer->sem_request_or_flush_or_exit, 1, 0);
  590. sem_init (&ipc_instance->control_buffer->sem_request, 1, 0);
  591. sem_init (&ipc_instance->control_buffer->sem_response, 1, 0);
  592. sem_init (&ipc_instance->control_buffer->sem_dispatch, 1, 0);
  593. #else
  594. {
  595. int i;
  596. /*
  597. * Allocate a semaphore segment
  598. */
  599. while (1) {
  600. semkey = random();
  601. ipc_instance->euid = geteuid ();
  602. if ((ipc_instance->control_buffer->semid
  603. = semget (semkey, 4, IPC_CREAT|IPC_EXCL|0600)) != -1) {
  604. break;
  605. }
  606. /*
  607. * EACCESS can be returned as non root user when opening a different
  608. * users semaphore.
  609. *
  610. * EEXIST can happen when we are a root or nonroot user opening
  611. * an existing shared memory segment for which we have access
  612. */
  613. if (errno != EEXIST && errno != EACCES) {
  614. res = CS_ERR_LIBRARY;
  615. goto error_exit;
  616. }
  617. }
  618. for (i = 0; i < 4; i++) {
  619. semun.val = 0;
  620. sys_res = semctl (ipc_instance->control_buffer->semid, i, SETVAL, semun);
  621. if (sys_res != 0) {
  622. res = CS_ERR_LIBRARY;
  623. goto error_exit;
  624. }
  625. }
  626. }
  627. #endif
  628. /*
  629. * Initialize IPC setup message
  630. */
  631. req_setup.service = service;
  632. strcpy (req_setup.control_file, control_map_path);
  633. strcpy (req_setup.request_file, request_map_path);
  634. strcpy (req_setup.response_file, response_map_path);
  635. strcpy (req_setup.dispatch_file, dispatch_map_path);
  636. req_setup.control_size = 8192;
  637. req_setup.request_size = request_size;
  638. req_setup.response_size = response_size;
  639. req_setup.dispatch_size = dispatch_size;
  640. #if _POSIX_THREAD_PROCESS_SHARED < 1
  641. req_setup.semkey = semkey;
  642. #endif
  643. res = socket_send (request_fd, &req_setup, sizeof (mar_req_setup_t));
  644. if (res != CS_OK) {
  645. goto error_exit;
  646. }
  647. res = socket_recv (request_fd, &res_setup, sizeof (mar_res_setup_t));
  648. if (res != CS_OK) {
  649. goto error_exit;
  650. }
  651. ipc_instance->fd = request_fd;
  652. if (res_setup.error == CS_ERR_TRY_AGAIN) {
  653. res = res_setup.error;
  654. goto error_exit;
  655. }
  656. ipc_instance->control_size = 8192;
  657. ipc_instance->request_size = request_size;
  658. ipc_instance->response_size = response_size;
  659. ipc_instance->dispatch_size = dispatch_size;
  660. pthread_mutex_init (&ipc_instance->mutex, NULL);
  661. hdb_handle_put (&ipc_hdb, *handle);
  662. return (res_setup.error);
  663. error_exit:
  664. #if _POSIX_THREAD_PROCESS_SHARED < 1
  665. if (ipc_instance->control_buffer->semid > 0)
  666. semctl (ipc_instance->control_buffer->semid, 0, IPC_RMID);
  667. #endif
  668. memory_unmap (ipc_instance->dispatch_buffer, dispatch_size);
  669. error_dispatch_buffer:
  670. memory_unmap (ipc_instance->response_buffer, response_size);
  671. error_response_buffer:
  672. memory_unmap (ipc_instance->request_buffer, request_size);
  673. error_request_buffer:
  674. memory_unmap (ipc_instance->control_buffer, 8192);
  675. error_connect:
  676. close (request_fd);
  677. hdb_handle_destroy (&ipc_hdb, *handle);
  678. hdb_handle_put (&ipc_hdb, *handle);
  679. return (res);
  680. }
  681. cs_error_t
  682. coroipcc_service_disconnect (
  683. hdb_handle_t handle)
  684. {
  685. cs_error_t res;
  686. struct ipc_instance *ipc_instance;
  687. res = hdb_error_to_cs (hdb_handle_get (&ipc_hdb, handle, (void **)&ipc_instance));
  688. if (res != CS_OK) {
  689. return (res);
  690. }
  691. shutdown (ipc_instance->fd, SHUT_RDWR);
  692. close (ipc_instance->fd);
  693. hdb_handle_destroy (&ipc_hdb, handle);
  694. hdb_handle_put (&ipc_hdb, handle);
  695. return (CS_OK);
  696. }
  697. cs_error_t
  698. coroipcc_dispatch_flow_control_get (
  699. hdb_handle_t handle,
  700. unsigned int *flow_control_state)
  701. {
  702. struct ipc_instance *ipc_instance;
  703. cs_error_t res;
  704. res = hdb_error_to_cs (hdb_handle_get (&ipc_hdb, handle, (void **)&ipc_instance));
  705. if (res != CS_OK) {
  706. return (res);
  707. }
  708. *flow_control_state = ipc_instance->control_buffer->flow_control_enabled;
  709. hdb_handle_put (&ipc_hdb, handle);
  710. return (res);
  711. }
  712. cs_error_t
  713. coroipcc_fd_get (
  714. hdb_handle_t handle,
  715. int *fd)
  716. {
  717. struct ipc_instance *ipc_instance;
  718. cs_error_t res;
  719. res = hdb_error_to_cs (hdb_handle_get (&ipc_hdb, handle, (void **)&ipc_instance));
  720. if (res != CS_OK) {
  721. return (res);
  722. }
  723. *fd = ipc_instance->fd;
  724. hdb_handle_put (&ipc_hdb, handle);
  725. return (res);
  726. }
  727. cs_error_t
  728. coroipcc_dispatch_get (
  729. hdb_handle_t handle,
  730. void **data,
  731. int timeout)
  732. {
  733. struct pollfd ufds;
  734. int poll_events;
  735. char buf;
  736. struct ipc_instance *ipc_instance;
  737. char *data_addr;
  738. cs_error_t error = CS_OK;
  739. int res;
  740. error = hdb_error_to_cs (hdb_handle_get (&ipc_hdb, handle, (void **)&ipc_instance));
  741. if (error != CS_OK) {
  742. return (error);
  743. }
  744. if (shared_mem_dispatch_bytes_left (ipc_instance) > (ipc_instance->dispatch_size/2)) {
  745. /*
  746. * Notify coroipcs to flush any pending dispatch messages
  747. */
  748. res = ipc_sem_post (ipc_instance->control_buffer, SEMAPHORE_REQUEST_OR_FLUSH_OR_EXIT);
  749. if (res != CS_OK) {
  750. error = CS_ERR_LIBRARY;
  751. goto error_put;
  752. }
  753. }
  754. *data = NULL;
  755. ufds.fd = ipc_instance->fd;
  756. ufds.events = POLLIN;
  757. ufds.revents = 0;
  758. poll_events = poll (&ufds, 1, timeout);
  759. if (poll_events == -1 && errno == EINTR) {
  760. error = CS_ERR_TRY_AGAIN;
  761. goto error_put;
  762. } else
  763. if (poll_events == -1) {
  764. error = CS_ERR_LIBRARY;
  765. goto error_put;
  766. } else
  767. if (poll_events == 0) {
  768. error = CS_ERR_TRY_AGAIN;
  769. goto error_put;
  770. }
  771. if (poll_events == 1 && (ufds.revents & (POLLERR|POLLHUP))) {
  772. error = CS_ERR_LIBRARY;
  773. goto error_put;
  774. }
  775. error = socket_recv (ipc_instance->fd, &buf, 1);
  776. #if defined(COROSYNC_SOLARIS) || defined(COROSYNC_BSD) || defined(COROSYNC_DARWIN)
  777. /* On many OS poll() never returns POLLHUP or POLLERR.
  778. * EOF is detected when recvmsg() return 0.
  779. */
  780. if ( error == CS_ERR_LIBRARY )
  781. goto error_put;
  782. #endif
  783. assert (error == CS_OK);
  784. if (shared_mem_dispatch_bytes_left (ipc_instance) > (ipc_instance->dispatch_size/2)) {
  785. /*
  786. * Notify coroipcs to flush any pending dispatch messages
  787. */
  788. res = ipc_sem_post (ipc_instance->control_buffer, SEMAPHORE_REQUEST_OR_FLUSH_OR_EXIT);
  789. if (res != CS_OK) {
  790. error = CS_ERR_LIBRARY;
  791. goto error_put;
  792. }
  793. }
  794. data_addr = ipc_instance->dispatch_buffer;
  795. data_addr = &data_addr[ipc_instance->control_buffer->read];
  796. *data = (void *)data_addr;
  797. return (CS_OK);
  798. error_put:
  799. hdb_handle_put (&ipc_hdb, handle);
  800. return (error);
  801. }
  802. cs_error_t
  803. coroipcc_dispatch_put (hdb_handle_t handle)
  804. {
  805. coroipc_response_header_t *header;
  806. struct ipc_instance *ipc_instance;
  807. cs_error_t res;
  808. char *addr;
  809. unsigned int read_idx;
  810. res = hdb_error_to_cs (hdb_handle_get_always (&ipc_hdb, handle, (void **)&ipc_instance));
  811. if (res != CS_OK) {
  812. return (res);
  813. }
  814. retry_ipc_sem_wait:
  815. res = ipc_sem_wait (ipc_instance->control_buffer, SEMAPHORE_DISPATCH, ipc_instance->fd);
  816. if (res != CS_OK) {
  817. if (res == CS_ERR_TRY_AGAIN) {
  818. priv_change_send (ipc_instance);
  819. goto retry_ipc_sem_wait;
  820. } else {
  821. goto error_exit;
  822. }
  823. }
  824. addr = ipc_instance->dispatch_buffer;
  825. read_idx = ipc_instance->control_buffer->read;
  826. header = (coroipc_response_header_t *) &addr[read_idx];
  827. ipc_instance->control_buffer->read =
  828. ((read_idx + header->size + 7) & 0xFFFFFFF8) %
  829. ipc_instance->dispatch_size;
  830. /*
  831. * Put from dispatch get and also from this call's get
  832. */
  833. res = CS_OK;
  834. error_exit:
  835. hdb_handle_put (&ipc_hdb, handle);
  836. hdb_handle_put (&ipc_hdb, handle);
  837. return (res);
  838. }
  839. cs_error_t
  840. coroipcc_msg_send_reply_receive (
  841. hdb_handle_t handle,
  842. const struct iovec *iov,
  843. unsigned int iov_len,
  844. void *res_msg,
  845. size_t res_len)
  846. {
  847. cs_error_t res;
  848. struct ipc_instance *ipc_instance;
  849. res = hdb_error_to_cs (hdb_handle_get (&ipc_hdb, handle, (void **)&ipc_instance));
  850. if (res != CS_OK) {
  851. return (res);
  852. }
  853. pthread_mutex_lock (&ipc_instance->mutex);
  854. res = msg_send (ipc_instance, iov, iov_len);
  855. if (res != CS_OK) {
  856. goto error_exit;
  857. }
  858. res = reply_receive (ipc_instance, res_msg, res_len);
  859. error_exit:
  860. pthread_mutex_unlock (&ipc_instance->mutex);
  861. hdb_handle_put (&ipc_hdb, handle);
  862. return (res);
  863. }
  864. cs_error_t
  865. coroipcc_msg_send_reply_receive_in_buf_get (
  866. hdb_handle_t handle,
  867. const struct iovec *iov,
  868. unsigned int iov_len,
  869. void **res_msg)
  870. {
  871. unsigned int res;
  872. struct ipc_instance *ipc_instance;
  873. res = hdb_error_to_cs (hdb_handle_get (&ipc_hdb, handle, (void **)&ipc_instance));
  874. if (res != CS_OK) {
  875. return (res);
  876. }
  877. pthread_mutex_lock (&ipc_instance->mutex);
  878. res = msg_send (ipc_instance, iov, iov_len);
  879. if (res != CS_OK) {
  880. goto error_exit;
  881. }
  882. res = reply_receive_in_buf (ipc_instance, res_msg);
  883. error_exit:
  884. pthread_mutex_unlock (&ipc_instance->mutex);
  885. return (res);
  886. }
  887. cs_error_t
  888. coroipcc_msg_send_reply_receive_in_buf_put (
  889. hdb_handle_t handle)
  890. {
  891. unsigned int res;
  892. struct ipc_instance *ipc_instance;
  893. res = hdb_error_to_cs (hdb_handle_get (&ipc_hdb, handle, (void **)&ipc_instance));
  894. if (res != CS_OK) {
  895. return (res);
  896. }
  897. hdb_handle_put (&ipc_hdb, handle);
  898. hdb_handle_put (&ipc_hdb, handle);
  899. return (res);
  900. }
  901. cs_error_t
  902. coroipcc_zcb_alloc (
  903. hdb_handle_t handle,
  904. void **buffer,
  905. size_t size,
  906. size_t header_size)
  907. {
  908. struct ipc_instance *ipc_instance;
  909. void *buf = NULL;
  910. char path[PATH_MAX];
  911. unsigned int res;
  912. mar_req_coroipcc_zc_alloc_t req_coroipcc_zc_alloc;
  913. coroipc_response_header_t res_coroipcs_zc_alloc;
  914. size_t map_size;
  915. struct iovec iovec;
  916. struct coroipcs_zc_header *hdr;
  917. res = hdb_error_to_cs (hdb_handle_get (&ipc_hdb, handle, (void **)&ipc_instance));
  918. if (res != CS_OK) {
  919. return (res);
  920. }
  921. map_size = size + header_size + sizeof (struct coroipcs_zc_header);
  922. res = memory_map (path, "corosync_zerocopy-XXXXXX", &buf, map_size);
  923. assert (res != -1);
  924. req_coroipcc_zc_alloc.header.size = sizeof (mar_req_coroipcc_zc_alloc_t);
  925. req_coroipcc_zc_alloc.header.id = ZC_ALLOC_HEADER;
  926. req_coroipcc_zc_alloc.map_size = map_size;
  927. strcpy (req_coroipcc_zc_alloc.path_to_file, path);
  928. iovec.iov_base = (void *)&req_coroipcc_zc_alloc;
  929. iovec.iov_len = sizeof (mar_req_coroipcc_zc_alloc_t);
  930. res = coroipcc_msg_send_reply_receive (
  931. handle,
  932. &iovec,
  933. 1,
  934. &res_coroipcs_zc_alloc,
  935. sizeof (coroipc_response_header_t));
  936. hdr = (struct coroipcs_zc_header *)buf;
  937. hdr->map_size = map_size;
  938. *buffer = ((char *)buf) + sizeof (struct coroipcs_zc_header);
  939. hdb_handle_put (&ipc_hdb, handle);
  940. return (res);
  941. }
  942. cs_error_t
  943. coroipcc_zcb_free (
  944. hdb_handle_t handle,
  945. void *buffer)
  946. {
  947. struct ipc_instance *ipc_instance;
  948. mar_req_coroipcc_zc_free_t req_coroipcc_zc_free;
  949. coroipc_response_header_t res_coroipcs_zc_free;
  950. struct iovec iovec;
  951. unsigned int res;
  952. struct coroipcs_zc_header *header = (struct coroipcs_zc_header *)((char *)buffer - sizeof (struct coroipcs_zc_header));
  953. res = hdb_error_to_cs (hdb_handle_get (&ipc_hdb, handle, (void **)&ipc_instance));
  954. if (res != CS_OK) {
  955. return (res);
  956. }
  957. req_coroipcc_zc_free.header.size = sizeof (mar_req_coroipcc_zc_free_t);
  958. req_coroipcc_zc_free.header.id = ZC_FREE_HEADER;
  959. req_coroipcc_zc_free.map_size = header->map_size;
  960. req_coroipcc_zc_free.server_address = header->server_address;
  961. iovec.iov_base = (void *)&req_coroipcc_zc_free;
  962. iovec.iov_len = sizeof (mar_req_coroipcc_zc_free_t);
  963. res = coroipcc_msg_send_reply_receive (
  964. handle,
  965. &iovec,
  966. 1,
  967. &res_coroipcs_zc_free,
  968. sizeof (coroipc_response_header_t));
  969. munmap ((void *)header, header->map_size);
  970. hdb_handle_put (&ipc_hdb, handle);
  971. return (res);
  972. }
  973. cs_error_t
  974. coroipcc_zcb_msg_send_reply_receive (
  975. hdb_handle_t handle,
  976. void *msg,
  977. void *res_msg,
  978. size_t res_len)
  979. {
  980. struct ipc_instance *ipc_instance;
  981. mar_req_coroipcc_zc_execute_t req_coroipcc_zc_execute;
  982. struct coroipcs_zc_header *hdr;
  983. struct iovec iovec;
  984. cs_error_t res;
  985. res = hdb_error_to_cs (hdb_handle_get (&ipc_hdb, handle, (void **)&ipc_instance));
  986. if (res != CS_OK) {
  987. return (res);
  988. }
  989. hdr = (struct coroipcs_zc_header *)(((char *)msg) - sizeof (struct coroipcs_zc_header));
  990. req_coroipcc_zc_execute.header.size = sizeof (mar_req_coroipcc_zc_execute_t);
  991. req_coroipcc_zc_execute.header.id = ZC_EXECUTE_HEADER;
  992. req_coroipcc_zc_execute.server_address = hdr->server_address;
  993. iovec.iov_base = (void *)&req_coroipcc_zc_execute;
  994. iovec.iov_len = sizeof (mar_req_coroipcc_zc_execute_t);
  995. res = coroipcc_msg_send_reply_receive (
  996. handle,
  997. &iovec,
  998. 1,
  999. res_msg,
  1000. res_len);
  1001. hdb_handle_put (&ipc_hdb, handle);
  1002. return (res);
  1003. }