src/charset-converter.cpp

   1 /* $Id: charset-converter.cpp,v 1.4 2005-06-25 15:58:33 adam Exp $
   2    Copyright (c) 1998-2005, Index Data.
   3
   4 This file is part of the yaz-proxy.
   5
   6 YAZ proxy is free software; you can redistribute it and/or modify it under
   7 the terms of the GNU General Public License as published by the Free
   8 Software Foundation; either version 2, or (at your option) any later
   9 version.
  10
  11 YAZ proxy is distributed in the hope that it will be useful, but WITHOUT ANY
  12 WARRANTY; without even the implied warranty of MERCHANTABILITY or
  13 FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  14 for more details.
  15
  16 You should have received a copy of the GNU General Public License
  17 along with YAZ proxy; see the file LICENSE.  If not, write to the
  18 Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
  19 02111-1307, USA.
  20  */
  21
  22 #include <yaz/log.h>
  23 #include <yaz/proto.h>
  24 #include "proxyp.h"
  25
  26 Yaz_CharsetConverter::Yaz_CharsetConverter()
  27 {
  28     m_wrbuf = wrbuf_alloc();
  29     m_target_query_charset = 0;
  30     m_client_query_charset = 0;
  31     m_client_charset_selected = 0;
  32 }
  33
  34 Yaz_CharsetConverter::~Yaz_CharsetConverter()
  35 {
  36     wrbuf_free(m_wrbuf, 1);
  37     xfree(m_target_query_charset);
  38     xfree(m_client_query_charset);
  39 }
  40
  41 const char *Yaz_CharsetConverter::get_target_query_charset()
  42 {
  43     return m_target_query_charset;
  44 }
  45
  46 void Yaz_CharsetConverter::set_target_query_charset(const char *s)
  47 {
  48     xfree(m_target_query_charset);
  49     m_target_query_charset = 0;
  50     if (s)
  51         m_target_query_charset = xstrdup(s);
  52 }
  53
  54 void Yaz_CharsetConverter::set_client_query_charset(const char *s)
  55 {
  56     xfree(m_client_query_charset);
  57     m_client_query_charset = 0;
  58     if (s)
  59         m_client_query_charset = xstrdup(s);
  60 }
  61
  62 const char *Yaz_CharsetConverter::get_client_query_charset()
  63 {
  64     return m_client_query_charset;
  65 }
  66
  67 void Yaz_CharsetConverter::set_client_charset_selected(int sel)
  68 {
  69     m_client_charset_selected = sel;
  70 }
  71
  72 int Yaz_CharsetConverter::get_client_charset_selected()
  73 {
  74     return m_client_charset_selected;
  75 }
  76
  77 void Yaz_CharsetConverter::convert_type_1(char *buf_in, int len_in,
  78                                           char **buf_out, int *len_out,
  79                                           ODR o)
  80 {
  81     wrbuf_rewind(m_wrbuf);
  82     wrbuf_iconv_write(m_wrbuf, m_ct, buf_in, len_in);
  83
  84     *len_out = wrbuf_len(m_wrbuf);
  85     if (*len_out == 0)
  86     {   // we assume conversion failed
  87         *buf_out = buf_in;
  88         *len_out = len_in;
  89     }
  90     else
  91     {
  92         *buf_out = (char*) odr_malloc(o, *len_out);
  93         memcpy(*buf_out, wrbuf_buf(m_wrbuf), *len_out);
  94     }
  95 }
  96
  97 void Yaz_CharsetConverter::convert_type_1(Z_Term *q, ODR o)
  98 {
  99     switch(q->which)
 100     {
 101     case Z_Term_general:
 102         convert_type_1((char *) q->u.general->buf, q->u.general->len,
 103                        (char **) &q->u.general->buf, &q->u.general->len, o);
 104         break;
 105     }
 106 }
 107
 108 void Yaz_CharsetConverter::convert_type_1(Z_Operand *q, ODR o)
 109 {
 110     switch(q->which)
 111     {
 112     case Z_Operand_APT:
 113         convert_type_1(q->u.attributesPlusTerm->term, o);
 114         break;
 115     case Z_Operand_resultSetId:
 116         break;
 117     case Z_Operand_resultAttr:
 118         break;
 119     }
 120 }
 121
 122 void Yaz_CharsetConverter::convert_type_1(Z_RPNStructure *q, ODR o)
 123 {
 124     switch(q->which)
 125     {
 126     case Z_RPNStructure_simple:
 127         convert_type_1(q->u.simple, o);
 128         break;
 129     case Z_RPNStructure_complex:
 130         convert_type_1(q->u.complex->s1, o);
 131         convert_type_1(q->u.complex->s2, o);
 132         break;
 133     }
 134 }
 135
 136 void Yaz_CharsetConverter::convert_type_1(Z_RPNQuery *q, ODR o)
 137 {
 138     if (m_target_query_charset && m_client_query_charset)
 139     {
 140         m_ct = yaz_iconv_open(m_target_query_charset,
 141                               m_client_query_charset);
 142         if (m_ct)
 143         {
 144             convert_type_1(q->RPNStructure, o);
 145             yaz_iconv_close(m_ct);
 146         }
 147     }
 148 }
 149 /*
 150  * Local variables:
 151  * c-basic-offset: 4
 152  * indent-tabs-mode: nil
 153  * End:
 154  * vim: shiftwidth=4 tabstop=8 expandtab
 155  */
 156