autotools are dumb, so I'm also adding gnioenumtypes.{c,h}
[cascardo/gnio.git] / gnio / gsocket.c
index 9d50665..0ca9091 100644 (file)
@@ -24,7 +24,9 @@
 #include <config.h>
 #include <glib.h>
 #include <gio/gio.h>
-#include <gio/gasynchelper.h>
+#include <gnio/gnio.h>
+#include "gasynchelper.h"
+#include "gnioenumtypes.h"
 
 #include <string.h>
 #ifndef G_OS_WIN32
@@ -51,22 +53,37 @@ G_DEFINE_TYPE (GSocket, g_socket, G_TYPE_OBJECT);
 enum
 {
   PROP_0,
+  PROP_DOMAIN,
+  PROP_TYPE,
+  PROP_PROTOCOL,
   PROP_FD,
   PROP_BLOCKING,
   PROP_BACKLOG,
+  PROP_REUSE_ADDRESS,
   PROP_LOCAL_ADDRESS,
   PROP_REMOTE_ADDRESS
 };
 
 struct _GSocketPrivate
 {
+  GSocketDomain domain;
+  GSocketType type;
+  gchar *protocol;
   gint fd;
   gboolean blocking;
   gint backlog;
+  gboolean reuse_address;
+  GError *error;
   GSocketAddress *local_address;
   GSocketAddress *remote_address;
 };
 
