src/util.cpp

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