X-Git-Url: http://git.indexdata.com/?p=yaz-moved-to-github.git;a=blobdiff_plain;f=src%2Fpquery.c;h=8db6ac62d5c6d3b7d707a505fa71ef8f45b387b1;hp=16d826a8baf4bf704c25767da65d6bbc372aae25;hb=2715f6522ca62ab5dbc886c21ed18945743216f0;hpb=379504a233e3e2cc85bca1e7b6d864f1395aec7c diff --git a/src/pquery.c b/src/pquery.c index 16d826a..8db6ac6 100644 --- a/src/pquery.c +++ b/src/pquery.c @@ -1,15 +1,18 @@ /* This file is part of the YAZ toolkit. - * Copyright (C) 1995-2009 Index Data + * Copyright (C) 1995-2011 Index Data * See the file LICENSE for details. */ /** * \file pquery.c * \brief Implements PQF parsing */ +#if HAVE_CONFIG_H +#include +#endif + #include #include #include -#include #include #include @@ -31,7 +34,7 @@ struct yaz_pqf_parser { static Z_RPNStructure *rpn_structure(struct yaz_pqf_parser *li, ODR o, int num_attr, int max_attr, - int *attr_list, char **attr_clist, + Odr_int *attr_list, char **attr_clist, Odr_oid **attr_set); static Odr_oid *query_oid_getvalbyname(struct yaz_pqf_parser *li, ODR o) @@ -40,7 +43,7 @@ static Odr_oid *query_oid_getvalbyname(struct yaz_pqf_parser *li, ODR o) if (li->lex_len >= sizeof(buf)-1) return 0; - memcpy (buf, li->lex_buf, li->lex_len); + memcpy(buf, li->lex_buf, li->lex_len); buf[li->lex_len] = '\0'; return yaz_string_to_oid_odr(yaz_oid_std(), CLASS_ATTSET, buf, o); } @@ -50,7 +53,7 @@ static int compare_term(struct yaz_pqf_parser *li, const char *src, { size_t len=strlen(src); - if (li->lex_len == len+off && !memcmp (li->lex_buf+off, src, len-off)) + if (li->lex_len == len+off && !memcmp(li->lex_buf+off, src, len-off)) return 1; return 0; } @@ -66,14 +69,14 @@ static int query_token(struct yaz_pqf_parser *li) if (**qptr == '\0') return 0; li->lex_len = 0; - if ((sep_match = strchr (li->left_sep, **qptr))) + if ((sep_match = strchr(li->left_sep, **qptr))) { sep_char = li->right_sep[sep_match - li->left_sep]; ++(*qptr); } li->lex_buf = *qptr; - if (**qptr == li->escape_char && isdigit (((const unsigned char *) *qptr)[1])) + if (**qptr == li->escape_char && yaz_isdigit((*qptr)[1])) { ++(li->lex_len); ++(*qptr); @@ -81,7 +84,7 @@ static int query_token(struct yaz_pqf_parser *li) } while (**qptr && **qptr != sep_char) { - if (**qptr == '\\') + if (**qptr == '\\' && (*qptr)[1]) { ++(li->lex_len); ++(*qptr); @@ -94,21 +97,21 @@ static int query_token(struct yaz_pqf_parser *li) if (sep_char == ' ' && li->lex_len >= 1 && li->lex_buf[0] == li->escape_char) { - if (compare_term (li, "and", 1)) + if (compare_term(li, "and", 1)) return 'a'; - if (compare_term (li, "or", 1)) + if (compare_term(li, "or", 1)) return 'o'; - if (compare_term (li, "not", 1)) + if (compare_term(li, "not", 1)) return 'n'; - if (compare_term (li, "attr", 1)) + if (compare_term(li, "attr", 1)) return 'l'; - if (compare_term (li, "set", 1)) + if (compare_term(li, "set", 1)) return 's'; - if (compare_term (li, "attrset", 1)) + if (compare_term(li, "attrset", 1)) return 'r'; - if (compare_term (li, "prox", 1)) + if (compare_term(li, "prox", 1)) return 'p'; - if (compare_term (li, "term", 1)) + if (compare_term(li, "term", 1)) return 'y'; } return 't'; @@ -119,7 +122,7 @@ static int lex(struct yaz_pqf_parser *li) return li->query_look = query_token(li); } -static int escape_string(char *out_buf, const char *in, int len) +int escape_string(char *out_buf, const char *in, int len) { char *out = out_buf; @@ -150,7 +153,7 @@ static int escape_string(char *out_buf, const char *in, int len) s[1] = *++in; s[2] = '\0'; len = len - 2; - sscanf (s, "%x", &n); + sscanf(s, "%x", &n); *out++ = n; } break; @@ -167,7 +170,7 @@ static int escape_string(char *out_buf, const char *in, int len) s[2] = *++in; s[3] = '\0'; len = len - 2; - sscanf (s, "%o", &n); + sscanf(s, "%o", &n); *out++ = n; } break; @@ -182,27 +185,28 @@ static int escape_string(char *out_buf, const char *in, int len) return out - out_buf; } -static int p_query_parse_attr(struct yaz_pqf_parser *li, ODR o, - int num_attr, int *attr_list, +int p_query_parse_attr(struct yaz_pqf_parser *li, ODR o, + int num_attr, Odr_int *attr_list, char **attr_clist, Odr_oid **attr_set) { const char *cp; + size_t i; - if (!(cp = strchr (li->lex_buf, '=')) || + if (!(cp = strchr(li->lex_buf, '=')) || (size_t) (cp-li->lex_buf) > li->lex_len) { - attr_set[num_attr] = query_oid_getvalbyname (li, o); + attr_set[num_attr] = query_oid_getvalbyname(li, o); if (attr_set[num_attr] == 0) { li->error = YAZ_PQF_ERROR_ATTSET; return 0; } - if (!lex (li)) + if (!lex(li)) { li->error = YAZ_PQF_ERROR_MISSING; return 0; } - if (!(cp = strchr (li->lex_buf, '='))) + if (!(cp = strchr(li->lex_buf, '='))) { li->error = YAZ_PQF_ERROR_BADATTR; return 0; @@ -220,103 +224,98 @@ static int p_query_parse_attr(struct yaz_pqf_parser *li, ODR o, li->error = YAZ_PQF_ERROR_BAD_INTEGER; return 0; } - attr_list[2*num_attr] = atoi(li->lex_buf); + attr_list[2*num_attr] = odr_atoi(li->lex_buf); cp++; - if (*cp >= '0' && *cp <= '9') - { - attr_list[2*num_attr+1] = atoi (cp); - attr_clist[num_attr] = 0; - } - else - { - int len = li->lex_len - (cp - li->lex_buf); - attr_list[2*num_attr+1] = 0; - attr_clist[num_attr] = (char *) odr_malloc (o, len+1); - len = escape_string(attr_clist[num_attr], cp, len); - attr_clist[num_attr][len] = '\0'; - } + + /* inspect value .. and make it a integer if it appears to be */ + for (i = cp - li->lex_buf; i < li->lex_len; i++) + if (li->lex_buf[i] < '0' || li->lex_buf[i] > '9') + { + int len = li->lex_len - (cp - li->lex_buf); + attr_list[2*num_attr+1] = 0; + attr_clist[num_attr] = (char *) odr_malloc(o, len+1); + len = escape_string(attr_clist[num_attr], cp, len); + attr_clist[num_attr][len] = '\0'; + return 1; + } + attr_list[2*num_attr+1] = odr_atoi(cp); + attr_clist[num_attr] = 0; return 1; } -static Z_AttributesPlusTerm *rpn_term(struct yaz_pqf_parser *li, ODR o, - int num_attr, int *attr_list, - char **attr_clist, Odr_oid **attr_set) +Z_AttributeList *get_attributeList(ODR o, int num_attr, Odr_int *attr_list, + char **attr_clist, Odr_oid **attr_set) { - Z_AttributesPlusTerm *zapt; - Odr_oct *term_octet; - Z_Term *term; + int i, k = 0; + Odr_int *attr_tmp; Z_AttributeElement **elements; - - zapt = (Z_AttributesPlusTerm *)odr_malloc (o, sizeof(*zapt)); - term_octet = (Odr_oct *)odr_malloc (o, sizeof(*term_octet)); - term = (Z_Term *)odr_malloc (o, sizeof(*term)); - + Z_AttributeList *attributes= (Z_AttributeList *) + odr_malloc(o, sizeof(*attributes)); + attributes->num_attributes = num_attr; if (!num_attr) - elements = (Z_AttributeElement**)odr_nullval(); - else { - int i, k = 0; - int *attr_tmp; - - elements = (Z_AttributeElement**) - odr_malloc (o, num_attr * sizeof(*elements)); + attributes->attributes = (Z_AttributeElement**)odr_nullval(); + return attributes; + } + elements = (Z_AttributeElement**) + odr_malloc(o, num_attr * sizeof(*elements)); - attr_tmp = (int *)odr_malloc (o, num_attr * 2 * sizeof(int)); - memcpy (attr_tmp, attr_list, num_attr * 2 * sizeof(int)); - for (i = num_attr; --i >= 0; ) + attr_tmp = (Odr_int *)odr_malloc(o, num_attr * 2 * sizeof(*attr_tmp)); + memcpy(attr_tmp, attr_list, num_attr * 2 * sizeof(*attr_tmp)); + for (i = num_attr; --i >= 0; ) + { + int j; + for (j = i+1; jattributeType = &attr_tmp[2*i]; + elements[k]->attributeSet = attr_set[i]; + + if (attr_clist[i]) { - int j; - for (j = i+1; jattributeType = &attr_tmp[2*i]; - elements[k]->attributeSet = attr_set[i]; - - if (attr_clist[i]) - { - elements[k]->which = Z_AttributeValue_complex; - elements[k]->value.complex = (Z_ComplexAttribute *) - odr_malloc (o, sizeof(Z_ComplexAttribute)); - elements[k]->value.complex->num_list = 1; - elements[k]->value.complex->list = - (Z_StringOrNumeric **) - odr_malloc (o, 1 * sizeof(Z_StringOrNumeric *)); - elements[k]->value.complex->list[0] = - (Z_StringOrNumeric *) - odr_malloc (o, sizeof(Z_StringOrNumeric)); - elements[k]->value.complex->list[0]->which = - Z_StringOrNumeric_string; - elements[k]->value.complex->list[0]->u.string = - attr_clist[i]; - elements[k]->value.complex->semanticAction = 0; - elements[k]->value.complex->num_semanticAction = 0; - } - else - { - elements[k]->which = Z_AttributeValue_numeric; - elements[k]->value.numeric = &attr_tmp[2*i+1]; - } - k++; + elements[k]->which = Z_AttributeValue_complex; + elements[k]->value.complex = (Z_ComplexAttribute *) + odr_malloc(o, sizeof(Z_ComplexAttribute)); + elements[k]->value.complex->num_list = 1; + elements[k]->value.complex->list = + (Z_StringOrNumeric **) + odr_malloc(o, 1 * sizeof(Z_StringOrNumeric *)); + elements[k]->value.complex->list[0] = + (Z_StringOrNumeric *) + odr_malloc(o, sizeof(Z_StringOrNumeric)); + elements[k]->value.complex->list[0]->which = + Z_StringOrNumeric_string; + elements[k]->value.complex->list[0]->u.string = + attr_clist[i]; + elements[k]->value.complex->semanticAction = 0; + elements[k]->value.complex->num_semanticAction = 0; } - num_attr = k; + else + { + elements[k]->which = Z_AttributeValue_numeric; + elements[k]->value.numeric = &attr_tmp[2*i+1]; + } + k++; } - zapt->attributes = (Z_AttributeList *) - odr_malloc (o, sizeof(*zapt->attributes)); - zapt->attributes->num_attributes = num_attr; - zapt->attributes->attributes = elements; - - zapt->term = term; + attributes->num_attributes = k; + attributes->attributes = elements; + return attributes; +} - term_octet->buf = (unsigned char *)odr_malloc (o, 1 + li->lex_len); - term_octet->size = term_octet->len = - escape_string ((char *) (term_octet->buf), li->lex_buf, li->lex_len); +Z_Term *z_Term_create(ODR o, int term_type, const char *buf, size_t len) +{ + Z_Term *term = (Z_Term *)odr_malloc(o, sizeof(*term)); + Odr_oct *term_octet = (Odr_oct *)odr_malloc(o, sizeof(*term_octet)); + term_octet->buf = (unsigned char *)odr_malloc(o, 1 + len); + memcpy(term_octet->buf, buf, len); + term_octet->size = term_octet->len = len; term_octet->buf[term_octet->size] = 0; /* null terminate */ - switch (li->term_type) + switch (term_type) { case Z_Term_general: term->which = Z_Term_general; @@ -325,11 +324,11 @@ static Z_AttributesPlusTerm *rpn_term(struct yaz_pqf_parser *li, ODR o, case Z_Term_characterString: term->which = Z_Term_characterString; term->u.characterString = (char*) term_octet->buf; - /* null terminated above */ + /* null terminated above */ break; case Z_Term_numeric: term->which = Z_Term_numeric; - term->u.numeric = odr_intdup (o, atoi((char*) (term_octet->buf))); + term->u.numeric = odr_intdup(o, odr_atoi((const char*) term_octet->buf)); break; case Z_Term_null: term->which = Z_Term_null; @@ -344,16 +343,38 @@ static Z_AttributesPlusTerm *rpn_term(struct yaz_pqf_parser *li, ODR o, term->u.null = odr_nullval(); break; } + return term; +} + +static Z_AttributesPlusTerm *rpn_term_attributes( + struct yaz_pqf_parser *li, ODR o, Z_AttributeList *attributes) +{ + char *es_str = odr_malloc(o, li->lex_len+1); + int es_len = escape_string(es_str, li->lex_buf, li->lex_len); + Z_Term *term = z_Term_create(o, li->term_type, es_str, es_len); + Z_AttributesPlusTerm *zapt = (Z_AttributesPlusTerm *) + odr_malloc(o, sizeof(*zapt)); + + zapt->term = term; + zapt->attributes = attributes; return zapt; } +static Z_AttributesPlusTerm *rpn_term(struct yaz_pqf_parser *li, ODR o, + int num_attr, Odr_int *attr_list, + char **attr_clist, Odr_oid **attr_set) +{ + return rpn_term_attributes(li, o, get_attributeList(o, num_attr, attr_list, attr_clist, attr_set)); +} + static Z_Operand *rpn_simple(struct yaz_pqf_parser *li, ODR o, - int num_attr, int *attr_list, char **attr_clist, + int num_attr, Odr_int *attr_list, + char **attr_clist, Odr_oid **attr_set) { Z_Operand *zo; - zo = (Z_Operand *)odr_malloc (o, sizeof(*zo)); + zo = (Z_Operand *)odr_malloc(o, sizeof(*zo)); switch (li->query_look) { case 't': @@ -361,20 +382,20 @@ static Z_Operand *rpn_simple(struct yaz_pqf_parser *li, ODR o, if (!(zo->u.attributesPlusTerm = rpn_term(li, o, num_attr, attr_list, attr_clist, attr_set))) return 0; - lex (li); + lex(li); break; case 's': - lex (li); + lex(li); if (!li->query_look) { li->error = YAZ_PQF_ERROR_MISSING; return 0; } zo->which = Z_Operand_resultSetId; - zo->u.resultSetId = (char *)odr_malloc (o, li->lex_len+1); - memcpy (zo->u.resultSetId, li->lex_buf, li->lex_len); + zo->u.resultSetId = (char *)odr_malloc(o, li->lex_len+1); + memcpy(zo->u.resultSetId, li->lex_buf, li->lex_len); zo->u.resultSetId[li->lex_len] = '\0'; - lex (li); + lex(li); break; default: /* we're only called if one of the above types are seens so @@ -385,19 +406,19 @@ static Z_Operand *rpn_simple(struct yaz_pqf_parser *li, ODR o, return zo; } -static Z_ProximityOperator *rpn_proximity (struct yaz_pqf_parser *li, ODR o) +static Z_ProximityOperator *rpn_proximity(struct yaz_pqf_parser *li, ODR o) { - Z_ProximityOperator *p = (Z_ProximityOperator *)odr_malloc (o, sizeof(*p)); + Z_ProximityOperator *p = (Z_ProximityOperator *)odr_malloc(o, sizeof(*p)); - if (!lex (li)) + if (!lex(li)) { li->error = YAZ_PQF_ERROR_MISSING; return NULL; } if (*li->lex_buf == '1') - p->exclusion = odr_intdup (o, 1); + p->exclusion = odr_booldup(o, 1); else if (*li->lex_buf == '0') - p->exclusion = odr_intdup (o, 0); + p->exclusion = odr_booldup(o, 0); else if (*li->lex_buf == 'v' || *li->lex_buf == 'n') p->exclusion = NULL; else @@ -406,28 +427,28 @@ static Z_ProximityOperator *rpn_proximity (struct yaz_pqf_parser *li, ODR o) return NULL; } - if (!lex (li)) + if (!lex(li)) { li->error = YAZ_PQF_ERROR_MISSING; return NULL; } if (*li->lex_buf >= '0' && *li->lex_buf <= '9') - p->distance = odr_intdup (o, atoi (li->lex_buf)); + p->distance = odr_intdup(o, odr_atoi(li->lex_buf)); else { li->error = YAZ_PQF_ERROR_BAD_INTEGER; return NULL; } - if (!lex (li)) + if (!lex(li)) { li->error = YAZ_PQF_ERROR_MISSING; return NULL; } if (*li->lex_buf == '1') - p->ordered = odr_intdup (o, 1); + p->ordered = odr_booldup(o, 1); else if (*li->lex_buf == '0') - p->ordered = odr_intdup (o, 0); + p->ordered = odr_booldup(o, 0); else { li->error = YAZ_PQF_ERROR_PROXIMITY; @@ -440,14 +461,14 @@ static Z_ProximityOperator *rpn_proximity (struct yaz_pqf_parser *li, ODR o) return NULL; } if (*li->lex_buf >= '0' && *li->lex_buf <= '9') - p->relationType = odr_intdup (o, atoi (li->lex_buf)); + p->relationType = odr_intdup(o, odr_atoi(li->lex_buf)); else { li->error = YAZ_PQF_ERROR_BAD_INTEGER; return NULL; } - if (!lex (li)) + if (!lex(li)) { li->error = YAZ_PQF_ERROR_MISSING; return NULL; @@ -457,7 +478,7 @@ static Z_ProximityOperator *rpn_proximity (struct yaz_pqf_parser *li, ODR o) else if (*li->lex_buf == 'p') p->which = Z_ProximityOperator_private; else - p->which = atoi (li->lex_buf); + p->which = atoi(li->lex_buf); if (p->which != Z_ProximityOperator_known && p->which != Z_ProximityOperator_private) @@ -466,13 +487,13 @@ static Z_ProximityOperator *rpn_proximity (struct yaz_pqf_parser *li, ODR o) return NULL; } - if (!lex (li)) + if (!lex(li)) { li->error = YAZ_PQF_ERROR_MISSING; return NULL; } if (*li->lex_buf >= '0' && *li->lex_buf <= '9') - p->u.known = odr_intdup (o, atoi(li->lex_buf)); + p->u.known = odr_intdup(o, odr_atoi(li->lex_buf)); else { li->error = YAZ_PQF_ERROR_BAD_INTEGER; @@ -483,14 +504,14 @@ static Z_ProximityOperator *rpn_proximity (struct yaz_pqf_parser *li, ODR o) static Z_Complex *rpn_complex(struct yaz_pqf_parser *li, ODR o, int num_attr, int max_attr, - int *attr_list, char **attr_clist, + Odr_int *attr_list, char **attr_clist, Odr_oid **attr_set) { Z_Complex *zc; Z_Operator *zo; - zc = (Z_Complex *)odr_malloc (o, sizeof(*zc)); - zo = (Z_Operator *)odr_malloc (o, sizeof(*zo)); + zc = (Z_Complex *)odr_malloc(o, sizeof(*zc)); + zo = (Z_Operator *)odr_malloc(o, sizeof(*zo)); zc->roperator = zo; switch (li->query_look) { @@ -508,7 +529,7 @@ static Z_Complex *rpn_complex(struct yaz_pqf_parser *li, ODR o, break; case 'p': zo->which = Z_Operator_prox; - zo->u.prox = rpn_proximity (li, o); + zo->u.prox = rpn_proximity(li, o); if (!zo->u.prox) return NULL; break; @@ -518,7 +539,7 @@ static Z_Complex *rpn_complex(struct yaz_pqf_parser *li, ODR o, li->error = YAZ_PQF_ERROR_INTERNAL; return NULL; } - lex (li); + lex(li); if (!(zc->s1 = rpn_structure(li, o, num_attr, max_attr, attr_list, attr_clist, attr_set))) @@ -530,21 +551,21 @@ static Z_Complex *rpn_complex(struct yaz_pqf_parser *li, ODR o, return zc; } -static void rpn_term_type(struct yaz_pqf_parser *li, ODR o) +static void rpn_term_type(struct yaz_pqf_parser *li) { if (!li->query_look) return ; - if (compare_term (li, "general", 0)) + if (compare_term(li, "general", 0)) li->term_type = Z_Term_general; - else if (compare_term (li, "numeric", 0)) + else if (compare_term(li, "numeric", 0)) li->term_type = Z_Term_numeric; - else if (compare_term (li, "string", 0)) + else if (compare_term(li, "string", 0)) li->term_type = Z_Term_characterString; - else if (compare_term (li, "oid", 0)) + else if (compare_term(li, "oid", 0)) li->term_type = Z_Term_oid; - else if (compare_term (li, "datetime", 0)) + else if (compare_term(li, "datetime", 0)) li->term_type = Z_Term_dateTime; - else if (compare_term (li, "null", 0)) + else if (compare_term(li, "null", 0)) li->term_type = Z_Term_null; #if 0 else if (compare_term(li, "range", 0)) @@ -554,18 +575,18 @@ static void rpn_term_type(struct yaz_pqf_parser *li, ODR o) li->external_type = VAL_MULTISRCH2; } #endif - lex (li); + lex(li); } static Z_RPNStructure *rpn_structure(struct yaz_pqf_parser *li, ODR o, int num_attr, int max_attr, - int *attr_list, + Odr_int *attr_list, char **attr_clist, Odr_oid **attr_set) { Z_RPNStructure *sz; - sz = (Z_RPNStructure *)odr_malloc (o, sizeof(*sz)); + sz = (Z_RPNStructure *)odr_malloc(o, sizeof(*sz)); switch (li->query_look) { case 'a': @@ -574,20 +595,20 @@ static Z_RPNStructure *rpn_structure(struct yaz_pqf_parser *li, ODR o, case 'p': sz->which = Z_RPNStructure_complex; if (!(sz->u.complex = - rpn_complex (li, o, num_attr, max_attr, attr_list, - attr_clist, attr_set))) + rpn_complex(li, o, num_attr, max_attr, attr_list, + attr_clist, attr_set))) return NULL; break; case 't': case 's': sz->which = Z_RPNStructure_simple; if (!(sz->u.simple = - rpn_simple (li, o, num_attr, attr_list, - attr_clist, attr_set))) + rpn_simple(li, o, num_attr, attr_list, + attr_clist, attr_set))) return NULL; break; case 'l': - lex (li); + lex(li); if (!li->query_look) { li->error = YAZ_PQF_ERROR_MISSING; @@ -602,16 +623,16 @@ static Z_RPNStructure *rpn_structure(struct yaz_pqf_parser *li, ODR o, attr_clist, attr_set)) return 0; num_attr++; - lex (li); + lex(li); return - rpn_structure (li, o, num_attr, max_attr, attr_list, - attr_clist, attr_set); + rpn_structure(li, o, num_attr, max_attr, attr_list, + attr_clist, attr_set); case 'y': - lex (li); - rpn_term_type (li, o); + lex(li); + rpn_term_type(li); return - rpn_structure (li, o, num_attr, max_attr, attr_list, - attr_clist, attr_set); + rpn_structure(li, o, num_attr, max_attr, attr_list, + attr_clist, attr_set); case 0: /* operator/operand expected! */ li->error = YAZ_PQF_ERROR_MISSING; return 0; @@ -619,26 +640,26 @@ static Z_RPNStructure *rpn_structure(struct yaz_pqf_parser *li, ODR o, return sz; } -Z_RPNQuery *p_query_rpn_mk(ODR o, struct yaz_pqf_parser *li, const char *qbuf) +static Z_RPNQuery *p_query_rpn_mk(ODR o, struct yaz_pqf_parser *li) { Z_RPNQuery *zq; - int attr_array[1024]; + Odr_int attr_array[1024]; char *attr_clist[512]; Odr_oid *attr_set[512]; Odr_oid *top_set = 0; - zq = (Z_RPNQuery *)odr_malloc (o, sizeof(*zq)); - lex (li); + zq = (Z_RPNQuery *)odr_malloc(o, sizeof(*zq)); + lex(li); if (li->query_look == 'r') { - lex (li); + lex(li); top_set = query_oid_getvalbyname(li, o); if (!top_set) { li->error = YAZ_PQF_ERROR_ATTSET; return NULL; } - lex (li); + lex(li); } if (!top_set) { @@ -664,6 +685,12 @@ Z_RPNQuery *p_query_rpn_mk(ODR o, struct yaz_pqf_parser *li, const char *qbuf) return zq; } +static void pqf_parser_begin(struct yaz_pqf_parser *li, const char *buf) +{ + li->query_buf = li->query_ptr = buf; + li->lex_buf = 0; +} + Z_RPNQuery *p_query_rpn(ODR o, const char *qbuf) { struct yaz_pqf_parser li; @@ -673,36 +700,33 @@ Z_RPNQuery *p_query_rpn(ODR o, const char *qbuf) li.right_sep = "}\""; li.escape_char = '@'; li.term_type = Z_Term_general; - li.query_buf = li.query_ptr = qbuf; - li.lex_buf = 0; - return p_query_rpn_mk(o, &li, qbuf); -} + pqf_parser_begin(&li, qbuf); + return p_query_rpn_mk(o, &li); +} -Z_AttributesPlusTerm *p_query_scan_mk(struct yaz_pqf_parser *li, - ODR o, oid_proto proto, - Odr_oid **attributeSetP, - const char *qbuf) +static Z_AttributeList *p_query_scan_attributes_mk(struct yaz_pqf_parser *li, + ODR o, + Odr_oid **attributeSetP) { - int attr_list[1024]; + Odr_int attr_list[1024]; char *attr_clist[512]; Odr_oid *attr_set[512]; int num_attr = 0; int max_attr = 512; Odr_oid *top_set = 0; - Z_AttributesPlusTerm *apt; - lex (li); + lex(li); if (li->query_look == 'r') { - lex (li); + lex(li); top_set = query_oid_getvalbyname(li, o); if (!top_set) { li->error = YAZ_PQF_ERROR_ATTSET; return NULL; } - lex (li); + lex(li); } if (!top_set) { @@ -714,7 +738,7 @@ Z_AttributesPlusTerm *p_query_scan_mk(struct yaz_pqf_parser *li, { if (li->query_look == 'l') { - lex (li); + lex(li); if (!li->query_look) { li->error = YAZ_PQF_ERROR_MISSING; @@ -729,24 +753,34 @@ Z_AttributesPlusTerm *p_query_scan_mk(struct yaz_pqf_parser *li, attr_clist, attr_set)) return 0; num_attr++; - lex (li); + lex(li); } else if (li->query_look == 'y') { - lex (li); - rpn_term_type (li, o); + lex(li); + rpn_term_type(li); } else break; } + return get_attributeList(o, num_attr, attr_list, attr_clist, attr_set); +} + +static Z_AttributesPlusTerm *p_query_scan_mk(struct yaz_pqf_parser *li, + ODR o, + Odr_oid **attributeSetP) +{ + Z_AttributeList *attr_list = p_query_scan_attributes_mk(li, o, attributeSetP); + Z_AttributesPlusTerm *apt; + if (!li->query_look) { li->error = YAZ_PQF_ERROR_MISSING; return 0; } - apt = rpn_term(li, o, num_attr, attr_list, attr_clist, attr_set); + apt = rpn_term_attributes(li, o, attr_list); - lex (li); + lex(li); if (li->query_look != 0) { @@ -756,26 +790,9 @@ Z_AttributesPlusTerm *p_query_scan_mk(struct yaz_pqf_parser *li, return apt; } -Z_AttributesPlusTerm *p_query_scan (ODR o, oid_proto proto, - Odr_oid **attributeSetP, - const char *qbuf) -{ - struct yaz_pqf_parser li; - - li.error = 0; - li.left_sep = "{\""; - li.right_sep = "}\""; - li.escape_char = '@'; - li.term_type = Z_Term_general; - li.query_buf = li.query_ptr = qbuf; - li.lex_buf = 0; - - return p_query_scan_mk (&li, o, proto, attributeSetP, qbuf); -} - -YAZ_PQF_Parser yaz_pqf_create (void) +YAZ_PQF_Parser yaz_pqf_create(void) { - YAZ_PQF_Parser p = (YAZ_PQF_Parser) xmalloc (sizeof(*p)); + YAZ_PQF_Parser p = (YAZ_PQF_Parser) xmalloc(sizeof(*p)); p->error = 0; p->left_sep = "{\""; @@ -788,16 +805,15 @@ YAZ_PQF_Parser yaz_pqf_create (void) void yaz_pqf_destroy(YAZ_PQF_Parser p) { - xfree (p); + xfree(p); } Z_RPNQuery *yaz_pqf_parse(YAZ_PQF_Parser p, ODR o, const char *qbuf) { if (!p) return 0; - p->query_buf = p->query_ptr = qbuf; - p->lex_buf = 0; - return p_query_rpn_mk (o, p, qbuf); + pqf_parser_begin(p, qbuf); + return p_query_rpn_mk(o, p); } Z_AttributesPlusTerm *yaz_pqf_scan(YAZ_PQF_Parser p, ODR o, @@ -806,12 +822,84 @@ Z_AttributesPlusTerm *yaz_pqf_scan(YAZ_PQF_Parser p, ODR o, { if (!p) return 0; - p->query_buf = p->query_ptr = qbuf; - p->lex_buf = 0; - return p_query_scan_mk (p, o, PROTO_Z3950, attributeSetP, qbuf); + pqf_parser_begin(p, qbuf); + return p_query_scan_mk(p, o, attributeSetP); +} + +Z_AttributeList *yaz_pqf_scan_attribute_list(YAZ_PQF_Parser p, ODR o, + Odr_oid **attributeSetP, + const char *qbuf) +{ + if (!p) + return 0; + pqf_parser_begin(p, qbuf); + return p_query_scan_attributes_mk(p, o, attributeSetP); +} + +static Z_FacetField* parse_facet(ODR odr, const char *facet) +{ + YAZ_PQF_Parser pqf_parser = yaz_pqf_create(); + struct yaz_pqf_parser *li = pqf_parser; + Odr_oid *attributeSetId; + Z_FacetField *facet_field = 0; + Z_AttributeList *attribute_list; + + pqf_parser_begin(pqf_parser, facet); + attribute_list = p_query_scan_attributes_mk(li, odr, &attributeSetId); + if (attribute_list) + { + facet_field = (Z_FacetField *) odr_malloc(odr, sizeof(*facet_field)); + facet_field->attributes = attribute_list; + facet_field->num_terms = 0; + facet_field->terms = odr_malloc(odr, 10 * sizeof(*facet_field->terms)); + while (li->query_look == 't') + { + if (facet_field->num_terms < 10) + { + char *es_str = odr_malloc(odr, li->lex_len+1); + int es_len = escape_string(es_str, li->lex_buf, li->lex_len); + Z_Term *term = z_Term_create(odr, li->term_type, es_str, es_len); + + facet_field->terms[facet_field->num_terms] = + (Z_FacetTerm *) odr_malloc(odr, sizeof(Z_FacetTerm)); + facet_field->terms[facet_field->num_terms]->term = term; + facet_field->terms[facet_field->num_terms]->count = + odr_intdup(odr, 0); + facet_field->num_terms++; + } + lex(li); + } + } + yaz_pqf_destroy(pqf_parser); + return facet_field; +} + +Z_FacetList *yaz_pqf_parse_facet_list(ODR o, const char *qbuf) +{ + char **darray; + int num; + + nmem_strsplit(odr_getmem(o), ",", qbuf, &darray, &num); + if (num > 0) + { + int i; + Z_FacetList *fl = (Z_FacetList*) odr_malloc(o, sizeof(*fl)); + fl->num = num; + fl->elements = (Z_FacetField **) + odr_malloc(o, num * sizeof(*fl->elements)); + for (i = 0; i < num; i++) + { + fl->elements[i] = parse_facet(o, darray[i]); + if (!fl->elements[i]) + return 0; + } + return fl; + } + else + return 0; } -int yaz_pqf_error (YAZ_PQF_Parser p, const char **msg, size_t *off) +int yaz_pqf_error(YAZ_PQF_Parser p, const char **msg, size_t *off) { switch (p->error) {