uclient-utils: Handle memory allocation failure for url file name
[project/uclient.git] / uclient.c
index 2ac052d..1137168 100644 (file)
--- a/uclient.c
+++ b/uclient.c
  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  */
  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  */
+#include <arpa/inet.h>
 #include <libubox/ustream-ssl.h>
 #include "uclient.h"
 #include "uclient-utils.h"
 #include "uclient-backend.h"
 
 #include <libubox/ustream-ssl.h>
 #include "uclient.h"
 #include "uclient-utils.h"
 #include "uclient-backend.h"
 
-struct uclient_url __hidden *
-uclient_get_url(const char *url_str, const char *auth_str)
+char *uclient_get_addr(char *dest, int *port, union uclient_addr *a)
 {
 {
-       static const struct uclient_backend *backends[] = {
-               &uclient_backend_http,
-       };
+       int portval;
+       void *ptr;
 
 
-       const struct uclient_backend *backend;
-       const char * const *prefix = NULL;
-       struct uclient_url *url;
-       const char *location;
-       char *host_buf, *uri_buf, *auth_buf, *next;
-       int i, host_len;
-
-       for (i = 0; i < ARRAY_SIZE(backends); i++) {
-               int prefix_len = 0;
-
-               for (prefix = backends[i]->prefix; *prefix; prefix++) {
-                       prefix_len = strlen(*prefix);
-
-                       if (!strncmp(url_str, *prefix, prefix_len))
-                               break;
-               }
-
-               if (!*prefix)
-                       continue;
-
-               url_str += prefix_len;
-               backend = backends[i];
+       switch(a->sa.sa_family) {
+       case AF_INET:
+               ptr = &a->sin.sin_addr;
+               portval = a->sin.sin_port;
                break;
                break;
+       case AF_INET6:
+               ptr = &a->sin6.sin6_addr;
+               portval = a->sin6.sin6_port;
+               break;
+       default:
+               return strcpy(dest, "Unknown");
        }
 
        }
 
-       if (!*prefix)
-               return NULL;
+       inet_ntop(a->sa.sa_family, ptr, dest, INET6_ADDRSTRLEN);
+       if (port)
+               *port = ntohs(portval);
 
 
-       next = strchr(url_str, '/');
-       if (next) {
-               location = next;
-               host_len = next - url_str;
-       } else {
-               location = "/";
-               host_len = strlen(url_str);
-       }
+       return dest;
+}
+
+static struct uclient_url *
+__uclient_get_url(const struct uclient_backend *backend,
+                 const char *host, int host_len,
+                 const char *location, const char *auth_str)
+{
+       struct uclient_url *url;
+       char *host_buf, *uri_buf, *auth_buf, *next;
 
        url = calloc_a(sizeof(*url),
                &host_buf, host_len + 1,
                &uri_buf, strlen(location) + 1,
                &auth_buf, auth_str ? strlen(auth_str) + 1 : 0);
 
 
        url = calloc_a(sizeof(*url),
                &host_buf, host_len + 1,
                &uri_buf, strlen(location) + 1,
                &auth_buf, auth_str ? strlen(auth_str) + 1 : 0);
 
+       if (!url)
+               return NULL;
+
        url->backend = backend;
        url->location = strcpy(uri_buf, location);
        url->backend = backend;
        url->location = strcpy(uri_buf, location);
-       url->prefix = prefix - backend->prefix;
-
-       url->host = strncpy(host_buf, url_str, host_len);
+       if (host)
+               url->host = strncpy(host_buf, host, host_len);
 
        next = strchr(host_buf, '@');
        if (next) {
 
        next = strchr(host_buf, '@');
        if (next) {
@@ -101,8 +93,10 @@ uclient_get_url(const char *url_str, const char *auth_str)
                        url->port = next + 1;
        } else {
                next = strrchr(url->host, ':');
                        url->port = next + 1;
        } else {
                next = strrchr(url->host, ':');
-               if (next)
+               if (next) {
+                       *next = 0;
                        url->port = next + 1;
                        url->port = next + 1;
+               }
        }
 
        return url;
        }
 
        return url;
@@ -112,6 +106,132 @@ free:
        return NULL;
 }
 
        return NULL;
 }
 
