example: handle uclient connection errors
[project/uclient.git] / uclient.c
index 2e3d3ae..fb45069 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.
  */
+#include <arpa/inet.h>
 #include <libubox/ustream-ssl.h>
 #include "uclient.h"
 #include "uclient-utils.h"
 #include "uclient-backend.h"
 
+char *uclient_get_addr(char *dest, int *port, union uclient_addr *a)
+{
+       int portval;
+       void *ptr;
+
+       switch(a->sa.sa_family) {
+       case AF_INET:
+               ptr = &a->sin.sin_addr;
+               portval = a->sin.sin_port;
+               break;
+       case AF_INET6:
+               ptr = &a->sin6.sin6_addr;
+               portval = a->sin6.sin6_port;
+               break;
+       default:
+               return strcpy(dest, "Unknown");
+       }
+
+       inet_ntop(a->sa.sa_family, ptr, dest, INET6_ADDRSTRLEN);
+       if (port)
+               *port = ntohs(portval);
+
+       return dest;
+}
+
+
 struct uclient_url __hidden *
 uclient_get_url(const char *url_str, const char *auth_str)
 {
@@ -101,8 +128,10 @@ uclient_get_url(const char *url_str, const char *auth_str)
                        url->port = next + 1;
        } else {
                next = strrchr(url->host, ':');
-               if (next)
+               if (next) {
+                       *next = 0;
                        url->port = next + 1;
+               }
        }
 
        return url;
@@ -132,27 +161,30 @@ struct uclient *uclient_new(const char *url_str, const char *auth_str, const str
        return cl;
 }
 
-int uclient_connect_url(struct uclient *cl, const char *url_str)
+int uclient_set_url(struct uclient *cl, const char *url_str, const char *auth_str)
 {
-       struct uclient_url *url = cl->url;
        const struct uclient_backend *backend = cl->backend;
+       struct uclient_url *url = cl->url;
 
-       if (url_str) {
-               url = uclient_get_url(url_str, NULL);
-               if (!url)
-                       return -1;
+       url = uclient_get_url(url_str, auth_str);
+       if (!url)
+               return -1;
 
-               if (url->backend != cl->backend)
-                       return -1;
+       if (url->backend != cl->backend)
+               return -1;
 
-               free(cl->url);
-               cl->url = url;
+       free(cl->url);
+       cl->url = url;
 
-               if (backend->update_url)
-                       backend->update_url(cl);
-       }
+       if (backend->update_url)
+               backend->update_url(cl);
 
-       return backend->connect(cl);
+       return 0;
+}
+
+int uclient_connect(struct uclient *cl)
+{
+       return cl->backend->connect(cl);
 }
 
 void uclient_free(struct uclient *cl)
@@ -191,6 +223,14 @@ int uclient_read(struct uclient *cl, char *buf, int len)
        return cl->backend->read(cl, buf, len);
 }
 
+void uclient_disconnect(struct uclient *cl)
+{
+       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);