src/util.cpp

   1 /* $Id: util.cpp,v 1.5 2006-01-16 15:51:56 adam Exp $
   2    Copyright (c) 2005, Index Data.
   3
   4 %LICENSE%
   5  */
   6
   7 #include "config.hpp"
   8
   9 #include <yaz/odr.h>
  10 #include <yaz/pquery.h>
  11 #include "util.hpp"
  12
  13
  14 bool yp2::util::pqf(ODR odr, Z_APDU *apdu, const std::string &q) {
  15     YAZ_PQF_Parser pqf_parser = yaz_pqf_create();
  16
  17     Z_RPNQuery *rpn = yaz_pqf_parse(pqf_parser, odr, q.c_str());
  18     if (!rpn)
  19     {
  20         yaz_pqf_destroy(pqf_parser);
  21         return false;
  22     }
  23     yaz_pqf_destroy(pqf_parser);
  24     Z_Query *query = (Z_Query *) odr_malloc(odr, sizeof(Z_Query));
  25     query->which = Z_Query_type_1;
  26     query->u.type_1 = rpn;
  27
  28     apdu->u.searchRequest->query = query;
  29     return true;
  30 }
  31
  32
  33 bool yp2::util::set_databases_from_zurl(ODR odr, std::string zurl,
  34                                         int *db_num, char ***db_strings)
  35 {
  36     const char *sep = strchr(zurl.c_str(), '/');
  37     if (!sep)
  38         return false;
  39
  40     int num = 0;
  41     const char *cp1 = sep+1;
  42     while(1)
  43     {
  44         const char *cp2 = strchr(cp1, '+');
  45         if (!cp2)
  46             break;
  47         cp1 = cp2+1;
  48         num++;
  49     }
  50     *db_num = num+1;
  51     *db_strings = (char **) odr_malloc(odr, sizeof(char*) * (*db_num));
  52
  53     num = 0;
  54     cp1 = sep+1;
  55     while(1)
  56     {
  57         const char *cp2 = strchr(cp1, '+');
  58         if (cp2)
  59             (*db_strings)[num] = odr_strdupn(odr, cp1, cp2-cp1-1);
  60         else
  61         {
  62             (*db_strings)[num] = odr_strdup(odr, cp1);
  63             break;
  64         }
  65         cp1 = cp2+1;
  66         num++;
  67     }
  68     return true;
  69 }
  70
  71 yp2::odr::odr(int type)
  72 {
  73     m_odr = odr_createmem(type);
  74 }
  75
  76 yp2::odr::odr()
  77 {
  78     m_odr = odr_createmem(ODR_ENCODE);
  79 }
  80
  81 yp2::odr::~odr()
  82 {
  83     odr_destroy(m_odr);
  84 }
  85
  86 yp2::odr::operator ODR() const
  87 {
  88     return m_odr;
  89 }
  90
  91 Z_APDU *yp2::odr::create_close(Z_APDU *in_apdu,
  92                                int reason, const char *addinfo)
  93 {
  94     Z_APDU *apdu = create_APDU(Z_APDU_close, in_apdu);
  95
  96     *apdu->u.close->closeReason = reason;
  97     if (addinfo)
  98         apdu->u.close->diagnosticInformation = odr_strdup(m_odr, addinfo);
  99     return apdu;
 100 }
 101
 102 Z_APDU *yp2::odr::create_APDU(int type, Z_APDU *in_apdu)
 103 {
 104     return yp2::util::create_APDU(m_odr, type, in_apdu);
 105 }
 106
 107 Z_APDU *yp2::util::create_APDU(ODR odr, int type, Z_APDU *in_apdu)
 108 {
 109     Z_APDU *out_apdu = zget_APDU(odr, type);
 110
 111     Z_ReferenceId **id_to = yp2::util::get_referenceId(out_apdu);
 112     *id_to = 0;
 113     if (in_apdu)
 114     {
 115         Z_ReferenceId **id_from = yp2::util::get_referenceId(in_apdu);
 116         if (id_from && *id_from && id_to)
 117         {
 118             *id_to = (Z_ReferenceId*) odr_malloc (odr, sizeof(**id_to));
 119             (*id_to)->size = (*id_to)->len = (*id_from)->len;
 120             (*id_to)->buf = (unsigned char*) odr_malloc(odr, (*id_to)->len);
 121             memcpy((*id_to)->buf, (*id_from)->buf, (*id_to)->len);
 122         }
 123         else if (id_to)
 124             *id_to = 0;
 125     }
 126     return out_apdu;
 127 }
 128
 129 Z_APDU *yp2::odr::create_initResponse(Z_APDU *in_apdu,
 130                                       int error, const char *addinfo)
 131 {
 132     Z_APDU *apdu = create_APDU(Z_APDU_initResponse, in_apdu);
 133     if (error)
 134     {
 135         apdu->u.initResponse->userInformationField =
 136             zget_init_diagnostics(m_odr, error, addinfo);
 137         *apdu->u.initResponse->result = 0;
 138     }
 139     return apdu;
 140 }
 141
 142 Z_APDU *yp2::odr::create_searchResponse(Z_APDU *in_apdu,
 143                                         int error, const char *addinfo)
 144 {
 145     Z_APDU *apdu = create_APDU(Z_APDU_searchResponse, in_apdu);
 146     if (error)
 147     {
 148         Z_Records *rec = (Z_Records *) odr_malloc(m_odr, sizeof(Z_Records));
 149         *apdu->u.searchResponse->searchStatus = 0;
 150         apdu->u.searchResponse->records = rec;
 151         rec->which = Z_Records_NSD;
 152         rec->u.nonSurrogateDiagnostic =
 153             zget_DefaultDiagFormat(m_odr, error, addinfo);
 154
 155     }
 156     return apdu;
 157 }
 158
 159 Z_APDU *yp2::odr::create_presentResponse(Z_APDU *in_apdu,
 160                                          int error, const char *addinfo)
 161 {
 162     Z_APDU *apdu = create_APDU(Z_APDU_presentResponse, in_apdu);
 163     if (error)
 164     {
 165         Z_Records *rec = (Z_Records *) odr_malloc(m_odr, sizeof(Z_Records));
 166         apdu->u.presentResponse->records = rec;
 167         rec->which = Z_Records_NSD;
 168         rec->u.nonSurrogateDiagnostic =
 169             zget_DefaultDiagFormat(m_odr, error, addinfo);
 170     }
 171     return apdu;
 172 }
 173
 174 Z_APDU *yp2::odr::create_scanResponse(Z_APDU *in_apdu,
 175                                       int error, const char *addinfo)
 176 {
 177     Z_APDU *apdu = create_APDU(Z_APDU_scanResponse, in_apdu);
 178     if (error)
 179     {
 180         Z_ScanResponse *res = apdu->u.scanResponse;
 181         res->entries = (Z_ListEntries *) odr_malloc(m_odr, sizeof(*res->entries));
 182         *res->scanStatus = Z_Scan_failure;
 183
 184         res->entries->num_entries = 0;
 185         res->entries->entries = 0;
 186         res->entries->num_nonsurrogateDiagnostics = 1;
 187         res->entries->nonsurrogateDiagnostics = (Z_DiagRec **)
 188             odr_malloc(m_odr, sizeof(Z_DiagRec *));
 189         res->entries->nonsurrogateDiagnostics[0] =
 190             zget_DiagRec(m_odr, error, addinfo);
 191     }
 192     return apdu;
 193 }
 194
 195 Z_ReferenceId **yp2::util::get_referenceId(Z_APDU *apdu)
 196 {
 197     switch (apdu->which)
 198     {
 199     case  Z_APDU_initRequest:
 200         return &apdu->u.initRequest->referenceId;
 201     case  Z_APDU_initResponse:
 202         return &apdu->u.initResponse->referenceId;
 203     case  Z_APDU_searchRequest:
 204         return &apdu->u.searchRequest->referenceId;
 205     case  Z_APDU_searchResponse:
 206         return &apdu->u.searchResponse->referenceId;
 207     case  Z_APDU_presentRequest:
 208         return &apdu->u.presentRequest->referenceId;
 209     case  Z_APDU_presentResponse:
 210         return &apdu->u.presentResponse->referenceId;
 211     case  Z_APDU_deleteResultSetRequest:
 212         return &apdu->u.deleteResultSetRequest->referenceId;
 213     case  Z_APDU_deleteResultSetResponse:
 214         return &apdu->u.deleteResultSetResponse->referenceId;
 215     case  Z_APDU_accessControlRequest:
 216         return &apdu->u.accessControlRequest->referenceId;
 217     case  Z_APDU_accessControlResponse:
 218         return &apdu->u.accessControlResponse->referenceId;
 219     case  Z_APDU_resourceControlRequest:
 220         return &apdu->u.resourceControlRequest->referenceId;
 221     case  Z_APDU_resourceControlResponse:
 222         return &apdu->u.resourceControlResponse->referenceId;
 223     case  Z_APDU_triggerResourceControlRequest:
 224         return &apdu->u.triggerResourceControlRequest->referenceId;
 225     case  Z_APDU_resourceReportRequest:
 226         return &apdu->u.resourceReportRequest->referenceId;
 227     case  Z_APDU_resourceReportResponse:
 228         return &apdu->u.resourceReportResponse->referenceId;
 229     case  Z_APDU_scanRequest:
 230         return &apdu->u.scanRequest->referenceId;
 231     case  Z_APDU_scanResponse:
 232         return &apdu->u.scanResponse->referenceId;
 233     case  Z_APDU_sortRequest:
 234         return &apdu->u.sortRequest->referenceId;
 235     case  Z_APDU_sortResponse:
 236         return &apdu->u.sortResponse->referenceId;
 237     case  Z_APDU_segmentRequest:
 238         return &apdu->u.segmentRequest->referenceId;
 239     case  Z_APDU_extendedServicesRequest:
 240         return &apdu->u.extendedServicesRequest->referenceId;
 241     case  Z_APDU_extendedServicesResponse:
 242         return &apdu->u.extendedServicesResponse->referenceId;
 243     case  Z_APDU_close:
 244         return &apdu->u.close->referenceId;
 245     }
 246     return 0;
 247 }
 248
 249
 250 /*
 251  * Local variables:
 252  * c-basic-offset: 4
 253  * indent-tabs-mode: nil
 254  * c-file-style: "stroustrup"
 255  * End:
 256  * vim: shiftwidth=4 tabstop=8 expandtab
 257  */