#include <stdlib.h>

 #include <stdio.h>

 #include <string.h>

 #include <unistd.h>

 #include <errno.h>

 #include <fcntl.h>

 #ifdef WIN32

 #include <io.h>

 #endif

 #include <glib.h>

 #include <glib/gstdio.h>

 #include <bl/bl.h>

 #include "testcase.h"

 #include "testcaseinput.h"

 #ifndef O_BINARY

 #define O_BINARY 0

 #endif

 /*

  * As write(), but with error handling.

  */

 static size_t write_file(int fd,const char *buf,size_t count,GError **error)

 {

     if (write(fd,buf,count)<count)

     {

 	g_set_error(error,G_FILE_ERROR,g_file_error_from_errno(errno),

 	  "Error writing bookloupe input file: %s",g_strerror(errno));

 	return -1;

     }

     return count;

 }

 /*

  * Replace \n with \r\n, U+240A (visible symbol for LF) with \n

  * and U+240D (visible symbol for CR) with \r.

  */

 static char *unix2dos(const char *text)

 {

     gunichar c;

     const gunichar visible_lf=0x240A;

     const gunichar visible_cr=0x240D;

     GString *string;

     string=g_string_new(NULL);

     while(*text)

     {

 	c=g_utf8_get_char(text);

 	text=g_utf8_next_char(text);

 	if (c=='\n')

 	    g_string_append(string,"\r\n");

 	else if (c==visible_lf)

 	    g_string_append_c(string,'\n');

 	else if (c==visible_cr)

 	    g_string_append_c(string,'\r');

 	else

 	    g_string_append_unichar(string,c);

     }

     return g_string_free(string,FALSE);

 }

 /*

  * Create an input file needed for a testcase (as specified in <input>).

  * The file is written in the encoding specified for communicating with

  * bookloupe. The name_used field of <input> is filled in with the name

  * of the created file (which may be different than the name specified

  * if that contained "XXXXXX" to be replaced by a unique string).

  */

 gboolean testcase_input_create(Testcase *testcase,TestcaseInput *input,

   GError **error)

 {

     int fd;

     size_t n;

     char *filename,*s,*t;

     GError *tmp_err=NULL;

     if (input->contents)

     {

 	if (testcase->encoding)

 	{

 	    if (testcase->flags&TESTCASE_UNIX_NEWLINES)

 		s=g_convert(input->contents,-1,testcase->encoding,"UTF-8",NULL,

 		  &n,&tmp_err);

 	    else

 	    {

 		t=unix2dos(input->contents);

 		s=g_convert(t,-1,testcase->encoding,"UTF-8",NULL,&n,&tmp_err);

 		g_free(t);

 	    }

 	    if (!s)

 	    {

 		g_propagate_prefixed_error(error,tmp_err,

 		  "Conversion to %s failed: ",testcase->encoding);

 		return FALSE;

 	    }

 	}

 	else if (testcase->flags&TESTCASE_UNIX_NEWLINES)

 	{

 	    s=g_strdup(input->contents);

 	    n=strlen(s);

 	}

 	else

 	{

 	    s=unix2dos(input->contents);

 	    n=strlen(s);

 	}

     }

     else

     {

 	n=0;

 	s=NULL;

     }

     g_free(input->name_used);

     input->name_used=NULL;

     if (testcase->tmpdir)

 	filename=g_build_filename(testcase->tmpdir,input->name,NULL);

     else

 	filename=g_strdup(input->name);

     if (strstr(input->name,"XXXXXX"))

 	fd=g_mkstemp(filename);

     else

 	fd=g_open(filename,O_WRONLY|O_CREAT|O_EXCL|O_BINARY,0600);

     if (fd<0)

     {

 	g_set_error(error,G_FILE_ERROR,g_file_error_from_errno(errno),

 	  "%s: %s",filename,g_strerror(errno));

 	g_free(s);

 	return FALSE;

     }

     input->name_used=g_strdup(filename+strlen(filename)-strlen(input->name));

     if (n && write_file(fd,s,n,error)!=n)

     {

 	g_free(s);

 	close(fd);

 	(void)g_unlink(filename);

 	g_free(filename);

 	g_free(input->name_used);

 	input->name_used=NULL;

 	return FALSE;

     }

     g_free(s);

     if (close(fd)<0)

     {

 	g_set_error(error,G_FILE_ERROR,g_file_error_from_errno(errno),

 	  "%s: %s",filename,g_strerror(errno));

 	(void)g_unlink(filename);

 	g_free(filename);

 	g_free(input->name_used);

 	input->name_used=NULL;

 	return FALSE;

     }

     g_free(filename);

     return TRUE;

 }

 /*

  * Remove an input file created with testcase_input_create()

  */

 gboolean testcase_input_remove(Testcase *testcase,TestcaseInput *input,

   GError **error)

 {

     char *filename;

     if (input->name_used)

     {

 	if (testcase->tmpdir)

 	    filename=g_build_filename(testcase->tmpdir,input->name_used,NULL);

 	else

 	    filename=g_strdup(input->name_used);

 	if (g_unlink(filename)<0)

 	{

 	    g_set_error(error,G_FILE_ERROR,g_file_error_from_errno(errno),

 	      "%s: %s",filename,g_strerror(errno));

 	    return FALSE;

 	}

 	g_free(filename);

 	g_free(input->name_used);

 	input->name_used=NULL;

     }

     return TRUE;

 }

 /* Create a new description of an input file needed for a testcase */

 TestcaseInput *testcase_input_new(const char *name,const char *contents)

 {

     TestcaseInput *input;

     input=g_new0(TestcaseInput,1);

     input->name=g_strdup(name);

     input->contents=g_strdup(contents);

     return input;

 }

 /* Free the description of a testcase input file */

 void testcase_input_free(TestcaseInput *input)

 {

     g_free(input->name);

     g_free(input->name_used);

     g_free(input->contents);

     g_free(input);

 }