Don't have the connection freeing peers -- it's the contexts responsibility.

[radsecproxy.git] / lib / conn.c

diff --git a/lib/conn.c b/lib/conn.c
index 62105e6..924c225 100644 (file)
--- a/lib/conn.c
+++ b/lib/conn.c
@@ -1,6 +1,13 @@
 /* See the file COPYING for licensing information.  */
 
+#if defined HAVE_CONFIG_H
+#include <config.h>
+#endif
+
+#include <string.h>
 #include <assert.h>
+#include <debug.h>
+#include <event2/event.h>
 #include <radsec/radsec.h>
 #include <radsec/radsec-impl.h>
 
@@ -11,37 +18,50 @@ rs_conn_create (struct rs_context *ctx, struct rs_connection **conn,
   struct rs_connection *c;
 
   c = (struct rs_connection *) malloc (sizeof(struct rs_connection));
-  if (c)
+  if (!c)
+    return rs_err_ctx_push_fl (ctx, RSE_NOMEM, __FILE__, __LINE__, NULL);
+
+  memset (c, 0, sizeof(struct rs_connection));
+  c->ctx = ctx;
+  c->fd = -1;
+  if (config)
     {
-      memset (c, 0, sizeof(struct rs_connection));
-      c->ctx = ctx;
-      if (config)
+      struct rs_realm *r = rs_conf_find_realm (ctx, config);
+      if (r)
+       {
+         struct rs_peer *p;
+
+         c->realm = r;
+         c->peers = r->peers;  /* FIXME: Copy instead?  */
+         for (p = c->peers; p; p = p->next)
+           p->conn = c;
+         c->tryagain = r->retries;
+       }
+      else
        {
-         struct rs_realm *r = rs_conf_find_realm (ctx, config);
-         if (r)
-           {
-             struct rs_peer *p;
-
-             c->type = r->type;
-             c->peers = r->peers; /* FIXME: Copy instead?  */
-             for (p = c->peers; p; p = p->next)
-               p->conn = c;
-           }
+         c->realm = rs_malloc (ctx, sizeof (struct rs_realm));
+         if (!c->realm)
+           return rs_err_ctx_push_fl (ctx, RSE_NOMEM, __FILE__, __LINE__,
+                                      NULL);
+         memset (c->realm, 0, sizeof (struct rs_realm));
        }
     }
+
   if (conn)
     *conn = c;
-  return c ? RSE_OK : rs_err_ctx_push (ctx, RSE_NOMEM, NULL);
+  return RSE_OK;
 }
 
 void
 rs_conn_set_type (struct rs_connection *conn, rs_conn_type_t type)
 {
-  conn->type = type;
+  assert (conn);
+  assert (conn->realm);
+  conn->realm->type = type;
 }
 
 
-struct rs_error *
+struct rs_error *         /* FIXME: Return int as all the others?  */
 _rs_resolv (struct evutil_addrinfo **addr, rs_conn_type_t type,
            const char *hostname, const char *service)
 {
@@ -49,7 +69,7 @@ _rs_resolv (struct evutil_addrinfo **addr, rs_conn_type_t type,
   struct evutil_addrinfo hints, *res = NULL;
 
   memset (&hints, 0, sizeof(struct evutil_addrinfo));
-  hints.ai_family = AF_UNSPEC; /* v4 or v6.  */
+  hints.ai_family = AF_INET;   /* IPv4 only.  TODO: Set AF_UNSPEC.  */
   hints.ai_flags = AI_ADDRCONFIG;
   switch (type)
     {
@@ -67,6 +87,8 @@ _rs_resolv (struct evutil_addrinfo **addr, rs_conn_type_t type,
       hints.ai_socktype = SOCK_DGRAM;
       hints.ai_protocol = IPPROTO_UDP;
       break;
+    default:
+      return _rs_err_create (RSE_INVALID_CONN, __FILE__, __LINE__, NULL, NULL);
     }
   err = evutil_getaddrinfo (hostname, service, &hints, &res);
   if (err)
@@ -85,23 +107,42 @@ rs_conn_add_listener (struct rs_connection *conn, rs_conn_type_t type,
                              "%s: NYI", __func__);
 }
 
-void
+
+int
+rs_conn_disconnect (struct rs_connection *conn)
+{
+  int err = 0;
+
+  assert (conn);
+
+  err = evutil_closesocket (conn->fd);
+  conn->fd = -1;
+  return err;
+}
+
+int
 rs_conn_destroy (struct rs_connection *conn)
 {
-  struct rs_peer *p;
+  int err = 0;
 
-#warning "TODO: Disconnect active_peer."
+  assert (conn);
 
-  for (p = conn->peers; p; p = p->next)
+  if (conn->is_connected)
     {
-      if (p->addr)
-       evutil_freeaddrinfo (p->addr);
-      if (p->secret)
-       rs_free (conn->ctx, p->secret);
+      err = rs_conn_disconnect (conn);
+      if (err)
+       return err;
     }
 
+  /* NOTE: conn->realm is owned by context.  */
+  /* NOTE: conn->peers is owned by context.  */
+
+  if (conn->tev)
+    event_free (conn->tev);
   if (conn->evb)
     event_base_free (conn->evb);
+
+  return 0;
 }
 
 int
@@ -111,23 +152,39 @@ rs_conn_set_eventbase (struct rs_connection *conn, struct event_base *eb)
                              "%s: NYI", __func__);
 }
 
-int
+void
 rs_conn_set_callbacks (struct rs_connection *conn, struct rs_conn_callbacks *cb)
 {
-  return rs_err_conn_push_fl (conn, RSE_NOSYS, __FILE__, __LINE__,
-                             "%s: NYI", __func__);
+  assert (conn);
+  conn->user_dispatch_flag = 1;
+  memcpy (&conn->callbacks, cb, sizeof (conn->callbacks));
+}
+
+void
+rs_conn_del_callbacks (struct rs_connection *conn)
+{
+  assert (conn);
+  conn->user_dispatch_flag = 0;
+  memset (&conn->callbacks, 0, sizeof (conn->callbacks));
+}
+
+struct rs_conn_callbacks *
+rs_conn_get_callbacks(struct rs_connection *conn)
+{
+  assert (conn);
+  return &conn->callbacks;
 }
 
 int
-rs_conn_select_server (struct rs_connection *conn, const char *name)
+rs_conn_select_peer (struct rs_connection *conn, const char *name)
 {
   return rs_err_conn_push_fl (conn, RSE_NOSYS, __FILE__, __LINE__,
                              "%s: NYI", __func__);
 }
 
 int
-rs_conn_get_current_server (struct rs_connection *conn, const char *name,
-                           size_t buflen)
+rs_conn_get_current_peer (struct rs_connection *conn, const char *name,
+                         size_t buflen)
 {
   return rs_err_conn_push_fl (conn, RSE_NOSYS, __FILE__, __LINE__,
                              "%s: NYI", __func__);
@@ -137,5 +194,5 @@ int rs_conn_fd (struct rs_connection *conn)
 {
   assert (conn);
   assert (conn->active_peer);
-  return conn->active_peer->fd;
+  return conn->fd;
 }