/* GLIB - Library of useful routines for C programming

 * Copyright (C) 2005 Matthias Clasen

 * Portion Copyright © 2008-09 Nokia Corporation and/or its subsidiary(-ies). All rights reserved.

 * This library is free software; you can redistribute it and/or

 * modify it under the terms of the GNU Lesser General Public

 * License as published by the Free Software Foundation; either

 * version 2 of the License, or (at your option) any later version.

 *

 * This library is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

 * Lesser General Public License for more details.

 *

 * You should have received a copy of the GNU Lesser General Public

 * License along with this library; if not, write to the

 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,

 * Boston, MA 02111-1307, USA.

 */

#include <stdlib.h>

#include <string.h>

#include <unistd.h>

#include <sys/types.h>

#include <pthread.h>

#include "glib.h"

#include "gstdio.h"

#ifdef SYMBIAN

#include "mrt2_glib2_test.h"

#endif /*SYMBIAN*/

static gchar *dir, *filename, *displayname, *childname;

static gboolean stop = FALSE;

#ifndef G_OS_WIN32

static void

handle_usr1 (int signum)

{

  stop = TRUE;

}

#endif

static gboolean

check_stop (gpointer data)

{

  GMainLoop *loop = data;

#if defined(G_OS_WIN32) || defined(SYMBIAN)

  stop = g_file_test ("STOP", G_FILE_TEST_EXISTS);

#endif

  if (stop)

    g_main_loop_quit (loop);

  return TRUE;

}

static void

write_or_die (const gchar *filename,

	      const gchar *contents,

	      gssize       length)

{

  GError *error = NULL;

  gchar *displayname;    

  if (!g_file_set_contents (filename, contents, length, &error)) 

    {

      displayname = g_filename_display_name (childname);

      g_print ("failed to write '%s': %s\n", 

	       displayname, error->message);

	  g_assert(FALSE && "mapping-test failed");

	  #ifdef SYMBIAN

  	  testResultXml("mapping-test");

  	  #endif /* EMULATOR */

      exit (1);

    }

}

static GMappedFile *

map_or_die (const gchar *filename,

	    gboolean     writable)

{

  GError *error = NULL;

  GMappedFile *map;

  gchar *displayname;

  map = g_mapped_file_new (filename, writable, &error);

  if (!map)

    {

      displayname = g_filename_display_name (childname);

      g_print ("failed to map '%s' non-writable, shared: %s\n", 

	       displayname, error->message);

	  g_assert(FALSE && "mapping-test failed");

	  #ifdef SYMBIAN

  	  testResultXml("mapping-test");

  	  #endif /* EMULATOR */

      exit (1);

    }

  return map;

}

void *child_main (void *dummy)

{

  GMappedFile *map;

  GMainLoop *loop; 

  map = map_or_die (filename, FALSE);

  loop = g_main_loop_new (NULL, FALSE);

#if !defined(G_OS_WIN32) && !defined(SYMBIAN)

  signal (SIGUSR1, handle_usr1);

#endif

  g_idle_add (check_stop, loop);

  g_main_loop_run (loop);

  write_or_die (childname, 

		g_mapped_file_get_contents (map),

		g_mapped_file_get_length (map));

  return NULL;

}

static void

test_mapping (void)

{

  GMappedFile *map;

  write_or_die (filename, "ABC", -1);

  map = map_or_die (filename, FALSE);

  g_assert (g_mapped_file_get_length (map) == 3);

  g_mapped_file_free (map);

  map = map_or_die (filename, TRUE);

  g_assert (g_mapped_file_get_length (map) == 3);

  g_mapped_file_free (map);

}

static void 

test_private (void)

