2 Unix SMB/CIFS implementation.
3 Low-level connections.tdb access functions
4 Copyright (C) Volker Lendecke 2007
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.
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.
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/>.
21 #include "system/filesys.h"
22 #include "smbd/globals.h"
23 #include "dbwrap/dbwrap.h"
24 #include "dbwrap/dbwrap_open.h"
25 #include "dbwrap/dbwrap_rbt.h"
27 #include "lib/conn_tdb.h"
30 static struct db_context
*connections_db_ctx(bool rw
)
32 static struct db_context
*db_ctx
;
39 open_flags
= rw
? (O_RDWR
|O_CREAT
) : O_RDONLY
;
41 db_ctx
= db_open(NULL
, lock_path("connections.tdb"), 0,
42 TDB_CLEAR_IF_FIRST
|TDB_INCOMPATIBLE_HASH
|TDB_DEFAULT
,
43 open_flags
, 0644, DBWRAP_LOCK_ORDER_1
);
47 static struct db_record
*connections_fetch_record(TALLOC_CTX
*mem_ctx
,
50 struct db_context
*ctx
= connections_db_ctx(True
);
56 return dbwrap_fetch_locked(ctx
, mem_ctx
, key
);
59 struct db_record
*connections_fetch_entry_ext(TALLOC_CTX
*mem_ctx
,
64 struct connections_key ckey
;
70 strlcpy(ckey
.name
, name
, sizeof(ckey
.name
));
72 key
.dsize
= sizeof(ckey
);
73 key
.dptr
= (uint8
*)&ckey
;
75 return connections_fetch_record(mem_ctx
, key
);
78 struct db_record
*connections_fetch_entry(TALLOC_CTX
*mem_ctx
,
79 connection_struct
*conn
,
82 struct server_id id
= messaging_server_id(conn
->sconn
->msg_ctx
);
83 return connections_fetch_entry_ext(mem_ctx
, id
, conn
->cnum
, name
);
86 struct connections_forall_state
{
87 struct db_context
*session_by_pid
;
88 int (*fn
)(const struct connections_key
*key
,
89 const struct connections_data
*data
,
95 struct connections_forall_session
{
98 char machine
[FSTRING_LEN
];
99 char addr
[FSTRING_LEN
];
102 static int collect_sessions_fn(struct smbXsrv_session_global0
*global
,
103 void *connections_forall_state
)
106 struct connections_forall_state
*state
=
107 (struct connections_forall_state
*)connections_forall_state
;
109 uint32_t id
= global
->session_global_id
;
110 struct connections_forall_session sess
;
112 sess
.uid
= global
->auth_session_info
->unix_token
->uid
;
113 sess
.gid
= global
->auth_session_info
->unix_token
->gid
;
114 strncpy(sess
.machine
, global
->channels
[0].remote_name
, sizeof(sess
.machine
));
115 strncpy(sess
.addr
, global
->channels
[0].remote_address
, sizeof(sess
.addr
));
117 status
= dbwrap_store(state
->session_by_pid
,
118 make_tdb_data((void*)&id
, sizeof(id
)),
119 make_tdb_data((void*)&sess
, sizeof(sess
)),
121 if (!NT_STATUS_IS_OK(status
)) {
122 DEBUG(0, ("Failed to store record: %s\n", nt_errstr(status
)));
127 static int traverse_tcon_fn(struct smbXsrv_tcon_global0
*global
,
128 void *connections_forall_state
)
131 struct connections_forall_state
*state
=
132 (struct connections_forall_state
*)connections_forall_state
;
134 struct connections_key key
;
135 struct connections_data data
;
137 uint32_t sess_id
= global
->session_global_id
;
138 struct connections_forall_session sess
= {
143 TDB_DATA val
= tdb_null
;
145 status
= dbwrap_fetch(state
->session_by_pid
, state
,
146 make_tdb_data((void*)&sess_id
, sizeof(sess_id
)),
148 if (NT_STATUS_IS_OK(status
)) {
149 memcpy((uint8_t *)&sess
, val
.dptr
, val
.dsize
);
155 key
.pid
= data
.pid
= global
->server_id
;
156 key
.cnum
= data
.cnum
= global
->tcon_global_id
;
157 strncpy(key
.name
, global
->share_name
, sizeof(key
.name
));
158 strncpy(data
.servicename
, global
->share_name
, sizeof(data
.servicename
));
161 strncpy(data
.addr
, sess
.addr
, sizeof(data
.addr
));
162 strncpy(data
.machine
, sess
.machine
, sizeof(data
.machine
));
163 data
.start
= nt_time_to_unix(global
->creation_time
);
167 return state
->fn(&key
, &data
, state
->private_data
);
170 int connections_forall_read(int (*fn
)(const struct connections_key
*key
,
171 const struct connections_data
*data
,
175 TALLOC_CTX
*frame
= talloc_stackframe();
176 struct connections_forall_state
*state
=
177 talloc_zero(talloc_tos(), struct connections_forall_state
);
181 state
->session_by_pid
= db_open_rbt(state
);
183 state
->private_data
= private_data
;
184 status
= smbXsrv_session_global_traverse(collect_sessions_fn
, state
);
185 if (!NT_STATUS_IS_OK(status
)) {
186 DEBUG(0, ("Failed to traverse sessions: %s\n",
191 status
= smbXsrv_tcon_global_traverse(traverse_tcon_fn
, state
);
192 if (!NT_STATUS_IS_OK(status
)) {
193 DEBUG(0, ("Failed to traverse tree connects: %s\n",
203 bool connections_init(bool rw
)
205 return (connections_db_ctx(rw
) != NULL
);