2 * uclient - ustream based protocol client library
4 * Copyright (C) 2014 Felix Fietkau <nbd@openwrt.org>
6 * Permission to use, copy, modify, and/or distribute this software for any
7 * purpose with or without fee is hereby granted, provided that the above
8 * copyright notice and this permission notice appear in all copies.
10 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
18 #include <arpa/inet.h>
19 #include <libubox/ustream-ssl.h>
21 #include "uclient-utils.h"
22 #include "uclient-backend.h"
24 char *uclient_get_addr(char *dest, int *port, union uclient_addr *a)
29 switch(a->sa.sa_family) {
31 ptr = &a->sin.sin_addr;
32 portval = a->sin.sin_port;
35 ptr = &a->sin6.sin6_addr;
36 portval = a->sin6.sin6_port;
39 return strcpy(dest, "Unknown");
42 inet_ntop(a->sa.sa_family, ptr, dest, INET6_ADDRSTRLEN);
44 *port = ntohs(portval);
50 struct uclient_url __hidden *
51 uclient_get_url(const char *url_str, const char *auth_str)
53 static const struct uclient_backend *backends[] = {
54 &uclient_backend_http,
57 const struct uclient_backend *backend;
58 const char * const *prefix = NULL;
59 struct uclient_url *url;
61 char *host_buf, *uri_buf, *auth_buf, *next;
64 for (i = 0; i < ARRAY_SIZE(backends); i++) {
67 for (prefix = backends[i]->prefix; *prefix; prefix++) {
68 prefix_len = strlen(*prefix);
70 if (!strncmp(url_str, *prefix, prefix_len))
77 url_str += prefix_len;
78 backend = backends[i];
85 next = strchr(url_str, '/');
88 host_len = next - url_str;
91 host_len = strlen(url_str);
94 url = calloc_a(sizeof(*url),
95 &host_buf, host_len + 1,
96 &uri_buf, strlen(location) + 1,
97 &auth_buf, auth_str ? strlen(auth_str) + 1 : 0);
99 url->backend = backend;
100 url->location = strcpy(uri_buf, location);
101 url->prefix = prefix - backend->prefix;
103 url->host = strncpy(host_buf, url_str, host_len);
105 next = strchr(host_buf, '@');
108 url->host = next + 1;
110 if (uclient_urldecode(host_buf, host_buf, false) < 0)
113 url->auth = host_buf;
116 if (!url->auth && auth_str)
117 url->auth = strcpy(auth_buf, auth_str);
119 /* Literal IPv6 address */
120 if (*url->host == '[') {
122 next = strrchr(url->host, ']');
128 url->port = next + 1;
130 next = strrchr(url->host, ':');
133 url->port = next + 1;
144 struct uclient *uclient_new(const char *url_str, const char *auth_str, const struct uclient_cb *cb)
147 struct uclient_url *url;
149 url = uclient_get_url(url_str, auth_str);
153 cl = url->backend->alloc();
157 cl->backend = url->backend;
164 int uclient_set_url(struct uclient *cl, const char *url_str, const char *auth_str)
166 const struct uclient_backend *backend = cl->backend;
167 struct uclient_url *url = cl->url;
169 url = uclient_get_url(url_str, auth_str);
173 if (url->backend != cl->backend)
179 if (backend->update_url)
180 backend->update_url(cl);
185 int uclient_connect(struct uclient *cl)
187 return cl->backend->connect(cl);
190 void uclient_free(struct uclient *cl)
192 struct uclient_url *url = cl->url;
194 if (cl->backend->free)
195 cl->backend->free(cl);
202 int uclient_write(struct uclient *cl, char *buf, int len)
204 if (!cl->backend->write)
207 return cl->backend->write(cl, buf, len);
210 int uclient_request(struct uclient *cl)
212 if (!cl->backend->request)
215 return cl->backend->request(cl);
218 int uclient_read(struct uclient *cl, char *buf, int len)
220 if (!cl->backend->read)
223 return cl->backend->read(cl, buf, len);
226 static void __uclient_backend_change_state(struct uloop_timeout *timeout)
228 struct uclient *cl = container_of(timeout, struct uclient, timeout);
230 if (cl->error_code && cl->cb->error)
231 cl->cb->error(cl, cl->error_code);
232 else if (cl->eof && cl->cb->data_eof)
233 cl->cb->data_eof(cl);
236 static void uclient_backend_change_state(struct uclient *cl)
238 cl->timeout.cb = __uclient_backend_change_state;
239 uloop_timeout_set(&cl->timeout, 1);
242 void __hidden uclient_backend_set_error(struct uclient *cl, int code)
247 cl->error_code = code;
248 uclient_backend_change_state(cl);
251 void __hidden uclient_backend_set_eof(struct uclient *cl)
253 if (cl->eof || cl->error_code)
257 uclient_backend_change_state(cl);
260 void __hidden uclient_backend_reset_state(struct uclient *cl)
264 uloop_timeout_cancel(&cl->timeout);