1 /* 2 * CDDL HEADER START 3 * 4 * The contents of this file are subject to the terms of the 5 * Common Development and Distribution License (the "License"). 6 * You may not use this file except in compliance with the License. 7 * 8 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE 9 * or http://www.opensolaris.org/os/licensing. 10 * See the License for the specific language governing permissions 11 * and limitations under the License. 12 * 13 * When distributing Covered Code, include this CDDL HEADER in each 14 * file and include the License file at usr/src/OPENSOLARIS.LICENSE. 15 * If applicable, add the following below this CDDL HEADER, with the 16 * fields enclosed by brackets "[]" replaced with your own identifying 17 * information: Portions Copyright [yyyy] [name of copyright owner] 18 * 19 * CDDL HEADER END 20 */ 21 /* 22 * Copyright 2009 Sun Microsystems, Inc. All rights reserved. 23 * Use is subject to license terms. 24 */ 25 26 #include <stdio.h> 27 #include <stdlib.h> 28 #include <string.h> 29 #include <strings.h> 30 #include <sys/types.h> 31 #include <errno.h> 32 #include <syslog.h> 33 #include <unistd.h> 34 #include <sys/types.h> 35 #include <sys/socket.h> 36 #include <sys/time.h> 37 #include <netinet/in.h> 38 #include <arpa/inet.h> 39 #include <netdb.h> 40 #include <sys/stat.h> 41 #include <sys/sdt.h> 42 #include <signal.h> 43 #include <fcntl.h> 44 #include <libstmfproxy.h> 45 46 /* 47 * NOTE: 48 * This is demo code to be used with the existing demo proxy daemon 49 * svc-stmfproxy in /usr/demo/comstar. 50 */ 51 52 struct _s_handle { 53 int sockfd; 54 }; 55 56 typedef struct _s_handle s_handle_t; 57 58 static ssize_t 59 pt_socket_recv(void *handle, void *buf, size_t len) 60 { 61 s_handle_t *sh = handle; 62 63 return (recv(sh->sockfd, buf, len, MSG_WAITALL)); 64 } 65 66 static ssize_t 67 pt_socket_send(void *handle, void *buf, size_t len) 68 { 69 s_handle_t *sh = handle; 70 71 return (send(sh->sockfd, buf, len, 0)); 72 } 73 74 static void * 75 pt_socket_connect(int server_node, char *server) 76 { 77 int sfd, new_sfd; 78 s_handle_t *sh = NULL; 79 int on = 1; 80 struct sockaddr_in cli_addr, serv_addr; 81 struct sockaddr_in sin; 82 int cliLen = sizeof (cli_addr); 83 84 if ((sfd = socket(AF_INET, SOCK_STREAM, 0)) <= 0) { 85 syslog(LOG_DAEMON|LOG_WARNING, 86 "socket() call failed: %d", errno); 87 return (NULL); 88 } 89 90 if (server_node) { 91 92 if (setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &on, 93 sizeof (on)) < 0) { 94 syslog(LOG_DAEMON|LOG_WARNING, 95 "setsockopt() failed: %d", errno); 96 goto serv_out; 97 } 98 99 bzero(&serv_addr, sizeof (serv_addr)); 100 serv_addr.sin_family = AF_INET; 101 serv_addr.sin_addr.s_addr = htonl(INADDR_ANY); 102 /* XXX get from smf? */ 103 serv_addr.sin_port = htons(6543); 104 105 if (bind(sfd, (struct sockaddr *)&serv_addr, 106 sizeof (serv_addr)) < 0) { 107 syslog(LOG_DAEMON|LOG_WARNING, "bind() call failed: %d", 108 errno); 109 goto serv_out; 110 } 111 112 (void) listen(sfd, 5); 113 114 new_sfd = accept(sfd, (struct sockaddr *)&cli_addr, &cliLen); 115 116 if (new_sfd < 0) { 117 syslog(LOG_DAEMON|LOG_WARNING, "accept failed: %d", 118 errno); 119 goto serv_out; 120 } 121 sh = malloc(sizeof (*sh)); 122 sh->sockfd = new_sfd; 123 serv_out: 124 close(sfd); 125 } else { 126 struct hostent *hp; 127 128 /* 129 * Assume IP dot notation or if that fails, gethostbyname() 130 * If that fails, return 131 */ 132 if ((inet_aton(server, &sin.sin_addr)) == 0) { 133 if ((hp = gethostbyname(server)) != NULL) { 134 memcpy(&sin.sin_addr.s_addr, hp->h_addr, 135 hp->h_length); 136 } else { 137 syslog(LOG_DAEMON|LOG_CRIT, 138 "Cannot get IP address for %s", server); 139 (void) close(sfd); 140 return (NULL); 141 } 142 } else { 143 fprintf(stderr, 144 "Sorry, cannot use ip address format\n"); 145 (void) close(sfd); 146 return (NULL); 147 } 148 sin.sin_family = AF_INET; 149 /* XXX pass in from smf */ 150 sin.sin_port = htons(6543); 151 152 while (connect(sfd, (struct sockaddr *)&sin, 153 sizeof (sin)) < 0) { 154 close(sfd); 155 if (errno == ECONNREFUSED) { 156 /* get a fresh socket and retry */ 157 sfd = socket(AF_INET, SOCK_STREAM, 0); 158 if (sfd < 0) { 159 syslog(LOG_DAEMON|LOG_WARNING, 160 "socket() call failed: %d", errno); 161 return (NULL); 162 } 163 sleep(2); 164 } else { 165 syslog(LOG_DAEMON|LOG_CRIT, 166 "Cannot connect %s - %d", server, errno); 167 return (NULL); 168 } 169 } 170 sh = malloc(sizeof (*sh)); 171 sh->sockfd = sfd; 172 } 173 return (sh); 174 } 175 176 pt_ops_t pt_socket_ops = { 177 pt_socket_connect, 178 pt_socket_send, 179 pt_socket_recv 180 }; 181 182 int 183 stmf_proxy_transport_init(char *transport, pt_ops_t **pt_ops) 184 { 185 if (strcmp(transport, "sockets") == 0) { 186 *pt_ops = &pt_socket_ops; 187 return (0); 188 } else { 189 return (-1); 190 } 191 } 192