test/swfdec_test_socket.c

   1 /* Swfdec
   2  * Copyright (C) 2007 Benjamin Otte <otte@gnome.org>
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin Street, Fifth Floor,
  17  * Boston, MA  02110-1301  USA
  18  */
  19
  20 #ifdef HAVE_CONFIG_H
  21 #include "config.h"
  22 #endif
  23
  24 #include "swfdec_test_socket.h"
  25 #include "swfdec_test_buffer.h"
  26 #include "swfdec_test_function.h"
  27
  28 static void
  29 swfdec_test_throw (SwfdecAsContext *cx, const char *message, ...)
  30 {
  31   SwfdecAsValue val;
  32
  33   if (!swfdec_as_context_catch (cx, &val)) {
  34     va_list varargs;
  35     char *s;
  36
  37     va_start (varargs, message);
  38     s = g_strdup_vprintf (message, varargs);
  39     va_end (varargs);
  40
  41     /* FIXME: Throw a real object here? */
  42     SWFDEC_AS_VALUE_SET_STRING (&val, swfdec_as_context_give_string (cx, s));
  43   }
  44   swfdec_as_context_throw (cx, &val);
  45 }
  46
  47 /*** SWFDEC_TEST_SOCKET ***/
  48
  49 G_DEFINE_TYPE (SwfdecTestSocket, swfdec_test_socket, SWFDEC_TYPE_AS_OBJECT)
  50
  51 static void
  52 swfdec_test_socket_do_close (SwfdecTestSocket *sock, gboolean success)
  53 {
  54   if (sock->plugin == NULL)
  55     return;
  56
  57   sock->plugin->finish (sock->plugin, success ? 0 : 1);
  58   sock->plugin = NULL;
  59
  60   if (sock->test->pending_sockets &&
  61       sock->test->pending_sockets->data == sock) {
  62     sock->test->pending_sockets = sock->test->pending_sockets->prev;
  63   }
  64   sock->test->sockets = g_list_remove (sock->test->sockets, sock);
  65   sock->test = NULL;
  66 }
  67
  68 static void
  69 swfdec_test_socket_dispose (GObject *object)
  70 {
  71   SwfdecTestSocket *sock = SWFDEC_TEST_SOCKET (object);
  72
  73   swfdec_test_socket_do_close (sock, FALSE);
  74   swfdec_buffer_queue_unref (sock->receive_queue);
  75   sock->receive_queue = NULL;
  76
  77   G_OBJECT_CLASS (swfdec_test_socket_parent_class)->dispose (object);
  78 }
  79
  80 static void
  81 swfdec_test_socket_class_init (SwfdecTestSocketClass *klass)
  82 {
  83   GObjectClass *object_class = G_OBJECT_CLASS (klass);
  84
  85   object_class->dispose = swfdec_test_socket_dispose;
  86 }
  87
  88 static void
  89 swfdec_test_socket_init (SwfdecTestSocket *sock)
  90 {
  91   sock->receive_queue = swfdec_buffer_queue_new ();
  92 }
  93
  94 /*** AS CODE ***/
  95
  96 SWFDEC_TEST_FUNCTION ("Socket_send", swfdec_test_socket_send)
  97 void
  98 swfdec_test_socket_send (SwfdecAsContext *cx, SwfdecAsObject *object, guint argc,
  99     SwfdecAsValue *argv, SwfdecAsValue *retval)
 100 {
 101   SwfdecTestSocket *sock;
 102   SwfdecBuffer *buffer;
 103
 104   SWFDEC_AS_CHECK (SWFDEC_TYPE_TEST_SOCKET, &sock, "");
 105
 106   if (sock->plugin == NULL) {
 107     swfdec_test_throw (cx, "socket is closed");
 108     return;
 109   }
 110   buffer = swfdec_test_buffer_from_args (cx, argc, argv);
 111   sock->plugin->receive (sock->plugin, buffer->data, buffer->length);
 112   swfdec_buffer_unref (buffer);
 113 }
 114
 115 SWFDEC_TEST_FUNCTION ("Socket_receive", swfdec_test_socket_receive)
 116 void
 117 swfdec_test_socket_receive (SwfdecAsContext *cx, SwfdecAsObject *object, guint argc,
 118     SwfdecAsValue *argv, SwfdecAsValue *retval)
 119 {
 120   SwfdecTestSocket *sock;
 121   SwfdecBuffer *buffer;
 122   gsize len = 0, depth;
 123
 124   SWFDEC_AS_CHECK (SWFDEC_TYPE_TEST_SOCKET, &sock, "|i", &len);
 125
 126   depth = swfdec_buffer_queue_get_depth (sock->receive_queue);
 127   if (len > 0) {
 128     if (depth < len) {
 129       swfdec_test_throw (cx, "only %zu bytes available", depth);
 130       return;
 131     } else {
 132       buffer = swfdec_buffer_queue_pull (sock->receive_queue, len);
 133     }
 134   } else {
 135     if (depth == 0) {
 136       SWFDEC_AS_VALUE_SET_NULL (retval);
 137       return;
 138     } else {
 139       buffer = swfdec_buffer_queue_pull (sock->receive_queue, depth);
 140     }
 141   }
 142   SWFDEC_AS_VALUE_SET_OBJECT (retval, swfdec_as_relay_get_as_object (
 143         SWFDEC_AS_RELAY (swfdec_test_buffer_new (cx, buffer))));
 144 }
 145
 146 SWFDEC_TEST_FUNCTION ("Socket_error", swfdec_test_socket_error)
 147 void
 148 swfdec_test_socket_error (SwfdecAsContext *cx, SwfdecAsObject *object, guint argc,
 149     SwfdecAsValue *argv, SwfdecAsValue *retval)
 150 {
 151   SwfdecTestSocket *sock;
 152
 153   SWFDEC_AS_CHECK (SWFDEC_TYPE_TEST_SOCKET, &sock, "");
 154
 155   swfdec_test_socket_do_close (sock, FALSE);
 156 }
 157
 158 SWFDEC_TEST_FUNCTION ("Socket_close", swfdec_test_socket_close_as)
 159 void
 160 swfdec_test_socket_close_as (SwfdecAsContext *cx, SwfdecAsObject *object, guint argc,
 161     SwfdecAsValue *argv, SwfdecAsValue *retval)
 162 {
 163   SwfdecTestSocket *sock;
 164
 165   SWFDEC_AS_CHECK (SWFDEC_TYPE_TEST_SOCKET, &sock, "");
 166
 167   swfdec_test_socket_do_close (sock, TRUE);
 168 }
 169
 170 SWFDEC_TEST_FUNCTION ("Socket_get_closed", swfdec_test_socket_get_closed)
 171 void
 172 swfdec_test_socket_get_closed (SwfdecAsContext *cx, SwfdecAsObject *object, guint argc,
 173     SwfdecAsValue *argv, SwfdecAsValue *retval)
 174 {
 175   SwfdecTestSocket *sock;
 176
 177   SWFDEC_AS_CHECK (SWFDEC_TYPE_TEST_SOCKET, &sock, "");
 178
 179   SWFDEC_AS_VALUE_SET_BOOLEAN (retval, sock->plugin == NULL);
 180 }
 181
 182 static void
 183 swfdec_test_socket_plugin_send (SwfdecTestPluginSocket *plugin, unsigned char *data,
 184     unsigned long length)
 185 {
 186   SwfdecTestSocket *sock = plugin->data;
 187   SwfdecBuffer *buffer;
 188
 189   buffer = swfdec_buffer_new_for_data (g_memdup (data, length), length);
 190   swfdec_buffer_queue_push (sock->receive_queue, buffer);
 191 }
 192
 193 static void
 194 swfdec_test_socket_plugin_close (SwfdecTestPluginSocket *plugin)
 195 {
 196   swfdec_test_socket_do_close (plugin->data, TRUE);
 197 }
 198
 199 SwfdecAsObject *
 200 swfdec_test_socket_new (SwfdecTestTest *test, SwfdecTestPluginSocket *plugin)
 201 {
 202   SwfdecAsValue val;
 203   SwfdecAsContext *cx;
 204   SwfdecAsObject *new;
 205
 206   g_return_val_if_fail (SWFDEC_IS_TEST_TEST (test), NULL);
 207   g_return_val_if_fail (plugin != NULL, NULL);
 208
 209   cx = swfdec_gc_object_get_context (test);
 210
 211   new = g_object_new (SWFDEC_TYPE_TEST_SOCKET, "context", cx, NULL);
 212   swfdec_as_object_get_variable (cx->global,
 213       swfdec_as_context_get_string (cx, "Socket"), &val);
 214   if (SWFDEC_AS_VALUE_IS_OBJECT (&val))
 215     swfdec_as_object_set_constructor (new, SWFDEC_AS_VALUE_GET_OBJECT (&val));
 216
 217   plugin->send = swfdec_test_socket_plugin_send;
 218   plugin->close = swfdec_test_socket_plugin_close;
 219   plugin->data = new;
 220   SWFDEC_TEST_SOCKET (new)->test = test;
 221   SWFDEC_TEST_SOCKET (new)->plugin = plugin;
 222   test->sockets = g_list_append (test->sockets, new);
 223
 224   SWFDEC_AS_VALUE_SET_STRING (&val,
 225       swfdec_as_context_get_string (cx, plugin->host));
 226   swfdec_as_object_set_variable (new,
 227       swfdec_as_context_get_string (cx, "host"), &val);
 228   swfdec_as_value_set_integer (cx, &val, plugin->port);
 229   swfdec_as_object_set_variable (new,
 230       swfdec_as_context_get_string (cx, "port"), &val);
 231
 232   return new;
 233 }
 234
 235 void
 236 swfdec_test_socket_close (SwfdecTestSocket *sock)
 237 {
 238   g_return_if_fail (SWFDEC_IS_TEST_SOCKET (sock));
 239
 240   return swfdec_test_socket_do_close (sock, FALSE);
 241 }
 242