2 * Copyright (c) 1995-2002, Index Data.
3 * See the file LICENSE for details.
5 * $Id: pquery.c,v 1.16 2002-07-25 12:48:39 adam Exp $
13 #include <yaz/proto.h>
15 #include <yaz/pquery.h>
17 static oid_value p_query_dfset = VAL_NONE;
20 const char *query_buf;
30 static Z_RPNStructure *rpn_structure (struct lex_info *li, ODR o, oid_proto,
31 int num_attr, int max_attr,
32 int *attr_list, char **attr_clist,
35 static enum oid_value query_oid_getvalbyname (struct lex_info *li)
42 memcpy (buf, li->lex_buf, li->lex_len);
43 buf[li->lex_len] = '\0';
44 value = oid_getvalbyname (buf);
48 static int compare_term (struct lex_info *li, const char *src, size_t off)
50 size_t len=strlen(src);
52 if (li->lex_len == len+off && !memcmp (li->lex_buf+off, src, len-off))
57 static int query_token (struct lex_info *li)
60 const char *sep_match;
61 const char **qptr = &li->query_buf;
68 if ((sep_match = strchr (li->left_sep, **qptr)))
70 sep_char = li->right_sep[sep_match - li->left_sep];
75 if (**qptr == li->escape_char && isdigit ((*qptr)[1]))
81 while (**qptr && **qptr != sep_char)
93 if (sep_char == ' ' &&
94 li->lex_len >= 1 && li->lex_buf[0] == li->escape_char)
96 if (compare_term (li, "and", 1))
98 if (compare_term (li, "or", 1))
100 if (compare_term (li, "not", 1))
102 if (compare_term (li, "attr", 1))
104 if (compare_term (li, "set", 1))
106 if (compare_term (li, "attrset", 1))
108 if (compare_term (li, "prox", 1))
110 if (compare_term (li, "term", 1))
116 static int lex (struct lex_info *li)
118 return li->query_look = query_token (li);
121 static int escape_string(char *out_buf, const char *in, int len)
126 if (*in == '\\' && len > 0)
152 sscanf (s, "%x", &n);
169 sscanf (s, "%o", &n);
181 return out - out_buf;
184 static int p_query_parse_attr(struct lex_info *li, ODR o,
185 int num_attr, int *attr_list,
186 char **attr_clist, oid_value *attr_set)
189 if (!(cp = strchr (li->lex_buf, '=')) ||
190 (size_t) (cp-li->lex_buf) > li->lex_len)
192 attr_set[num_attr] = query_oid_getvalbyname (li);
193 if (attr_set[num_attr] == VAL_NONE)
197 if (!(cp = strchr (li->lex_buf, '=')))
203 attr_set[num_attr] = attr_set[num_attr-1];
205 attr_set[num_attr] = VAL_NONE;
207 attr_list[2*num_attr] = atoi(li->lex_buf);
209 if (*cp >= '0' && *cp <= '9')
211 attr_list[2*num_attr+1] = atoi (cp);
212 attr_clist[num_attr] = 0;
216 int len = li->lex_len - (cp - li->lex_buf);
217 attr_list[2*num_attr+1] = 0;
218 attr_clist[num_attr] = (char *) odr_malloc (o, len+1);
219 len = escape_string(attr_clist[num_attr], cp, len);
220 attr_clist[num_attr][len] = '\0';
225 static Z_AttributesPlusTerm *rpn_term (struct lex_info *li, ODR o,
227 int num_attr, int *attr_list,
228 char **attr_clist, oid_value *attr_set)
230 Z_AttributesPlusTerm *zapt;
233 Z_AttributeElement **elements;
235 zapt = (Z_AttributesPlusTerm *)odr_malloc (o, sizeof(*zapt));
236 term_octet = (Odr_oct *)odr_malloc (o, sizeof(*term_octet));
237 term = (Z_Term *)odr_malloc (o, sizeof(*term));
240 elements = (Z_AttributeElement**)odr_nullval();
246 elements = (Z_AttributeElement**)
247 odr_malloc (o, num_attr * sizeof(*elements));
249 attr_tmp = (int *)odr_malloc (o, num_attr * 2 * sizeof(int));
250 memcpy (attr_tmp, attr_list, num_attr * 2 * sizeof(int));
251 for (i = num_attr; --i >= 0; )
254 for (j = i+1; j<num_attr; j++)
255 if (attr_tmp[2*j] == attr_tmp[2*i])
260 (Z_AttributeElement*)odr_malloc (o,sizeof(**elements));
261 elements[k]->attributeType = &attr_tmp[2*i];
262 elements[k]->attributeSet =
263 yaz_oidval_to_z3950oid(o, CLASS_ATTSET, attr_set[i]);
267 elements[k]->which = Z_AttributeValue_complex;
268 elements[k]->value.complex = (Z_ComplexAttribute *)
269 odr_malloc (o, sizeof(Z_ComplexAttribute));
270 elements[k]->value.complex->num_list = 1;
271 elements[k]->value.complex->list =
272 (Z_StringOrNumeric **)
273 odr_malloc (o, 1 * sizeof(Z_StringOrNumeric *));
274 elements[k]->value.complex->list[0] =
275 (Z_StringOrNumeric *)
276 odr_malloc (o, sizeof(Z_StringOrNumeric));
277 elements[k]->value.complex->list[0]->which =
278 Z_StringOrNumeric_string;
279 elements[k]->value.complex->list[0]->u.string =
281 elements[k]->value.complex->semanticAction = (int **)
283 elements[k]->value.complex->num_semanticAction = 0;
287 elements[k]->which = Z_AttributeValue_numeric;
288 elements[k]->value.numeric = &attr_tmp[2*i+1];
294 zapt->attributes = (Z_AttributeList *)
295 odr_malloc (o, sizeof(*zapt->attributes));
296 zapt->attributes->num_attributes = num_attr;
297 zapt->attributes->attributes = elements;
301 term_octet->buf = (unsigned char *)odr_malloc (o, 1 + li->lex_len);
302 term_octet->size = term_octet->len =
303 escape_string ((char *) (term_octet->buf), li->lex_buf, li->lex_len);
304 term_octet->buf[term_octet->size] = 0; /* null terminate */
306 switch (li->term_type)
309 term->which = Z_Term_general;
310 term->u.general = term_octet;
312 case Z_Term_characterString:
313 term->which = Z_Term_characterString;
314 term->u.characterString = term_octet->buf; /* null terminated above */
317 term->which = Z_Term_numeric;
318 term->u.numeric = odr_intdup (o, atoi(term_octet->buf));
321 term->which = Z_Term_null;
322 term->u.null = odr_nullval();
325 term->which = Z_Term_null;
326 term->u.null = odr_nullval();
332 static Z_Operand *rpn_simple (struct lex_info *li, ODR o, oid_proto proto,
333 int num_attr, int *attr_list, char **attr_clist,
338 zo = (Z_Operand *)odr_malloc (o, sizeof(*zo));
339 switch (li->query_look)
342 zo->which = Z_Operand_APT;
343 if (!(zo->u.attributesPlusTerm =
344 rpn_term (li, o, proto, num_attr, attr_list, attr_clist,
353 zo->which = Z_Operand_resultSetId;
354 zo->u.resultSetId = (char *)odr_malloc (o, li->lex_len+1);
355 memcpy (zo->u.resultSetId, li->lex_buf, li->lex_len);
356 zo->u.resultSetId[li->lex_len] = '\0';
365 static Z_ProximityOperator *rpn_proximity (struct lex_info *li, ODR o)
367 Z_ProximityOperator *p = (Z_ProximityOperator *)odr_malloc (o, sizeof(*p));
371 if (*li->lex_buf == '1')
373 p->exclusion = (int *)odr_malloc (o, sizeof(*p->exclusion));
376 else if (*li->lex_buf == '0')
378 p->exclusion = (int *)odr_malloc (o, sizeof(*p->exclusion));
386 p->distance = (int *)odr_malloc (o, sizeof(*p->distance));
387 *p->distance = atoi (li->lex_buf);
391 p->ordered = (int *)odr_malloc (o, sizeof(*p->ordered));
392 *p->ordered = atoi (li->lex_buf);
396 p->relationType = (int *)odr_malloc (o, sizeof(*p->relationType));
397 *p->relationType = atoi (li->lex_buf);
401 if (*li->lex_buf == 'k')
403 else if (*li->lex_buf == 'p')
406 p->which = atoi (li->lex_buf);
410 p->which = Z_ProximityOperator_known;
411 p->u.known = (int *)odr_malloc (o, sizeof(*p->u.known));
412 *p->u.known = atoi (li->lex_buf);
416 static Z_Complex *rpn_complex (struct lex_info *li, ODR o, oid_proto proto,
417 int num_attr, int max_attr,
418 int *attr_list, char **attr_clist,
424 zc = (Z_Complex *)odr_malloc (o, sizeof(*zc));
425 zo = (Z_Operator *)odr_malloc (o, sizeof(*zo));
427 switch (li->query_look)
430 zo->which = Z_Operator_and;
431 zo->u.and_not = odr_nullval();
434 zo->which = Z_Operator_or;
435 zo->u.and_not = odr_nullval();
438 zo->which = Z_Operator_and_not;
439 zo->u.and_not = odr_nullval();
442 zo->which = Z_Operator_prox;
443 zo->u.prox = rpn_proximity (li, o);
452 rpn_structure (li, o, proto, num_attr, max_attr, attr_list,
453 attr_clist, attr_set)))
456 rpn_structure (li, o, proto, num_attr, max_attr, attr_list,
457 attr_clist, attr_set)))
462 static void rpn_term_type (struct lex_info *li, ODR o)
466 if (compare_term (li, "general", 0))
467 li->term_type = Z_Term_general;
468 else if (compare_term (li, "numeric", 0))
469 li->term_type = Z_Term_numeric;
470 else if (compare_term (li, "string", 0))
471 li->term_type = Z_Term_characterString;
472 else if (compare_term (li, "oid", 0))
473 li->term_type = Z_Term_oid;
474 else if (compare_term (li, "datetime", 0))
475 li->term_type = Z_Term_dateTime;
476 else if (compare_term (li, "null", 0))
477 li->term_type = Z_Term_null;
481 static Z_RPNStructure *rpn_structure (struct lex_info *li, ODR o,
483 int num_attr, int max_attr,
490 sz = (Z_RPNStructure *)odr_malloc (o, sizeof(*sz));
491 switch (li->query_look)
497 sz->which = Z_RPNStructure_complex;
498 if (!(sz->u.complex =
499 rpn_complex (li, o, proto, num_attr, max_attr, attr_list,
500 attr_clist, attr_set)))
505 sz->which = Z_RPNStructure_simple;
507 rpn_simple (li, o, proto, num_attr, attr_list,
508 attr_clist, attr_set)))
515 if (num_attr >= max_attr)
517 if (!p_query_parse_attr(li, o, num_attr, attr_list,
518 attr_clist, attr_set))
523 rpn_structure (li, o, proto, num_attr, max_attr, attr_list,
524 attr_clist, attr_set);
527 rpn_term_type (li, o);
529 rpn_structure (li, o, proto, num_attr, max_attr, attr_list,
530 attr_clist, attr_set);
531 case 0: /* operator/operand expected! */
537 Z_RPNQuery *p_query_rpn_mk (ODR o, struct lex_info *li, oid_proto proto,
541 int attr_array[1024];
542 char *attr_clist[512];
543 oid_value attr_set[512];
544 oid_value topSet = VAL_NONE;
546 zq = (Z_RPNQuery *)odr_malloc (o, sizeof(*zq));
548 if (li->query_look == 'r')
551 topSet = query_oid_getvalbyname (li);
552 if (topSet == VAL_NONE)
557 if (topSet == VAL_NONE)
558 topSet = p_query_dfset;
559 if (topSet == VAL_NONE)
562 zq->attributeSetId = yaz_oidval_to_z3950oid(o, CLASS_ATTSET, topSet);
564 if (!zq->attributeSetId)
567 if (!(zq->RPNStructure = rpn_structure (li, o, proto, 0, 512,
568 attr_array, attr_clist, attr_set)))
573 Z_RPNQuery *p_query_rpn (ODR o, oid_proto proto,
579 li.right_sep = "}\"";
580 li.escape_char = '@';
581 li.term_type = Z_Term_general;
583 return p_query_rpn_mk (o, &li, proto, qbuf);
587 Z_AttributesPlusTerm *p_query_scan_mk (struct lex_info *li,
588 ODR o, oid_proto proto,
589 Odr_oid **attributeSetP,
593 char *attr_clist[512];
594 oid_value attr_set[512];
597 oid_value topSet = VAL_NONE;
600 if (li->query_look == 'r')
603 topSet = query_oid_getvalbyname (li);
607 if (topSet == VAL_NONE)
608 topSet = p_query_dfset;
609 if (topSet == VAL_NONE)
612 *attributeSetP = yaz_oidval_to_z3950oid (o, CLASS_ATTSET, topSet);
616 if (li->query_look == 'l')
621 if (num_attr >= max_attr)
623 if (!p_query_parse_attr(li, o, num_attr, attr_list,
624 attr_clist, attr_set))
629 else if (li->query_look == 'y')
632 rpn_term_type (li, o);
639 return rpn_term (li, o, proto, num_attr, attr_list, attr_clist, attr_set);
642 Z_AttributesPlusTerm *p_query_scan (ODR o, oid_proto proto,
643 Odr_oid **attributeSetP,
649 li.right_sep = "}\"";
650 li.escape_char = '@';
651 li.term_type = Z_Term_general;
654 return p_query_scan_mk (&li, o, proto, attributeSetP, qbuf);
657 int p_query_attset (const char *arg)
659 p_query_dfset = oid_getvalbyname (arg);
660 return (p_query_dfset == VAL_NONE) ? -1 : 0;