2 *------------------------------------------------------------------
5 * Copyright (c) 2009 Cisco and/or its affiliates.
6 * Licensed under the Apache License, Version 2.0 (the "License");
7 * you may not use this file except in compliance with the License.
8 * You may obtain a copy of the License at:
10 * http://www.apache.org/licenses/LICENSE-2.0
12 * Unless required by applicable law or agreed to in writing, software
13 * distributed under the License is distributed on an "AS IS" BASIS,
14 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 * See the License for the specific language governing permissions and
16 * limitations under the License.
17 *------------------------------------------------------------------
20 #include <sys/types.h>
21 #include <sys/socket.h>
23 #include <netinet/in.h>
24 #include <sys/ioctl.h>
25 #include <vppinfra/byte_order.h>
31 #include <vlibmemory/api.h>
32 #include <vlibsocket/api.h>
34 #include <vlibsocket/vl_socket_msg_enum.h>
36 #define vl_typedefs /* define message structures */
37 #include <vlibsocket/vl_socket_api_h.h>
40 /* instantiate all the print functions we know about */
41 #define vl_print(handle, ...) vlib_cli_output (handle, __VA_ARGS__)
43 #include <vlibsocket/vl_socket_api_h.h>
46 /* instantiate all the endian swap functions we know about */
48 #include <vlibsocket/vl_socket_api_h.h>
52 vl_api_sockclnt_create_reply_t_handler (vl_api_sockclnt_create_reply_t * mp)
54 vl_api_registration_t *rp = socket_main.current_rp;
56 rp->server_handle = mp->handle;
57 rp->server_index = mp->index;
61 vl_api_sockclnt_delete_reply_t_handler (vl_api_sockclnt_delete_reply_t * mp)
63 unix_main_t *um = &unix_main;
64 unix_file_t *uf = socket_main.current_uf;
65 vl_api_registration_t *rp = socket_main.current_rp;
67 unix_file_del (um, uf);
68 vl_free_socket_registration_index (rp->vl_api_registration_pool_index);
72 sockclnt_open_index (char *client_name, char *hostname, int port)
74 vl_api_registration_t *rp;
75 unix_main_t *um = &unix_main;
76 unix_file_t template = { 0 };
80 struct sockaddr_in serv_addr;
81 struct hostent *server;
82 vl_api_sockclnt_create_t *mp;
85 server = gethostbyname (hostname);
88 clib_warning ("Couldn't translate server name %s", hostname);
92 /* Set up non-blocking server socket on CLIENT_API_SERVER_PORT */
93 sockfd = socket (AF_INET, SOCK_STREAM, 0);
97 clib_unix_warning ("socket");
101 bzero ((char *) &serv_addr, sizeof (serv_addr));
102 serv_addr.sin_family = AF_INET;
103 bcopy ((char *) server->h_addr,
104 (char *) &serv_addr.sin_addr.s_addr, server->h_length);
105 serv_addr.sin_port = htons (port);
107 if (connect (sockfd, (const void *) &serv_addr, sizeof (serv_addr)) < 0)
109 clib_unix_warning ("Connect failure to (%s, %d)", hostname, port);
114 rv = ioctl (sockfd, FIONBIO, &one);
117 clib_unix_warning ("FIONBIO");
122 pool_get (socket_main.registration_pool, rp);
123 memset (rp, 0, sizeof (*rp));
124 rp->registration_type = REGISTRATION_TYPE_SOCKET_CLIENT;
125 rp->vl_api_registration_pool_index = rp - socket_main.registration_pool;
127 template.read_function = vl_socket_read_ready;
128 template.write_function = vl_socket_write_ready;
129 template.file_descriptor = sockfd;
130 template.private_data = rp - socket_main.registration_pool;
132 rp->unix_file_index = unix_file_add (um, &template);
133 rp->name = format (0, "%s:%d", hostname, port);
135 mp = vl_msg_api_alloc (sizeof (*mp));
136 mp->_vl_msg_id = ntohs (VL_API_SOCKCLNT_CREATE);
137 mp->context = rp - socket_main.registration_pool;
139 if (gethostname (my_hostname, sizeof (my_hostname)) < 0)
141 clib_unix_warning ("gethostname");
142 strncpy (my_hostname, "unknown!", sizeof (my_hostname) - 1);
144 strncpy ((char *) mp->name, my_hostname, sizeof (mp->name) - 1);
146 vl_msg_api_send (rp, (u8 *) mp);
147 return rp - socket_main.registration_pool;
151 sockclnt_close_index (u32 index)
153 vl_api_sockclnt_delete_t *mp;
154 vl_api_registration_t *rp;
156 /* Don't crash / assert if fed garbage */
157 if (pool_is_free_index (socket_main.registration_pool, index))
159 clib_warning ("registration_pool index %d already free", index);
162 rp = pool_elt_at_index (socket_main.registration_pool, index);
164 mp = vl_msg_api_alloc (sizeof (*mp));
165 mp->_vl_msg_id = ntohs (VL_API_SOCKCLNT_DELETE);
166 mp->handle = rp->server_handle;
167 mp->index = rp->server_index;
168 vl_msg_api_send (rp, (u8 *) mp);
171 vl_api_registration_t *
172 sockclnt_get_registration (u32 index)
174 return pool_elt_at_index (socket_main.registration_pool, index);
178 * Both rx and tx msgs MUST be initialized, or we'll have
179 * precisely no idea how many bytes to write into the API trace...
181 #define foreach_sockclnt_api_msg \
182 _(SOCKCLNT_CREATE_REPLY, sockclnt_create_reply) \
183 _(SOCKCLNT_DELETE_REPLY, sockclnt_delete_reply)
186 static clib_error_t *
187 sockclnt_vlib_api_init (vlib_main_t * vm)
190 vl_msg_api_set_handlers(VL_API_##N, #n, \
191 vl_api_##n##_t_handler, \
193 vl_api_##n##_t_endian, \
194 vl_api_##n##_t_print, \
195 sizeof(vl_api_##n##_t), 1);
196 foreach_sockclnt_api_msg;
201 VLIB_API_INIT_FUNCTION (sockclnt_vlib_api_init);
204 * fd.io coding-style-patch-verification: ON
207 * eval: (c-set-style "gnu")