Moved to gedcom subdirectory.
[gedcom-parse.git] / gedcom / encoding.c
1 /*  This program is free software; you can redistribute it and/or modify  *
2  *  it under the terms of the GNU General Public License as published by  *
3  *  the Free Software Foundation; either version 2 of the License, or     *
4  *  (at your option) any later version.                                   *
5
6  (C) 2001 by The Genes Development Team
7  Original author: Peter Verthez (Peter.Verthez@advalvas.be)
8 */
9
10 /* $Id$ */
11 /* $Name$ */
12
13 #include <string.h>
14 #include <iconv.h>
15 #include <search.h>
16 #include <stdio.h>
17 #include "gedcom_internal.h"
18 #include "encoding.h"
19
20 #define INTERNAL_ENCODING "UTF8"
21 #define ENCODING_CONF_FILE "gedcom.enc"
22 #define MAXBUF 255
23
24 static iconv_t cd_to_internal = (iconv_t) -1;
25 static void *encoding_mapping = NULL;
26 static ENCODING the_enc = ONE_BYTE;
27
28 struct node {
29   char *gedcom_name;
30   char *iconv_name;
31 };
32
33 char* charwidth_string[] = { "1", "2_HILO", "2_LOHI" };
34
35 int node_compare(const void *node1, const void *node2)
36 {
37   return strcmp(((const struct node *) node1)->gedcom_name,
38                 ((const struct node *) node2)->gedcom_name);
39 }
40
41 void add_encoding(char *gedcom_n, char* charwidth, char *iconv_n)
42 {
43   void **datum;
44   struct node *nodeptr = (struct node *) malloc(sizeof *nodeptr);
45   nodeptr->gedcom_name = (char *) malloc(strlen(gedcom_n)
46                                          + strlen(charwidth) + 3);
47   nodeptr->iconv_name  = (char *) malloc(strlen(iconv_n) + 1);
48   /* sprintf is safe here (malloc'ed before) */
49   sprintf(nodeptr->gedcom_name, "%s(%s)", gedcom_n, charwidth);
50   strcpy(nodeptr->iconv_name, iconv_n);
51   datum = tsearch(nodeptr, &encoding_mapping, node_compare);
52   if ((datum == NULL) || (*datum != nodeptr)) {
53     gedcom_warning("Duplicate entry found for encoding '%s', ignoring",
54                    gedcom_n);
55   }
56 }
57
58 char* get_encoding(char* gedcom_n, ENCODING enc)
59 {
60   void **datum;
61   struct node search_node;
62   char *buffer;
63   buffer = (char*)malloc(strlen(gedcom_n) + strlen(charwidth_string[enc]) + 3);
64   /* sprintf is safe here (malloc'ed before) */
65   sprintf(buffer, "%s(%s)", gedcom_n, charwidth_string[enc]);
66   search_node.gedcom_name = buffer;
67   datum = tfind(&search_node, &encoding_mapping, node_compare);
68   free(buffer);
69   if (datum == NULL) {
70     gedcom_error("No encoding found for '%s'", gedcom_n);
71     return NULL;
72   }
73   else {
74     return ((const struct node *) *datum)->iconv_name;
75   }
76 }
77
78 void init_encodings()
79 {
80   if (encoding_mapping == NULL) {
81     FILE *in;
82     char buffer[MAXBUF + 1];
83     char gedcom_n[MAXBUF + 1];
84     char charwidth[MAXBUF + 1];
85     char iconv_n[MAXBUF + 1];
86     in = fopen(ENCODING_CONF_FILE, "r");
87     if (in != NULL) {
88       while (fgets(buffer, sizeof(buffer), in) != NULL) {
89         if (buffer[strlen(buffer) - 1] != '\n') {
90           gedcom_error("Line too long in encoding configuration file '%s'",
91                        ENCODING_CONF_FILE);
92           return;
93         }
94         else if ((buffer[0] != '#') && (strcmp(buffer, "\n") != 0)) {
95           if (sscanf(buffer, "%s %s %s", gedcom_n, charwidth, iconv_n) == 3) {
96             add_encoding(gedcom_n, charwidth, iconv_n);
97           }
98           else {
99             gedcom_error("Missing data in encoding configuration file '%s'",
100                          ENCODING_CONF_FILE);
101             return;
102           }
103         }
104       }
105       fclose(in);
106     }
107     else {
108       gedcom_warning("Could not open encoding configuration file '%s'",
109                      ENCODING_CONF_FILE);
110     }
111   }
112 }
113
114 void set_encoding_width(ENCODING enc)
115 {
116   the_enc = enc;
117 }
118
119 static char conv_buf[MAXGEDCLINELEN * 2];
120 static size_t conv_buf_size;
121
122 int open_conv_to_internal(char* fromcode)
123 {
124   char *encoding = get_encoding(fromcode, the_enc);
125   if (cd_to_internal != (iconv_t) -1)
126     iconv_close(cd_to_internal);
127   if (encoding == NULL) {
128     cd_to_internal = (iconv_t) -1;
129   }
130   else {
131     memset(conv_buf, 0, sizeof(conv_buf));
132     conv_buf_size = 0;
133     cd_to_internal = iconv_open(INTERNAL_ENCODING, encoding);
134     if (cd_to_internal == (iconv_t) -1) {
135       gedcom_error("Error opening conversion context for encoding %s: %s",
136                    encoding, strerror(errno));
137     }
138   }
139   return (cd_to_internal != (iconv_t) -1);  
140 }
141
142 void close_conv_to_internal()
143 {
144   iconv_close(cd_to_internal);
145   cd_to_internal = (iconv_t) -1;
146 }
147
148 char* to_internal(char* str, size_t len,
149                   char* output_buffer, size_t out_len)
150 {
151   size_t outsize = out_len;
152   char *wrptr = output_buffer;
153   char *rdptr = conv_buf;
154   /* set up input buffer (concatenate to what was left previous time) */
155   /* can't use strcpy, because possible null bytes from unicode */
156   memcpy(conv_buf + conv_buf_size, str, len);
157   conv_buf_size += len;
158   /* set up output buffer (empty it) */
159   memset(output_buffer, 0, out_len);
160   /* do the conversion */
161   iconv(cd_to_internal, &rdptr, &conv_buf_size, &wrptr, &outsize);
162   /* then shift what is left over to the head of the input buffer */
163   memmove(conv_buf, rdptr, conv_buf_size);
164   memset(conv_buf + conv_buf_size, 0, sizeof(conv_buf) - conv_buf_size);
165   return output_buffer;
166 }