← Back to branch summary

~zulcss/samba/server-dailies-3.4

« back to all changes in this revision

Viewing changes to source3/lib/conn_tdb.c

  • Committer: Chuck Short
  • Date: 2010-09-28 20:38:39 UTC
  • Revision ID: zulcss@ubuntu.com-20100928203839-pgjulytsi9ue63x1
Initial version

Show diffs side-by-side

added added

removed removed

Lines of Context:
source3/lib/conn_tdb.c
 
1
/* 
 
2
   Unix SMB/CIFS implementation.
 
3
   Low-level connections.tdb access functions
 
4
   Copyright (C) Volker Lendecke 2007
 
5
   
 
6
   This program is free software; you can redistribute it and/or modify
 
7
   it under the terms of the GNU General Public License as published by
 
8
   the Free Software Foundation; either version 3 of the License, or
 
9
   (at your option) any later version.
 
10
   
 
11
   This program is distributed in the hope that it will be useful,
 
12
   but WITHOUT ANY WARRANTY; without even the implied warranty of
 
13
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
14
   GNU General Public License for more details.
 
15
   
 
16
   You should have received a copy of the GNU General Public License
 
17
   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
18
*/
 
19
 
 
20
#include "includes.h"
 
21
 
 
22
static struct db_context *connections_db_ctx(bool rw)
 
23
{
 
24
        static struct db_context *db_ctx;
 
25
 
 
26
        if (db_ctx != NULL) {
 
27
                return db_ctx;
 
28
        }
 
29
 
 
30
        if (rw) {
 
31
                db_ctx = db_open(NULL, lock_path("connections.tdb"), 0,
 
32
                                 TDB_CLEAR_IF_FIRST|TDB_DEFAULT, 
 
33
                                 O_RDWR | O_CREAT, 0644);
 
34
        }
 
35
        else {
 
36
                db_ctx = db_open(NULL, lock_path("connections.tdb"), 0,
 
37
                                 TDB_CLEAR_IF_FIRST|TDB_DEFAULT, O_RDONLY, 0);
 
38
        }
 
39
 
 
40
        return db_ctx;
 
41
}
 
42
 
 
43
struct db_record *connections_fetch_record(TALLOC_CTX *mem_ctx,
 
44
                                           TDB_DATA key)
 
45
{
 
46
        struct db_context *ctx = connections_db_ctx(True);
 
47
 
 
48
        if (ctx == NULL) {
 
49
                return NULL;
 
50
        }
 
51
 
 
52
        return ctx->fetch_locked(ctx, mem_ctx, key);
 
53
}
 
54
 
 
55
struct db_record *connections_fetch_entry(TALLOC_CTX *mem_ctx,
 
56
                                          connection_struct *conn,
 
57
                                          const char *name)
 
58
{
 
59
        struct connections_key ckey;
 
60
        TDB_DATA key;
 
61
 
 
62
        ZERO_STRUCT(ckey);
 
63
        ckey.pid = procid_self();
 
64
        ckey.cnum = conn ? conn->cnum : -1;
 
65
        strlcpy(ckey.name, name, sizeof(ckey.name));
 
66
 
 
67
        key.dsize = sizeof(ckey);
 
68
        key.dptr = (uint8 *)&ckey;
 
69
 
 
70
        return connections_fetch_record(mem_ctx, key);
 
71
}
 
72
 
 
73
struct conn_traverse_state {
 
74
        int (*fn)(struct db_record *rec,
 
75
                  const struct connections_key *key,
 
76
                  const struct connections_data *data,
 
77
                  void *private_data);
 
78
        void *private_data;
 
79
};
 
80
 
 
81
static int conn_traverse_fn(struct db_record *rec, void *private_data)
 
82
{
 
83
        struct conn_traverse_state *state =
 
84
                (struct conn_traverse_state *)private_data;
 
85
 
 
86
        if ((rec->key.dsize != sizeof(struct connections_key))
 
87
            || (rec->value.dsize != sizeof(struct connections_data))) {
 
88
                return 0;
 
89
        }
 
90
 
 
91
        return state->fn(rec, (const struct connections_key *)rec->key.dptr,
 
92
                         (const struct connections_data *)rec->value.dptr,
 
93
                         state->private_data);
 
94
}
 
95
 
 
96
int connections_traverse(int (*fn)(struct db_record *rec,
 
97
                                   void *private_data),
 
98
                         void *private_data)
 
99
{
 
100
        struct db_context *ctx = connections_db_ctx(False);
 
101
 
 
102
        if (ctx == NULL) {
 
103
                return -1;
 
104
        }
 
105
 
 
106
        return ctx->traverse(ctx, fn, private_data);
 
107
}
 
108
 
 
109
int connections_forall(int (*fn)(struct db_record *rec,
 
110
                                 const struct connections_key *key,
 
111
                                 const struct connections_data *data,
 
112
                                 void *private_data),
 
113
                       void *private_data)
 
114
{
 
115
        struct conn_traverse_state state;
 
116
 
 
117
        state.fn = fn;
 
118
        state.private_data = private_data;
 
119
 
 
120
        return connections_traverse(conn_traverse_fn, (void *)&state);
 
121
}
 
122
 
 
123
bool connections_init(bool rw)
 
124
{
 
125
        return (connections_db_ctx(rw) != NULL);
 
126
}