summaryrefslogtreecommitdiff
path: root/usr.sbin/smtpd/map.c
blob: 28466c024621d64a75a9f8d680739e4fba289638 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
/*	$OpenBSD: map.c,v 1.6 2009/05/13 21:20:55 jacekm Exp $	*/

/*
 * Copyright (c) 2008 Gilles Chehade <gilles@openbsd.org>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <sys/types.h>
#include <sys/queue.h>
#include <sys/tree.h>
#include <sys/param.h>
#include <sys/socket.h>

#include <db.h>
#include <errno.h>
#include <event.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "smtpd.h"

struct map *
map_findbyname(struct smtpd *env, const char *name)
{
	struct map	*m;

	TAILQ_FOREACH(m, env->sc_maps, m_entry) {
		if (strcmp(m->m_name, name) == 0)
			break;
	}
	return (m);
}

struct map *
map_find(struct smtpd *env, objid_t id)
{
	struct map	*m;

	TAILQ_FOREACH(m, env->sc_maps, m_entry) {
		if (m->m_id == id)
			break;
	}
	return (m);
}

char *
map_dblookup(struct smtpd *env, char *mapname, char *keyname)
{
	int ret;
	DBT key;
	DBT val;
	DB *db;
	struct map *map;
	char *result = NULL;

	map = map_findbyname(env, mapname);
	if (map == NULL)
		return NULL;

	if (map->m_src != S_DB) {
		log_warn("invalid map type for map \"%s\"", mapname);
		return NULL;
	}

	db = dbopen(map->m_config, O_RDONLY, 0600, DB_HASH, NULL);
	if (db == NULL) {
		log_warn("map_dblookup: can't open %s", map->m_config);
		return NULL;
	}

	key.data = keyname;
	key.size = strlen(key.data) + 1;

	if ((ret = db->get(db, &key, &val, 0)) == -1) {
		log_warn("map_dblookup: map '%s'", mapname);
		db->close(db);
		return NULL;
	}

	if (ret == 0) {
		result = calloc(val.size, 1);
		if (result == NULL)
			fatal("calloc");
		(void)strlcpy(result, val.data, val.size);
	}

	db->close(db);

	return ret == 0 ? result : NULL;
}