+static void
+g_socket_constructed (GObject *object)
+{
+
+}
+
 static void
 g_socket_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec)
 {
@@ -74,6 +91,18 @@ g_socket_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec
 
   switch (prop_id)
     {
+      case PROP_DOMAIN:
+        g_value_set_int (value, socket->priv->domain);
+        break;
+
+      case PROP_TYPE:
+        g_value_set_int (value, socket->priv->type);
+        break;
+
+      case PROP_PROTOCOL:
+        g_value_set_string (value, socket->priv->protocol);
+        break;
+
       case PROP_FD:
         g_value_set_int (value, socket->priv->fd);
         break;
@@ -86,6 +115,10 @@ g_socket_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec
         g_value_set_int (value, socket->priv->backlog);
         break;
 
+      case PROP_REUSE_ADDRESS:
+        g_value_set_boolean (value, socket->priv->reuse_address);
+        break;
+
       case PROP_LOCAL_ADDRESS:
         g_value_set_object (value, socket->priv->local_address);
         break;
@@ -118,6 +151,10 @@ g_socket_set_property (GObject *object, guint prop_id, const GValue *value, GPar
         socket->priv->backlog = g_value_get_int (value);
         break;
 
+      case PROP_REUSE_ADDRESS:
+        g_socket_set_reuse_address (socket, g_value_get_boolean (value));
+        break;
+
       default:
         G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
     }
@@ -160,9 +197,37 @@ g_socket_class_init (GSocketClass *klass)
 
   gobject_class->finalize = g_socket_finalize;
   gobject_class->dispose = g_socket_dispose;
+  gobject_class->constructed = g_socket_constructed;
   gobject_class->set_property = g_socket_set_property;
   gobject_class->get_property = g_socket_get_property;
 
+  g_object_class_install_property (gobject_class, PROP_DOMAIN,
+                                   g_param_spec_int ("domain",
+                                                     "file descriptor",
+                                                     "the socket's file descriptor",
+                                                     G_MININT,
+                                                     G_MAXINT,
+                                                     -1,
+                                                     G_PARAM_CONSTRUCT_ONLY | G_PARAM_READWRITE | G_PARAM_STATIC_NAME | G_PARAM_STATIC_BLURB | G_PARAM_STATIC_NICK));
+
+  g_object_class_install_property (gobject_class, PROP_TYPE,
+                                   g_param_spec_int ("fd",
+                                                     "file descriptor",
+                                                     "the socket's file descriptor",
+                                                     G_MININT,
+                                                     G_MAXINT,
+                                                     -1,
+                                                     G_PARAM_CONSTRUCT_ONLY | G_PARAM_READWRITE | G_PARAM_STATIC_NAME | G_PARAM_STATIC_BLURB | G_PARAM_STATIC_NICK));
+
+  g_object_class_install_property (gobject_class, PROP_PROTOCOL,
+                                   g_param_spec_int ("fd",
+                                                     "file descriptor",
+                                                     "the socket's file descriptor",
+                                                     G_MININT,
+                                                     G_MAXINT,
+                                                     -1,
+                                                     G_PARAM_CONSTRUCT_ONLY | G_PARAM_READWRITE | G_PARAM_STATIC_NAME | G_PARAM_STATIC_BLURB | G_PARAM_STATIC_NICK));
+
   g_object_class_install_property (gobject_class, PROP_FD,
                                    g_param_spec_int ("fd",
                                                      "file descriptor",
@@ -188,6 +253,13 @@ g_socket_class_init (GSocketClass *klass)
                                                      10,
                                                      G_PARAM_READWRITE | G_PARAM_STATIC_NAME | G_PARAM_STATIC_BLURB | G_PARAM_STATIC_NICK));
 
+  g_object_class_install_property (gobject_class, PROP_REUSE_ADDRESS,
+                                   g_param_spec_boolean ("reuse-address",
+                                                         "reuse address",
+                                                         "allow reuse of local addresses when binding",
+                                                         FALSE,
+                                                         G_PARAM_READWRITE | G_PARAM_STATIC_NAME | G_PARAM_STATIC_BLURB | G_PARAM_STATIC_NICK));
+
   g_object_class_install_property (gobject_class, PROP_LOCAL_ADDRESS,
                                    g_param_spec_object ("local-address",
                                                         "local address",
@@ -211,13 +283,16 @@ g_socket_init (GSocket *socket)
   socket->priv->fd = -1;
   socket->priv->blocking = TRUE;
   socket->priv->backlog = 10;
+  socket->priv->reuse_address = FALSE;
+  socket->priv->error = NULL;
   socket->priv->remote_address = NULL;
   socket->priv->local_address = NULL;
 }
 
 GSocket *
-g_socket_new (GSocketDomain domain, GSocketType type, const gchar *protocol, GError **error)
+g_socket_new (GSocketDomain domain, GSocketType type, const gchar *protocol)
 {
+  GError *error = NULL;
   static GStaticMutex getprotobyname_mutex = G_STATIC_MUTEX_INIT;
   gint fd, native_domain, native_type, native_protocol;
 
@@ -236,7 +311,7 @@ g_socket_new (GSocketDomain domain, GSocketType type, const gchar *protocol, GEr
         break;
 
       default:
-        g_set_error (error, G_IO_ERROR, G_IO_ERROR_NOT_SUPPORTED, "unsupported socket domain");
+        g_set_error (&error, G_IO_ERROR, G_IO_ERROR_NOT_SUPPORTED, "unsupported socket domain");
         return NULL;
     }
 
@@ -255,7 +330,7 @@ g_socket_new (GSocketDomain domain, GSocketType type, const gchar *protocol, GEr
         break;
 
       default:
-        g_set_error (error, G_IO_ERROR, G_IO_ERROR_NOT_SUPPORTED, "unsupported socket type");
+        g_set_error (&error, G_IO_ERROR, G_IO_ERROR_NOT_SUPPORTED, "unsupported socket type");
         return NULL;
     }
 
@@ -268,7 +343,7 @@ g_socket_new (GSocketDomain domain, GSocketType type, const gchar *protocol, GEr
       if (!(ent = getprotobyname (protocol)))
         {
           g_static_mutex_unlock (&getprotobyname_mutex);
-          g_set_error (error, G_IO_ERROR, G_IO_ERROR_NOT_SUPPORTED, "unsupported socket protocol");
+          g_set_error (&error, G_IO_ERROR, G_IO_ERROR_NOT_SUPPORTED, "unsupported socket protocol");
           return NULL;
         }
       native_protocol = ent->p_proto;
@@ -279,11 +354,13 @@ g_socket_new (GSocketDomain domain, GSocketType type, const gchar *protocol, GEr
 
   if (fd < 0)
     {
-      g_set_error (error, G_IO_ERROR, g_io_error_from_errno (errno), "unable to create socket: %s", g_strerror (errno));
+      g_set_error (&error, G_IO_ERROR, g_io_error_from_errno (errno), "unable to create socket: %s", g_strerror (errno));
       return NULL;
     }
 
-  return G_SOCKET (g_object_new (G_TYPE_SOCKET, "fd", fd, "blocking", TRUE, NULL));
+  g_print ("finishing socket_new\n");
+
+  return G_SOCKET (g_object_new (G_TYPE_SOCKET, "blocking", TRUE, "fd", fd, NULL));
 }
 
 GSocket *
@@ -327,6 +404,28 @@ g_socket_get_blocking (GSocket *socket)
   return socket->priv->blocking;
 }
 
+void
+g_socket_set_reuse_address (GSocket  *socket,
+                            gboolean  reuse)
+{
+  gint value = (gint) reuse;
+
+  g_return_if_fail (G_IS_SOCKET (socket));
+
+  if (setsockopt (socket->priv->fd, SOL_SOCKET, SO_REUSEADDR, (gpointer) &value, sizeof (value)) < 0)
+    g_warning ("error setting reuse address: %s", g_strerror (errno));
+
+  socket->priv->reuse_address = reuse;
+}
+
+gboolean
+g_socket_get_reuse_address (GSocket *socket)
+{
+  g_return_val_if_fail (G_IS_SOCKET (socket), FALSE);
+
+  return socket->priv->reuse_address;
+}
+
 GSocketAddress *
 g_socket_get_local_address (GSocket  *socket,
                             GError  **error)
@@ -369,12 +468,34 @@ g_socket_get_remote_address (GSocket  *socket,
   return (socket->priv->remote_address = g_object_ref_sink (g_socket_address_from_native (buffer, len)));
 }
 
-void
-g_socket_listen (GSocket *socket)
+gboolean
+g_socket_has_error (GSocket  *socket,
+                    GError  **error)
 {
-  g_return_if_fail (G_IS_SOCKET (socket));
+  g_return_val_if_fail (G_IS_SOCKET (socket), FALSE);
 
-  listen (socket->priv->fd, socket->priv->backlog);
+  if (!socket->priv->error)
+    return FALSE;
+
+  return TRUE;
+}
+
+gboolean
+g_socket_listen (GSocket  *socket,
+                 GError  **error)
+{
+  g_return_val_if_fail (G_IS_SOCKET (socket), FALSE);
+
+  if (g_socket_has_error (socket, error))
+    return FALSE;
+
+  if (listen (socket->priv->fd, socket->priv->backlog) < 0)
+    {
+      g_set_error (error, G_IO_ERROR, g_io_error_from_errno (errno), "could not listen: %s", g_strerror (errno));
+      return FALSE;
+    }
+
+  return TRUE;
 }
 
 gboolean
@@ -384,6 +505,9 @@ g_socket_bind (GSocket         *socket,
 {
   g_return_val_if_fail (G_IS_SOCKET (socket) && G_IS_SOCKET_ADDRESS (address), FALSE);
 
+  if (g_socket_has_error (socket, error))
+    return FALSE;
+
   {
     gchar addr[256];
 
@@ -410,6 +534,11 @@ g_socket_accept (GSocket       *socket,
 {
   gint ret;
 
+  g_return_val_if_fail (G_IS_SOCKET (socket), NULL);
+
+  if (g_socket_has_error (socket, error))
+    return NULL;
+
   if ((ret = accept (socket->priv->fd, NULL, 0)) < 0)
     {
       g_set_error (error, G_IO_ERROR, g_io_error_from_errno (errno), "error accepting connection: %s", g_strerror (errno));
@@ -428,6 +557,9 @@ g_socket_connect (GSocket         *socket,
 
   g_return_val_if_fail (G_IS_SOCKET (socket) && G_IS_SOCKET_ADDRESS (address), FALSE);
 
+  if (g_socket_has_error (socket, error))
+    return FALSE;
+
   g_socket_address_to_native (address, buffer);
 
   if (connect (socket->priv->fd, (struct sockaddr *) buffer, g_socket_address_native_size (address)) < 0)