uic.c 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
  1. /*
  2. * Copyright (C) 2006 Steven Dake (sdake@mvista.com)
  3. *
  4. * This software licensed under BSD license, the text of which follows:
  5. *
  6. * Redistribution and use in source and binary forms, with or without
  7. * modification, are permitted provided that the following conditions are met:
  8. *
  9. * - Redistributions of source code must retain the above copyright notice,
  10. * this list of conditions and the following disclaimer.
  11. * - Redistributions in binary form must reproduce the above copyright notice,
  12. * this list of conditions and the following disclaimer in the documentation
  13. * and/or other materials provided with the distribution.
  14. * - Neither the name of the MontaVista Software, Inc. nor the names of its
  15. * contributors may be used to endorse or promote products derived from this
  16. * software without specific prior written permission.
  17. *
  18. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  19. * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  20. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  21. * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
  22. * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  23. * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  24. * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  25. * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  26. * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  27. * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
  28. * THE POSSIBILITY OF SUCH DAMAGE.
  29. */
  30. #include <sys/uio.h>
  31. #include <sys/mman.h>
  32. #include <sys/types.h>
  33. #include <sys/socket.h>
  34. #include <sys/un.h>
  35. #include <sys/time.h>
  36. #include <sys/resource.h>
  37. #include <netinet/in.h>
  38. #include <arpa/inet.h>
  39. #include <unistd.h>
  40. #include <fcntl.h>
  41. #include <stdlib.h>
  42. #include <stdio.h>
  43. #include <errno.h>
  44. #include <signal.h>
  45. #include <sched.h>
  46. #include <time.h>
  47. #include <pthread.h>
  48. #include <sys/poll.h>
  49. #include <string.h>
  50. #if defined(OPENAIS_LINUX)
  51. /* SUN_LEN is broken for abstract namespace
  52. */
  53. #define AIS_SUN_LEN(a) sizeof(*(a))
  54. static char *socketname = "lcr.socket";
  55. #else
  56. #define AIS_SUN_LEN(a) SUN_LEN(a)
  57. static char *socketname = "/var/run/lcr.socket";
  58. #endif
  59. int uic_connect (int *fd)
  60. {
  61. int res;
  62. struct sockaddr_un addr;
  63. memset (&addr, 0, sizeof (struct sockaddr_un));
  64. #if defined(OPENAIS_BSD) || defined(OPENAIS_DARWIN)
  65. addr.sun_len = sizeof(struct sockaddr_un);
  66. #endif
  67. addr.sun_family = PF_UNIX;
  68. #if defined(OPENAIS_LINUX)
  69. strcpy (addr.sun_path + 1, socketname);
  70. #else
  71. strcpy (addr.sun_path, socketname);
  72. #endif
  73. *fd = socket (PF_UNIX, SOCK_STREAM, 0);
  74. if (*fd == -1) {
  75. return -errno;
  76. }
  77. res = connect (*fd, (struct sockaddr *)&addr, AIS_SUN_LEN(&addr));
  78. if (res == -1) {
  79. return -errno;
  80. }
  81. return 0;
  82. }
  83. struct req_msg {
  84. int len;
  85. char msg[0];
  86. };
  87. int uic_msg_send (int fd, char *msg)
  88. {
  89. struct msghdr msg_send;
  90. struct iovec iov_send[2];
  91. struct req_msg req_msg;
  92. int res;
  93. req_msg.len = strlen (msg) + 1;
  94. iov_send[0].iov_base = (void *)&req_msg;
  95. iov_send[0].iov_len = sizeof (struct req_msg);
  96. iov_send[1].iov_base = (void *)msg;
  97. iov_send[1].iov_len = req_msg.len;
  98. msg_send.msg_iov = iov_send;
  99. msg_send.msg_iovlen = 2;
  100. msg_send.msg_name = 0;
  101. msg_send.msg_namelen = 0;
  102. msg_send.msg_control = 0;
  103. msg_send.msg_controllen = 0;
  104. msg_send.msg_flags = 0;
  105. retry_send:
  106. res = sendmsg (fd, &msg_send, 0);
  107. if (res == -1 && errno == EINTR) {
  108. goto retry_send;
  109. }
  110. if (res == -1) {
  111. res = -errno;
  112. }
  113. return (res);
  114. }
  115. int main (void)
  116. {
  117. int client_fd;
  118. int res;
  119. res = uic_connect (&client_fd);
  120. if (res != 0) {
  121. printf ("Couldn't connect to live replacement service\n");
  122. }
  123. uic_msg_send (client_fd, "livereplace ckpt version 2");
  124. return 0;
  125. }