{

  GError *error = NULL;

  GMappedFile *map;

  gchar *buffer;

  gsize len;

  write_or_die (filename, "ABC", -1);

  map = map_or_die (filename, TRUE);

  buffer = (gchar *)g_mapped_file_get_contents (map);

  buffer[0] = '1';

  buffer[1] = '2';

  buffer[2] = '3';

  g_mapped_file_free (map);

  if (!g_file_get_contents (filename, &buffer, &len, &error))

    {

      g_print ("failed to read '%s': %s\n", 

	       displayname, error->message);

	  g_assert(FALSE && "mapping-test failed");

	  #ifdef SYMBIAN

  	  testResultXml("mapping-test");

  	  #endif /* EMULATOR */

      exit (1);

    }

  g_assert (len == 3);

  g_assert (strcmp (buffer, "ABC") == 0);

  g_free (buffer);

}

static void

test_child_private (gchar *argv0)

{

  GError *error = NULL;

  GMappedFile *map;

  gchar *buffer;

  gsize len;

  gchar *child_argv[3];

  GPid  child_pid;

  pthread_t child_thread;

#if defined(G_OS_WIN32) && defined(SYMBIAN)

  g_remove ("STOP");

  g_assert (!g_file_test ("STOP", G_FILE_TEST_EXISTS));

#endif

  write_or_die (filename, "ABC", -1);

  map = map_or_die (filename, TRUE);

  child_argv[0] = argv0;

  child_argv[1] = "mapchild";

  child_argv[2] = NULL;

  pthread_create(&child_thread, NULL, child_main, NULL);

  //pthread_join(thread1, NULL);

  /*if (!g_spawn_async (dir, child_argv, NULL,

		      0, NULL, NULL, &child_pid, &error))

    {

      g_print ("failed to spawn child: %s\n", 

	       error->message);

      exit (1);            

    }*/

  /* give the child some time to set up its mapping */

  g_usleep (2000000);

  buffer = (gchar *)g_mapped_file_get_contents (map);

  buffer[0] = '1';

  buffer[1] = '2';

  buffer[2] = '3';

  g_mapped_file_free (map);

#if !defined(G_OS_WIN32) && !defined(SYMBIAN)

  kill (child_pid, SIGUSR1);

#else

  g_file_set_contents ("STOP", "Hey there\n", -1, NULL);

#endif

  /* give the child some time to write the file */

  g_usleep (2000000);

  pthread_join(child_thread, NULL);

  if (!g_file_get_contents (childname, &buffer, &len, &error))

    {

      gchar *name;

      name = g_filename_display_name (childname);

      g_print ("failed to read '%s': %s\n", name, error->message);

      g_assert(FALSE && "mapping-test failed");

	  #ifdef SYMBIAN

  	  testResultXml("mapping-test");

  	  #endif /* EMULATOR */

      exit (1);      

    }

  g_assert (len == 3);

  g_assert (strcmp (buffer, "ABC") == 0);

  g_free (buffer);

}

static int 

parent_main (int   argc,

	     char *argv[])

{

  /* test mapping with various flag combinations */

  test_mapping ();

  /* test private modification */

  test_private ();

  /* test multiple clients, non-shared */

  test_child_private (argv[0]);

  return 0;

}

int

main (int argc, 

      char *argv[])

{

  int retval;

  #ifndef EMULATOR

  mkdir("C:\\Private\\e0000009", 0666);

  chdir("C:\\Private\\e0000009");

  #endif//EMULATOR

  dir = g_get_current_dir ();

  filename = g_build_filename (dir, "maptest", NULL);

  displayname = g_filename_display_name (filename);

  childname = g_build_filename (dir, "mapchild", NULL);

  #ifdef SYMBIAN

  g_log_set_handler (NULL,  (GLogLevelFlags)(G_LOG_FLAG_FATAL| G_LOG_FLAG_RECURSION | G_LOG_LEVEL_CRITICAL | G_LOG_LEVEL_WARNING | G_LOG_LEVEL_MESSAGE | G_LOG_LEVEL_INFO | G_LOG_LEVEL_DEBUG), &mrtLogHandler, NULL);

  g_set_print_handler(mrtPrintHandler);

  #endif

  /*if (argc > 1)

    return child_main (argc, argv);

  else */

  retval = parent_main (argc, argv);

  #ifdef SYMBIAN

  testResultXml("mapping-test");

  #endif /* EMULATOR */

  return retval;

}