+static const char *
+uclient_split_host(const char *base, int *host_len)
+{
+       char *next, *location;
+
+       next = strchr(base, '/');
+       if (next) {
+               location = next;
+               *host_len = next - base;
+       } else {
+               location = "/";
+               *host_len = strlen(base);
+       }
+
+       return location;
+}
+
+struct uclient_url __hidden *
+uclient_get_url_location(struct uclient_url *url, const char *location)
+{
+       struct uclient_url *new_url;
+       char *host_buf, *uri_buf, *auth_buf, *port_buf;
+       int host_len = strlen(url->host) + 1;
+       int auth_len = url->auth ? strlen(url->auth) + 1 : 0;
+       int port_len = url->port ? strlen(url->port) + 1 : 0;
+       int uri_len;
+
+       if (strstr(location, "://"))
+               return uclient_get_url(location, url->auth);
+
+       if (location[0] == '/')
+               uri_len = strlen(location) + 1;
+       else
+               uri_len = strlen(url->location) + strlen(location) + 2;
+
+       new_url = calloc_a(sizeof(*url),
+               &host_buf, host_len,
+               &port_buf, port_len,
+               &uri_buf, uri_len,
+               &auth_buf, auth_len);
+
+       if (!new_url)
+               return NULL;
+
+       new_url->backend = url->backend;
+       new_url->prefix = url->prefix;
+       new_url->host = strcpy(host_buf, url->host);
+       if (url->port)
+               new_url->port = strcpy(port_buf, url->port);
+       if (url->auth)
+               new_url->auth = strcpy(auth_buf, url->auth);
+
+       new_url->location = uri_buf;
+       if (location[0] == '/')
+               strcpy(uri_buf, location);
+       else {
+               int len = strcspn(url->location, "?#");
+               char *buf = uri_buf;
+
+               memcpy(buf, url->location, len);
+               if (buf[len - 1] != '/') {
+                       buf[len] = '/';
+                       len++;
+               }
+
+               buf += len;
+               strcpy(buf, location);
+       }
+
+       return new_url;
+}
+
+struct uclient_url __hidden *
+uclient_get_url(const char *url_str, const char *auth_str)
+{
+       static const struct uclient_backend *backends[] = {
+               &uclient_backend_http,
+       };
+
+       const struct uclient_backend *backend;
+       const char * const *prefix = NULL;
+       struct uclient_url *url;
+       const char *location;
+       int host_len;
+       int i;
+
+       for (i = 0; i < ARRAY_SIZE(backends); i++) {
+               int prefix_len = 0;
+
+               for (prefix = backends[i]->prefix; *prefix; prefix++) {
+                       prefix_len = strlen(*prefix);
+
+                       if (!strncmp(url_str, *prefix, prefix_len))
+                               break;
+               }
+
+               if (!*prefix)
+                       continue;
+
+               url_str += prefix_len;
+               backend = backends[i];
+               break;
+       }
+
+       if (!*prefix)
+               return NULL;
+
+       location = uclient_split_host(url_str, &host_len);
+       url = __uclient_get_url(backend, url_str, host_len, location, auth_str);
+       if (!url)
+               return NULL;
+
+       url->prefix = prefix - backend->prefix;
+       return url;
+}
+
+static void uclient_connection_timeout(struct uloop_timeout *timeout)
+{
+       struct uclient *cl = container_of(timeout, struct uclient, connection_timeout);
+
+       if (cl->backend->disconnect)
+               cl->backend->disconnect(cl);
+
+       uclient_backend_set_error(cl, UCLIENT_ERROR_TIMEDOUT);
+}
+
 struct uclient *uclient_new(const char *url_str, const char *auth_str, const struct uclient_cb *cb)
 {
        struct uclient *cl;
 struct uclient *uclient_new(const char *url_str, const char *auth_str, const struct uclient_cb *cb)
 {
        struct uclient *cl;
@@ -128,10 +248,42 @@ struct uclient *uclient_new(const char *url_str, const char *auth_str, const str
        cl->backend = url->backend;
        cl->cb = cb;
        cl->url = url;
        cl->backend = url->backend;
        cl->cb = cb;
        cl->url = url;
+       cl->timeout_msecs = UCLIENT_DEFAULT_TIMEOUT_MS;
+       cl->connection_timeout.cb = uclient_connection_timeout;
 
        return cl;
 }
 
 
        return cl;
 }
 
+int uclient_set_proxy_url(struct uclient *cl, const char *url_str, const char *auth_str)
+{
+       const struct uclient_backend *backend = cl->backend;
+       struct uclient_url *url;
+       int host_len;
+       char *next, *host;
+
+       if (!backend->update_proxy_url)
+               return -1;
+
+       next = strstr(url_str, "://");
+       if (!next)
+               return -1;
+
+       host = next + 3;
+       uclient_split_host(host, &host_len);
+
+       url = __uclient_get_url(NULL, host, host_len, url_str, auth_str);
+       if (!url)
+               return -1;
+
+       free(cl->proxy_url);
+       cl->proxy_url = url;
+
+       if (backend->update_proxy_url)
+               backend->update_proxy_url(cl);
+
+       return 0;
+}
+
 int uclient_set_url(struct uclient *cl, const char *url_str, const char *auth_str)
 {
        const struct uclient_backend *backend = cl->backend;
 int uclient_set_url(struct uclient *cl, const char *url_str, const char *auth_str)
 {
        const struct uclient_backend *backend = cl->backend;
@@ -141,8 +293,13 @@ int uclient_set_url(struct uclient *cl, const char *url_str, const char *auth_st
        if (!url)
                return -1;
 
        if (!url)
                return -1;
 
-       if (url->backend != cl->backend)
+       if (url->backend != cl->backend) {
+               free(url);
                return -1;
                return -1;
+       }
+
+       free(cl->proxy_url);
+       cl->proxy_url = NULL;
 
        free(cl->url);
        cl->url = url;
 
        free(cl->url);
        cl->url = url;
@@ -153,6 +310,16 @@ int uclient_set_url(struct uclient *cl, const char *url_str, const char *auth_st
        return 0;
 }
 
        return 0;
 }
 
+int uclient_set_timeout(struct uclient *cl, int msecs)
+{
+       if (msecs <= 0)
+               return -EINVAL;
+
+       cl->timeout_msecs = msecs;
+
+       return 0;
+}
+
 int uclient_connect(struct uclient *cl)
 {
        return cl->backend->connect(cl);
 int uclient_connect(struct uclient *cl)
 {
        return cl->backend->connect(cl);
@@ -170,7 +337,7 @@ void uclient_free(struct uclient *cl)
        free(url);
 }
 
        free(url);
 }
 
-int uclient_write(struct uclient *cl, char *buf, int len)
+int uclient_write(struct uclient *cl, const char *buf, int len)
 {
        if (!cl->backend->write)
                return -1;
 {
        if (!cl->backend->write)
                return -1;
@@ -180,10 +347,18 @@ int uclient_write(struct uclient *cl, char *buf, int len)
 
 int uclient_request(struct uclient *cl)
 {
 
 int uclient_request(struct uclient *cl)
 {
+       int err;
+
        if (!cl->backend->request)
                return -1;
 
        if (!cl->backend->request)
                return -1;
 
-       return cl->backend->request(cl);
+       err = cl->backend->request(cl);
+       if (err)
+               return err;
+
+       uloop_timeout_set(&cl->connection_timeout, cl->timeout_msecs);
+
+       return 0;
 }
 
 int uclient_read(struct uclient *cl, char *buf, int len)
 }
 
 int uclient_read(struct uclient *cl, char *buf, int len)
@@ -194,6 +369,16 @@ int uclient_read(struct uclient *cl, char *buf, int len)
        return cl->backend->read(cl, buf, len);
 }
 
        return cl->backend->read(cl, buf, len);
 }
 
+void uclient_disconnect(struct uclient *cl)
+{
+       uloop_timeout_cancel(&cl->connection_timeout);
+
+       if (!cl->backend->disconnect)
+               return;
+
+       cl->backend->disconnect(cl);
+}
+
 static void __uclient_backend_change_state(struct uloop_timeout *timeout)
 {
        struct uclient *cl = container_of(timeout, struct uclient, timeout);
 static void __uclient_backend_change_state(struct uloop_timeout *timeout)
 {
        struct uclient *cl = container_of(timeout, struct uclient, timeout);
@@ -215,6 +400,7 @@ void __hidden uclient_backend_set_error(struct uclient *cl, int code)
        if (cl->error_code)
                return;
 
        if (cl->error_code)
                return;
 
+       uloop_timeout_cancel(&cl->connection_timeout);
        cl->error_code = code;
        uclient_backend_change_state(cl);
 }
        cl->error_code = code;
        uclient_backend_change_state(cl);
 }
@@ -224,12 +410,14 @@ void __hidden uclient_backend_set_eof(struct uclient *cl)
        if (cl->eof || cl->error_code)
                return;
 
        if (cl->eof || cl->error_code)
                return;
 
+       uloop_timeout_cancel(&cl->connection_timeout);
        cl->eof = true;
        uclient_backend_change_state(cl);
 }
 
 void __hidden uclient_backend_reset_state(struct uclient *cl)
 {
        cl->eof = true;
        uclient_backend_change_state(cl);
 }
 
 void __hidden uclient_backend_reset_state(struct uclient *cl)
 {
+       cl->data_eof = false;
        cl->eof = false;
        cl->error_code = 0;
        uloop_timeout_cancel(&cl->timeout);
        cl->eof = false;
        cl->error_code = 0;
        uloop_timeout_cancel(&cl->timeout);