test-io-stream-closing-read.c 3.5 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49
/*
 * This file is part of the Nice GLib ICE library.
 *
 * (C) 2014 Collabora Ltd.
 *  Contact: Philip Withnall
 *
 * The contents of this file are subject to the Mozilla Public License Version
 * 1.1 (the "License"); you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 * http://www.mozilla.org/MPL/
 *
 * Software distributed under the License is distributed on an "AS IS" basis,
 * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
 * for the specific language governing rights and limitations under the
 * License.
 *
 * The Original Code is the Nice GLib ICE library.
 *
 * The Initial Developers of the Original Code are Collabora Ltd and Nokia
 * Corporation. All Rights Reserved.
 *
 * Contributors:
 *   Philip Withnall, Collabora Ltd.
 *
 * Alternatively, the contents of this file may be used under the terms of the
 * the GNU Lesser General Public License Version 2.1 (the "LGPL"), in which
 * case the provisions of LGPL are applicable instead of those above. If you
 * wish to allow use of your version of this file only under the terms of the
 * LGPL and not to allow others to use your version of this file under the
 * MPL, indicate your decision by deleting the provisions above and replace
 * them with the notice and other provisions required by the LGPL. If you do
 * not delete the provisions above, a recipient may use your version of this
 * file under either the MPL or the LGPL.
 */
#ifdef HAVE_CONFIG_H
# include <config.h>
#endif

#include "agent.h"
#include "test-io-stream-common.h"

#include <stdlib.h>
#include <string.h>
#ifndef G_OS_WIN32
#include <unistd.h>
#endif

#define NUM_MESSAGES 10

50 51 52 53
guint count = 0;
GMutex count_lock;
GCond count_cond;

54 55 56 57
static void
read_thread_cb (GInputStream *input_stream, TestIOStreamThreadData *data)
{
  GError *error = NULL;
58 59
  gssize len;
  guint8 buf[MESSAGE_SIZE];
60 61


62 63 64 65
  g_mutex_lock (&count_lock);
  count++;
  g_cond_broadcast (&count_cond);
  g_mutex_unlock (&count_lock);
66

67 68 69 70 71 72
  /* Block on receiving some data. */
  do {
    len = g_input_stream_read (input_stream, buf, sizeof (buf), NULL, &error);
    if (!data->user_data) {
      g_assert_cmpint (len, ==, sizeof(buf));
      return;
73
    }
74 75
  } while (len > 0);
  g_assert_cmpint (len, ==, -1);
76

77 78
  g_assert_error (error, G_IO_ERROR, G_IO_ERROR_BROKEN_PIPE);
  g_clear_error (&error);
79

80 81
  stop_main_loop (data->error_loop);
}
82

83 84 85 86 87 88 89 90
static void
write_thread_cb (GOutputStream *output_stream, TestIOStreamThreadData *data)
{
  gchar buf[MESSAGE_SIZE] = {0};
  gssize ret;
  GError *error = NULL;
  gpointer tmp;
  guint stream_id;
91

92 93
  ret = g_output_stream_write (output_stream, buf, sizeof (buf), NULL,
      &error);
94

95 96 97 98 99 100 101
  g_mutex_lock (&count_lock);
  count++;
  g_cond_broadcast (&count_cond);
  if (data->user_data) {
    g_assert_cmpint (ret, ==, sizeof(buf));
    g_mutex_unlock (&count_lock);
    return;
102 103
  }

104 105 106
  while (count != 4)
    g_cond_wait (&count_cond, &count_lock);
  g_mutex_unlock (&count_lock);
107 108


109
  /* Now we remove the stream, lets see how the writer handles that */
110

111 112
  tmp = g_object_get_data (G_OBJECT (data->other->agent), "stream-id");
  stream_id = GPOINTER_TO_UINT (tmp);
113

114
  nice_agent_remove_stream (data->other->agent, stream_id);
115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130
}

int main (void)
{
  const TestIOStreamCallbacks callbacks = {
    read_thread_cb,
    write_thread_cb,
    NULL,
    NULL,
  };

#ifdef G_OS_WIN32
  WSADATA w;
  WSAStartup (0x0202, &w);
#endif

131
  run_io_stream_test (30, TRUE, &callbacks, (gpointer) TRUE, NULL, NULL, NULL);
132 133 134 135 136 137 138

#ifdef G_OS_WIN32
  WSACleanup ();
#endif

  return 0;
}