/*++
/* NAME
/* dict_pipe 3
/* SUMMARY
/* dictionary manager interface for pipelined tables
/* SYNOPSIS
/* #include <dict_pipe.h>
/*
/* DICT *dict_pipe_open(name, open_flags, dict_flags)
/* const char *name;
/* int open_flags;
/* int dict_flags;
/* DESCRIPTION
/* dict_pipe_open() opens a pipeline of one or more tables.
/* Example: "\fBpipemap:{\fItype_1:name_1, ..., type_n:name_n\fR}".
/*
/* Each "pipemap:" query is given to the first table.  Each
/* lookup result becomes the query for the next table in the
/* pipeline, and the last table produces the final result.
/* When any table lookup produces no result, the pipeline
/* produces no result.
/*
/* The first and last characters of the "pipemap:" table name
/* must be '{' and '}'. Within these, individual maps are
/* separated with comma or whitespace.
/*
/* The open_flags and dict_flags arguments are passed on to
/* the underlying dictionaries.
/* SEE ALSO
/* dict(3) generic dictionary manager
/* LICENSE
/* .ad
/* .fi
/* The Secure Mailer license must be distributed with this software.
/* AUTHOR(S)
/* Wietse Venema
/* IBM T.J. Watson Research
/* P.O. Box 704
/* Yorktown Heights, NY 10598, USA
/*--*/

/* System library. */

#include <sys_defs.h>
#include <string.h>

/* Utility library. */

#include <msg.h>
#include "mymalloc.h"
#include "htable.h"
#include "dict.h"
#include "dict_pipe.h"
#include "stringops.h"
#include "vstring.h"

/* Application-specific. */

typedef struct {
    DICT    dict;   /* generic members */
    ARGV   *map_pipe;   /* pipelined tables */
    VSTRING *qr_buf;   /* query/reply buffer */
} DICT_PIPE;

#define STR(x) vstring_str(x)

/* dict_pipe_lookup - search pipelined tables */

static const char *dict_pipe_lookup(DICT *dict, const char *query)
{
    static const char myname[] = "dict_pipe_lookup";
    DICT_PIPE *dict_pipe = (DICT_PIPE *) dict;
    DICT   *map;
    char  **cpp;
    char   *reg_name;
    const char *result = 0;

    vstring_strcpy(dict_pipe->qr_buf, query);
    for (cpp = dict_pipe->map_pipe->argv; (reg_name = *cpp) != 0; cpp++) {
 if ((map = dict_handle(reg_name)) == 0)
     msg_panic("%s: dictionary \"%s\" not found", myname, reg_name);
 if ((result = dict_get(map, STR(dict_pipe->qr_buf))) == 0)
     DICT_ERR_VAL_RETURN(dict, map->error, result);
 vstring_strcpy(dict_pipe->qr_buf, result);
    }
    DICT_ERR_VAL_RETURN(dict, DICT_ERR_NONE, STR(dict_pipe->qr_buf));
}

/* dict_pipe_close - disassociate from pipelined tables */

static void dict_pipe_close(DICT *dict)
{
    static const char myname[] = "dict_pipe_close";
    DICT_PIPE *dict_pipe = (DICT_PIPE *) dict;
    DICT   *map;
    char  **cpp;
    char   *reg_name;

    for (cpp = dict_pipe->map_pipe->argv; (reg_name = *cpp) != 0; cpp++) {
 if ((map = dict_handle(reg_name)) == 0)
     msg_panic("%s: dictionary \"%s\" not found", myname, reg_name);
 dict_close(map);
    }
    argv_free(dict_pipe->map_pipe);
    vstring_free(dict_pipe->qr_buf);
    dict_free(dict);
}

/* dict_pipe_open - open pipelined tables */

DICT   *dict_pipe_open(const char *name, int open_flags, int dict_flags)
{
    static const char myname[] = "dict_pipe_open";
    DICT_PIPE *dict_pipe;
    char   *saved_name = 0;
    char   *dict_type_name;
    ARGV   *argv = 0;
    char  **cpp;
    DICT   *dict;
    int     match_flags = 0;
    struct DICT_OWNER aggr_owner;
    size_t  len;

    /*
     * Clarity first. Let the optimizer worry about redundant code.
     */
#define DICT_PIPE_RETURN(x) do { \
     if (saved_name != 0) \
  myfree(saved_name); \
     if (argv != 0) \
  argv_free(argv); \
     return (x); \
 } while (0)

    /*
     * Sanity checks.
     */
    if (open_flags != O_RDONLY)
 DICT_PIPE_RETURN(dict_surrogate(DICT_TYPE_PIPE, name,
     open_flags, dict_flags,
      "%s:%s map requires O_RDONLY access mode",
     DICT_TYPE_PIPE, name));

    /*
     * Split the table name into its constituent parts.
     */
    if ((len = balpar(name, CHARS_BRACE)) == 0 || name[len] != 0
 || *(saved_name = mystrndup(name + 1, len - 2)) == 0
 || ((argv = argv_splitq(saved_name, CHARS_COMMA_SP, CHARS_BRACE)),
     (argv->argc == 0)))
 DICT_PIPE_RETURN(dict_surrogate(DICT_TYPE_PIPE, name,
     open_flags, dict_flags,
     "bad syntax: \"%s:%s\"; "
     "need \"%s:{type:name...}\"",
     DICT_TYPE_PIPE, name,
     DICT_TYPE_PIPE));

    /*
     * Check all underlying table specs before registering any of them to
     * avoid leaking refcounts if one of them is bad.
     */
    for (cpp = argv->argv; (dict_type_name = *cpp) != 0; cpp++) {
 if (strchr(dict_type_name, ':') == 0)
     DICT_PIPE_RETURN(dict_surrogate(DICT_TYPE_PIPE, name,
         open_flags, dict_flags,
         "bad syntax: \"%s:%s\"; "
         "need \"%s:{type:name...}\"",
         DICT_TYPE_PIPE, name,
         DICT_TYPE_PIPE));
    }

    /*
     * The least-trusted table in the pipeline determines the over-all trust
     * level. The first table determines the pattern-matching flags.
     */
    DICT_OWNER_AGGREGATE_INIT(aggr_owner);
    for (cpp = argv->argv; (dict_type_name = *cpp) != 0; cpp++) {
 if (msg_verbose)
     msg_info("%s: %s", myname, dict_type_name);
 dict = dict_open(dict_type_name, open_flags, dict_flags);
 argv_replace_one(argv, cpp - argv->argv, dict->reg_name);
 DICT_OWNER_AGGREGATE_UPDATE(aggr_owner, dict->owner);
 if (cpp == argv->argv)
     match_flags = dict->flags & (DICT_FLAG_FIXED | DICT_FLAG_PATTERN);
    }

    /*
     * Bundle up the result.
     */
    dict_pipe =
 (DICT_PIPE *) dict_alloc(DICT_TYPE_PIPE, name, sizeof(*dict_pipe));
    dict_pipe->dict.lookup = dict_pipe_lookup;
    dict_pipe->dict.close = dict_pipe_close;
    dict_pipe->dict.flags = dict_flags | match_flags;
    dict_pipe->dict.owner = aggr_owner;
    dict_pipe->qr_buf = vstring_alloc(100);
    dict_pipe->map_pipe = argv;
    argv = 0;
    DICT_PIPE_RETURN(&dict_pipe->dict);
}