coroipcc.c 27 KB

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