#include <sys_defs.h>
#include <sys/stat.h>
#include <stdlib.h>
#include <unistd.h>
#include <fcntl.h>
#include <ctype.h>
#include <string.h>
#include <msg.h>
#include <mymalloc.h>
#include <vstring.h>
#include <vstream.h>
#include <msg_vstream.h>
#include <msg_syslog.h>
#include <readlline.h>
#include <stringops.h>
#include <split_at.h>
#include <vstring_vstream.h>
#include <set_eugid.h>
#include <mail_conf.h>
#include <mail_dict.h>
#include <mail_params.h>
#include <mail_version.h>
#include <mkmap.h>
#include <mail_task.h>
#include <dict_proxy.h>
#include <mime_state.h>
#include <rec_type.h>
#define STR vstring_str
#define LEN VSTRING_LEN
#define POSTMAP_FLAG_AS_OWNER (1<<0)
#define POSTMAP_FLAG_SAVE_PERM (1<<1)
#define POSTMAP_FLAG_HEADER_KEY (1<<2)
#define POSTMAP_FLAG_BODY_KEY (1<<3)
#define POSTMAP_FLAG_MIME_KEY (1<<4)
#ifdef __APPLE_OS_X_SERVER__
#define POSTMAP_NO_LOGGING (1<<5)
#endif
#define POSTMAP_FLAG_HB_KEY (POSTMAP_FLAG_HEADER_KEY | POSTMAP_FLAG_BODY_KEY)
#define POSTMAP_FLAG_FULL_KEY (POSTMAP_FLAG_BODY_KEY | POSTMAP_FLAG_MIME_KEY)
#define POSTMAP_FLAG_ANY_KEY (POSTMAP_FLAG_HB_KEY | POSTMAP_FLAG_MIME_KEY)
typedef struct {
DICT **dicts;
char **maps;
int map_count;
int dict_flags;
int header_done;
int found;
} POSTMAP_KEY_STATE;
static void postmap(char *map_type, char *path_name, int postmap_flags,
int open_flags, int dict_flags)
{
VSTREAM *source_fp;
VSTRING *line_buffer;
MKMAP *mkmap;
int lineno;
char *key;
char *value;
struct stat st;
mode_t saved_mask;
line_buffer = vstring_alloc(100);
if ((open_flags & O_TRUNC) == 0) {
source_fp = VSTREAM_IN;
vstream_control(source_fp, VSTREAM_CTL_PATH, "stdin", VSTREAM_CTL_END);
} else if (strcmp(map_type, DICT_TYPE_PROXY) == 0) {
msg_fatal("can't create maps via the proxy service");
} else if ((source_fp = vstream_fopen(path_name, O_RDONLY, 0)) == 0) {
msg_fatal("open %s: %m", path_name);
}
if (fstat(vstream_fileno(source_fp), &st) < 0)
msg_fatal("fstat %s: %m", path_name);
if ((postmap_flags & POSTMAP_FLAG_SAVE_PERM) && S_ISREG(st.st_mode))
saved_mask = umask(022 | (~st.st_mode & 077));
if ((postmap_flags & POSTMAP_FLAG_AS_OWNER) && getuid() == 0
&& (st.st_uid != geteuid() || st.st_gid != getegid()))
set_eugid(st.st_uid, st.st_gid);
mkmap = mkmap_open(map_type, path_name, open_flags, dict_flags);
if ((postmap_flags & POSTMAP_FLAG_SAVE_PERM) && S_ISREG(st.st_mode))
umask(saved_mask);
lineno = 0;
while (readlline(line_buffer, source_fp, &lineno)) {
key = STR(line_buffer);
value = key + strcspn(key, " \t\r\n");
if (*value)
*value++ = 0;
while (ISSPACE(*value))
value++;
trimblanks(key, 0)[0] = 0;
trimblanks(value, 0)[0] = 0;
if (*key == 0 || *value == 0) {
msg_warn("%s, line %d: expected format: key whitespace value",
VSTREAM_PATH(source_fp), lineno);
continue;
}
if (key[strlen(key) - 1] == ':')
msg_warn("%s, line %d: record is in \"key: value\" format; is this an alias file?",
VSTREAM_PATH(source_fp), lineno);
mkmap_append(mkmap, key, value);
}
mkmap_close(mkmap);
vstring_free(line_buffer);
if (source_fp != VSTREAM_IN)
vstream_fclose(source_fp);
}
static void postmap_body(void *ptr, int unused_rec_type,
const char *keybuf,
ssize_t unused_len,
off_t unused_offset)
{
POSTMAP_KEY_STATE *state = (POSTMAP_KEY_STATE *) ptr;
DICT **dicts = state->dicts;
char **maps = state->maps;
int map_count = state->map_count;
int dict_flags = state->dict_flags;
const char *map_name;
const char *value;
int n;
for (n = 0; n < map_count; n++) {
if (dicts[n] == 0)
dicts[n] = ((map_name = split_at(maps[n], ':')) != 0 ?
dict_open3(maps[n], map_name, O_RDONLY, dict_flags) :
dict_open3(var_db_type, maps[n], O_RDONLY, dict_flags));
if ((value = dict_get(dicts[n], keybuf)) != 0) {
if (*value == 0) {
msg_warn("table %s:%s: key %s: empty string result is not allowed",
dicts[n]->type, dicts[n]->name, keybuf);
msg_warn("table %s:%s should return NO RESULT in case of NOT FOUND",
dicts[n]->type, dicts[n]->name);
}
vstream_printf("%s %s\n", keybuf, value);
state->found = 1;
break;
}
}
}
static void postmap_header(void *ptr, int unused_header_class,
const HEADER_OPTS *unused_header_info,
VSTRING *header_buf,
off_t offset)
{
postmap_body(ptr, 0, STR(header_buf), LEN(header_buf), offset);
}
static void postmap_head_end(void *ptr)
{
POSTMAP_KEY_STATE *state = (POSTMAP_KEY_STATE *) ptr;
state->header_done = 1;
}
static int postmap_queries(VSTREAM *in, char **maps, const int map_count,
const int postmap_flags,
const int dict_flags)
{
int found = 0;
VSTRING *keybuf = vstring_alloc(100);
DICT **dicts;
const char *map_name;
const char *value;
int n;
if (map_count <= 0)
msg_panic("postmap_queries: bad map count");
dicts = (DICT **) mymalloc(sizeof(*dicts) * map_count);
for (n = 0; n < map_count; n++)
dicts[n] = 0;
if ((postmap_flags & POSTMAP_FLAG_HB_KEY) == 0) {
while (vstring_get_nonl(keybuf, in) != VSTREAM_EOF) {
for (n = 0; n < map_count; n++) {
if (dicts[n] == 0)
dicts[n] = ((map_name = split_at(maps[n], ':')) != 0 ?
dict_open3(maps[n], map_name, O_RDONLY, dict_flags) :
dict_open3(var_db_type, maps[n], O_RDONLY, dict_flags));
if ((value = dict_get(dicts[n], STR(keybuf))) != 0) {
if (*value == 0) {
msg_warn("table %s:%s: key %s: empty string result is not allowed",
dicts[n]->type, dicts[n]->name, STR(keybuf));
msg_warn("table %s:%s should return NO RESULT in case of NOT FOUND",
dicts[n]->type, dicts[n]->name);
}
vstream_printf("%s %s\n", STR(keybuf), value);
found = 1;
break;
}
}
}
} else {
POSTMAP_KEY_STATE key_state;
MIME_STATE *mime_state;
int mime_errs = 0;
key_state.dicts = dicts;
key_state.maps = maps;
key_state.map_count = map_count;
key_state.dict_flags = dict_flags;
key_state.header_done = 0;
key_state.found = 0;
mime_state =
mime_state_alloc((postmap_flags & POSTMAP_FLAG_MIME_KEY) ?
0 : MIME_OPT_DISABLE_MIME,
(postmap_flags & POSTMAP_FLAG_HEADER_KEY) ?
postmap_header : (MIME_STATE_HEAD_OUT) 0,
(postmap_flags & POSTMAP_FLAG_FULL_KEY) ?
(MIME_STATE_ANY_END) 0 : postmap_head_end,
(postmap_flags & POSTMAP_FLAG_BODY_KEY) ?
postmap_body : (MIME_STATE_BODY_OUT) 0,
(MIME_STATE_ANY_END) 0,
(MIME_STATE_ERR_PRINT) 0,
(void *) &key_state);
while (vstring_get_nonl(keybuf, in) != VSTREAM_EOF
&& key_state.header_done == 0 && mime_errs == 0)
mime_errs = mime_state_update(mime_state, REC_TYPE_NORM,
STR(keybuf), LEN(keybuf));
if (mime_errs == 0)
mime_errs = mime_state_update(mime_state, REC_TYPE_END, "", 0);
if (mime_errs)
msg_fatal("message format error: %s",
mime_state_detail(mime_errs)->text);
mime_state_free(mime_state);
found = key_state.found;
}
if (found)
vstream_fflush(VSTREAM_OUT);
for (n = 0; n < map_count; n++)
if (dicts[n])
dict_close(dicts[n]);
myfree((char *) dicts);
vstring_free(keybuf);
return (found);
}
static int postmap_query(const char *map_type, const char *map_name,
const char *key, int dict_flags)
{
DICT *dict;
const char *value;
dict = dict_open3(map_type, map_name, O_RDONLY, dict_flags);
if ((value = dict_get(dict, key)) != 0) {
if (*value == 0) {
msg_warn("table %s:%s: key %s: empty string result is not allowed",
map_type, map_name, key);
msg_warn("table %s:%s should return NO RESULT in case of NOT FOUND",
map_type, map_name);
}
vstream_printf("%s\n", value);
}
vstream_fflush(VSTREAM_OUT);
dict_close(dict);
return (value != 0);
}
static int postmap_deletes(VSTREAM *in, char **maps, const int map_count,
int dict_flags)
{
int found = 0;
VSTRING *keybuf = vstring_alloc(100);
DICT **dicts;
const char *map_name;
int n;
int open_flags;
if (map_count <= 0)
msg_panic("postmap_deletes: bad map count");
dicts = (DICT **) mymalloc(sizeof(*dicts) * map_count);
for (n = 0; n < map_count; n++) {
map_name = split_at(maps[n], ':');
if (map_name && strcmp(maps[n], DICT_TYPE_PROXY) == 0)
open_flags = O_RDWR | O_CREAT;
else
open_flags = O_RDWR;
dicts[n] = (map_name != 0 ?
dict_open3(maps[n], map_name, open_flags, dict_flags) :
dict_open3(var_db_type, maps[n], open_flags, dict_flags));
}
while (vstring_get_nonl(keybuf, in) != VSTREAM_EOF)
for (n = 0; n < map_count; n++)
found |= (dict_del(dicts[n], STR(keybuf)) == 0);
for (n = 0; n < map_count; n++)
if (dicts[n])
dict_close(dicts[n]);
myfree((char *) dicts);
vstring_free(keybuf);
return (found);
}
static int postmap_delete(const char *map_type, const char *map_name,
const char *key, int dict_flags)
{
DICT *dict;
int status;
int open_flags;
if (strcmp(map_type, DICT_TYPE_PROXY) == 0)
open_flags = O_RDWR | O_CREAT;
else
open_flags = O_RDWR;
dict = dict_open3(map_type, map_name, open_flags, dict_flags);
status = dict_del(dict, key);
dict_close(dict);
return (status == 0);
}
static void postmap_seq(const char *map_type, const char *map_name,
int dict_flags)
{
DICT *dict;
const char *key;
const char *value;
int func;
if (strcmp(map_type, DICT_TYPE_PROXY) == 0)
msg_fatal("can't sequence maps via the proxy service");
dict = dict_open3(map_type, map_name, O_RDONLY, dict_flags);
for (func = DICT_SEQ_FUN_FIRST; ; func = DICT_SEQ_FUN_NEXT) {
if (dict_seq(dict, func, &key, &value) != 0)
break;
if (*key == 0) {
msg_warn("table %s:%s: empty lookup key value is not allowed",
map_type, map_name);
} else if (*value == 0) {
msg_warn("table %s:%s: key %s: empty string result is not allowed",
map_type, map_name, key);
msg_warn("table %s:%s should return NO RESULT in case of NOT FOUND",
map_type, map_name);
}
vstream_printf("%s %s\n", key, value);
}
vstream_fflush(VSTREAM_OUT);
dict_close(dict);
}
static NORETURN usage(char *myname)
{
msg_fatal("usage: %s [-Nfinoprsvw] [-c config_dir] [-d key] [-q key] [map_type:]file...",
myname);
}
MAIL_VERSION_STAMP_DECLARE;
int main(int argc, char **argv)
{
char *path_name;
int ch;
int fd;
char *slash;
struct stat st;
int postmap_flags = POSTMAP_FLAG_AS_OWNER | POSTMAP_FLAG_SAVE_PERM;
int open_flags = O_RDWR | O_CREAT | O_TRUNC;
int dict_flags = DICT_FLAG_DUP_WARN | DICT_FLAG_FOLD_FIX;
char *query = 0;
char *delkey = 0;
int sequence = 0;
int found;
MAIL_VERSION_STAMP_ALLOCATE;
umask(022);
for (fd = 0; fd < 3; fd++)
if (fstat(fd, &st) == -1
&& (close(fd), open("/dev/null", O_RDWR, 0)) != fd)
msg_fatal("open /dev/null: %m");
if (getenv(CONF_ENV_VERB))
msg_verbose = 1;
if ((slash = strrchr(argv[0], '/')) != 0 && slash[1])
argv[0] = slash + 1;
msg_vstream_init(argv[0], VSTREAM_ERR);
msg_syslog_init(mail_task(argv[0]), LOG_PID, LOG_FACILITY);
#ifdef __APPLE_OS_X_SERVER__
int log_mask = setlogmask(LOG_UPTO(LOG_CRIT));
#endif
#ifdef __APPLE_OS_X_SERVER__
while ((ch = GETOPT(argc, argv, "Nbc:d:fhimnopq:rsvwx")) > 0) {
#else
while ((ch = GETOPT(argc, argv, "Nbc:d:fhimnopq:rsvw")) > 0) {
#endif
switch (ch) {
default:
usage(argv[0]);
break;
case 'N':
dict_flags |= DICT_FLAG_TRY1NULL;
dict_flags &= ~DICT_FLAG_TRY0NULL;
break;
case 'b':
postmap_flags |= POSTMAP_FLAG_BODY_KEY;
break;
case 'c':
if (setenv(CONF_ENV_PATH, optarg, 1) < 0)
msg_fatal("out of memory");
break;
case 'd':
if (sequence || query || delkey)
msg_fatal("specify only one of -s -q or -d");
delkey = optarg;
break;
case 'f':
dict_flags &= ~DICT_FLAG_FOLD_FIX;
break;
case 'h':
postmap_flags |= POSTMAP_FLAG_HEADER_KEY;
break;
case 'i':
open_flags &= ~O_TRUNC;
break;
case 'm':
postmap_flags |= POSTMAP_FLAG_MIME_KEY;
break;
case 'n':
dict_flags |= DICT_FLAG_TRY0NULL;
dict_flags &= ~DICT_FLAG_TRY1NULL;
break;
case 'o':
postmap_flags &= ~POSTMAP_FLAG_AS_OWNER;
break;
case 'p':
postmap_flags &= ~POSTMAP_FLAG_SAVE_PERM;
break;
case 'q':
if (sequence || query || delkey)
msg_fatal("specify only one of -s -q or -d");
query = optarg;
break;
case 'r':
dict_flags &= ~(DICT_FLAG_DUP_WARN | DICT_FLAG_DUP_IGNORE);
dict_flags |= DICT_FLAG_DUP_REPLACE;
break;
case 's':
if (query || delkey)
msg_fatal("specify only one of -s or -q or -d");
sequence = 1;
break;
case 'v':
msg_verbose++;
break;
case 'w':
dict_flags &= ~(DICT_FLAG_DUP_WARN | DICT_FLAG_DUP_REPLACE);
dict_flags |= DICT_FLAG_DUP_IGNORE;
break;
#ifdef __APPLE_OS_X_SERVER__
case 'x':
postmap_flags |= POSTMAP_NO_LOGGING;
break;
#endif
}
}
mail_conf_read();
#ifdef __APPLE_OS_X_SERVER__
if ( !(postmap_flags & POSTMAP_NO_LOGGING) )
setlogmask(log_mask);
#endif
if (strcmp(var_syslog_name, DEF_SYSLOG_NAME) != 0)
msg_syslog_init(mail_task(argv[0]), LOG_PID, LOG_FACILITY);
mail_dict_init();
if ((query == 0 || strcmp(query, "-") != 0)
&& (postmap_flags & POSTMAP_FLAG_ANY_KEY))
msg_fatal("specify -b -h or -m only with \"-q -\"");
if (delkey) {
if (optind + 1 > argc)
usage(argv[0]);
if (strcmp(delkey, "-") == 0)
exit(postmap_deletes(VSTREAM_IN, argv + optind, argc - optind,
dict_flags | DICT_FLAG_LOCK) == 0);
found = 0;
while (optind < argc) {
if ((path_name = split_at(argv[optind], ':')) != 0) {
found |= postmap_delete(argv[optind], path_name, delkey,
dict_flags | DICT_FLAG_LOCK);
} else {
found |= postmap_delete(var_db_type, argv[optind], delkey,
dict_flags | DICT_FLAG_LOCK);
}
optind++;
}
exit(found ? 0 : 1);
} else if (query) {
if (optind + 1 > argc)
usage(argv[0]);
if (strcmp(query, "-") == 0)
exit(postmap_queries(VSTREAM_IN, argv + optind, argc - optind,
postmap_flags, dict_flags | DICT_FLAG_LOCK) == 0);
while (optind < argc) {
if ((path_name = split_at(argv[optind], ':')) != 0) {
found = postmap_query(argv[optind], path_name, query,
dict_flags | DICT_FLAG_LOCK);
} else {
found = postmap_query(var_db_type, argv[optind], query,
dict_flags | DICT_FLAG_LOCK);
}
if (found)
exit(0);
optind++;
}
exit(1);
} else if (sequence) {
while (optind < argc) {
if ((path_name = split_at(argv[optind], ':')) != 0) {
postmap_seq(argv[optind], path_name,
dict_flags | DICT_FLAG_LOCK);
} else {
postmap_seq(var_db_type, argv[optind],
dict_flags | DICT_FLAG_LOCK);
}
exit(0);
}
exit(1);
} else {
if (optind + 1 > argc)
usage(argv[0]);
while (optind < argc) {
if ((path_name = split_at(argv[optind], ':')) != 0) {
postmap(argv[optind], path_name, postmap_flags,
open_flags, dict_flags);
} else {
postmap(var_db_type, argv[optind], postmap_flags,
open_flags, dict_flags);
}
optind++;
}
exit(0);
}
}