2 Unix SMB/CIFS implementation.
6 Copyright (C) Andrew Tridgell 2005
7 Copyright (C) Stefan Metzmacher 2005
9 This program is free software; you can redistribute it and/or modify
10 it under the terms of the GNU General Public License as published by
11 the Free Software Foundation; either version 2 of the License, or
12 (at your option) any later version.
14 This program is distributed in the hope that it will be useful,
15 but WITHOUT ANY WARRANTY; without even the implied warranty of
16 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 GNU General Public License for more details.
19 You should have received a copy of the GNU General Public License
20 along with this program; if not, write to the Free Software
21 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
25 #include "nbt_server/nbt_server.h"
26 #include "nbt_server/wins/winsdb.h"
27 #include "lib/ldb/include/ldb.h"
28 #include "lib/ldb/include/ldb_errors.h"
29 #include "system/time.h"
31 uint64_t winsdb_get_maxVersion(struct winsdb_handle
*h
)
34 struct ldb_context
*ldb
= h
->ldb
;
36 struct ldb_result
*res
= NULL
;
37 TALLOC_CTX
*tmp_ctx
= talloc_new(ldb
);
38 uint64_t maxVersion
= 0;
40 dn
= ldb_dn_explode(tmp_ctx
, "CN=VERSION");
43 /* find the record in the WINS database */
44 ret
= ldb_search(ldb
, dn
, LDB_SCOPE_BASE
,
46 if (ret
!= LDB_SUCCESS
) goto failed
;
47 talloc_steal(tmp_ctx
, res
);
48 if (res
->count
> 1) goto failed
;
50 if (res
->count
== 1) {
51 maxVersion
= ldb_msg_find_uint64(res
->msgs
[0], "maxVersion", 0);
60 if newVersion == 0 return the old maxVersion + 1 and save it
61 if newVersion > 0 return MAX(oldMaxVersion, newMaxVersion) and save it
63 uint64_t winsdb_set_maxVersion(struct winsdb_handle
*h
, uint64_t newMaxVersion
)
68 struct ldb_result
*res
= NULL
;
69 struct ldb_message
*msg
= NULL
;
70 struct ldb_context
*wins_db
= h
->ldb
;
71 TALLOC_CTX
*tmp_ctx
= talloc_new(wins_db
);
72 uint64_t oldMaxVersion
= 0;
74 trans
= ldb_transaction_start(wins_db
);
75 if (trans
!= LDB_SUCCESS
) goto failed
;
77 dn
= ldb_dn_explode(tmp_ctx
, "CN=VERSION");
80 /* find the record in the WINS database */
81 ret
= ldb_search(wins_db
, dn
, LDB_SCOPE_BASE
, NULL
, NULL
, &res
);
83 if (ret
!= LDB_SUCCESS
) goto failed
;
84 if (res
->count
> 1) goto failed
;
86 talloc_steal(tmp_ctx
, res
);
88 if (res
->count
== 1) {
89 oldMaxVersion
= ldb_msg_find_uint64(res
->msgs
[0], "maxVersion", 0);
92 if (newMaxVersion
== 0) {
93 newMaxVersion
= oldMaxVersion
+ 1;
95 newMaxVersion
= MAX(oldMaxVersion
, newMaxVersion
);
98 msg
= ldb_msg_new(tmp_ctx
);
99 if (!msg
) goto failed
;
103 ret
= ldb_msg_add_empty(msg
, "objectClass", LDB_FLAG_MOD_REPLACE
);
104 if (ret
!= 0) goto failed
;
105 ret
= ldb_msg_add_string(msg
, "objectClass", "winsMaxVersion");
106 if (ret
!= 0) goto failed
;
107 ret
= ldb_msg_add_empty(msg
, "maxVersion", LDB_FLAG_MOD_REPLACE
);
108 if (ret
!= 0) goto failed
;
109 ret
= ldb_msg_add_fmt(msg
, "maxVersion", "%llu", (long long)newMaxVersion
);
110 if (ret
!= 0) goto failed
;
112 ret
= ldb_modify(wins_db
, msg
);
113 if (ret
!= 0) ret
= ldb_add(wins_db
, msg
);
114 if (ret
!= 0) goto failed
;
116 trans
= ldb_transaction_commit(wins_db
);
117 if (trans
!= LDB_SUCCESS
) goto failed
;
119 talloc_free(tmp_ctx
);
120 return newMaxVersion
;
123 if (trans
== LDB_SUCCESS
) ldb_transaction_cancel(wins_db
);
124 talloc_free(tmp_ctx
);
128 uint64_t winsdb_get_seqnumber(struct winsdb_handle
*h
)
131 struct ldb_context
*ldb
= h
->ldb
;
133 struct ldb_result
*res
= NULL
;
134 TALLOC_CTX
*tmp_ctx
= talloc_new(ldb
);
135 uint64_t seqnumber
= 0;
137 dn
= ldb_dn_explode(tmp_ctx
, "@BASEINFO");
138 if (!dn
) goto failed
;
140 /* find the record in the WINS database */
141 ret
= ldb_search(ldb
, dn
, LDB_SCOPE_BASE
,
143 if (ret
!= LDB_SUCCESS
) goto failed
;
144 talloc_steal(tmp_ctx
, res
);
145 if (res
->count
> 1) goto failed
;
147 if (res
->count
== 1) {
148 seqnumber
= ldb_msg_find_uint64(res
->msgs
[0], "sequenceNumber", 0);
152 talloc_free(tmp_ctx
);
157 return a DN for a nbt_name
159 static struct ldb_dn
*winsdb_dn(TALLOC_CTX
*mem_ctx
, struct nbt_name
*name
)
163 dn
= ldb_dn_string_compose(mem_ctx
, NULL
, "type=0x%02X", name
->type
);
164 if (dn
&& name
->name
&& *name
->name
) {
165 dn
= ldb_dn_string_compose(mem_ctx
, dn
, "name=%s", name
->name
);
167 if (dn
&& name
->scope
&& *name
->scope
) {
168 dn
= ldb_dn_string_compose(mem_ctx
, dn
, "scope=%s", name
->scope
);
173 static NTSTATUS
winsdb_nbt_name(TALLOC_CTX
*mem_ctx
, struct ldb_dn
*dn
, struct nbt_name
**_name
)
176 struct nbt_name
*name
;
179 name
= talloc(mem_ctx
, struct nbt_name
);
181 status
= NT_STATUS_NO_MEMORY
;
185 if (dn
->comp_num
> 3) {
186 status
= NT_STATUS_INTERNAL_DB_CORRUPTION
;
190 if (dn
->comp_num
> cur
&& strcasecmp("scope", dn
->components
[cur
].name
) == 0) {
191 name
->scope
= talloc_steal(name
, dn
->components
[cur
].value
.data
);
197 if (dn
->comp_num
> cur
&& strcasecmp("name", dn
->components
[cur
].name
) == 0) {
198 name
->name
= talloc_steal(name
, dn
->components
[cur
].value
.data
);
201 name
->name
= talloc_strdup(name
, "");
203 status
= NT_STATUS_NO_MEMORY
;
208 if (dn
->comp_num
> cur
&& strcasecmp("type", dn
->components
[cur
].name
) == 0) {
209 name
->type
= strtoul((char *)dn
->components
[cur
].value
.data
, NULL
, 0);
212 status
= NT_STATUS_INTERNAL_DB_CORRUPTION
;
224 decode the winsdb_addr("address") attribute:
226 "172.31.1.1;winsOwner:172.31.9.202;expireTime:20050923032330.0Z;"
229 static NTSTATUS
winsdb_addr_decode(struct winsdb_handle
*h
, struct winsdb_record
*rec
, struct ldb_val
*val
,
230 TALLOC_CTX
*mem_ctx
, struct winsdb_addr
**_addr
)
233 struct winsdb_addr
*addr
;
235 const char *wins_owner
;
236 const char *expire_time
;
239 addr
= talloc(mem_ctx
, struct winsdb_addr
);
241 status
= NT_STATUS_NO_MEMORY
;
245 address
= (char *)val
->data
;
247 p
= strchr(address
, ';');
249 /* support old entries, with only the address */
250 addr
->address
= talloc_steal(addr
, val
->data
);
251 addr
->wins_owner
= talloc_reference(addr
, rec
->wins_owner
);
252 if (!addr
->wins_owner
) {
253 status
= NT_STATUS_NO_MEMORY
;
256 addr
->expire_time
= rec
->expire_time
;
262 addr
->address
= talloc_strdup(addr
, address
);
263 if (!addr
->address
) {
264 status
= NT_STATUS_NO_MEMORY
;
268 if (strncmp("winsOwner:", p
, 10) != 0) {
269 status
= NT_STATUS_INTERNAL_DB_CORRUPTION
;
273 p
= strchr(wins_owner
, ';');
275 status
= NT_STATUS_INTERNAL_DB_CORRUPTION
;
280 if (strcmp(wins_owner
, "0.0.0.0") == 0) {
281 wins_owner
= h
->local_owner
;
283 addr
->wins_owner
= talloc_strdup(addr
, wins_owner
);
284 if (!addr
->wins_owner
) {
285 status
= NT_STATUS_NO_MEMORY
;
289 if (strncmp("expireTime:", p
, 11) != 0) {
290 status
= NT_STATUS_INTERNAL_DB_CORRUPTION
;
294 expire_time
= p
+ 11;
295 p
= strchr(expire_time
, ';');
297 status
= NT_STATUS_INTERNAL_DB_CORRUPTION
;
302 addr
->expire_time
= ldb_string_to_time(expire_time
);
312 encode the winsdb_addr("address") attribute like this:
314 "172.31.1.1;winsOwner:172.31.9.202;expireTime:20050923032330.0Z;"
318 static int ldb_msg_add_winsdb_addr(struct ldb_message
*msg
, struct winsdb_record
*rec
,
319 const char *attr_name
, struct winsdb_addr
*addr
)
324 if (rec
->is_static
) {
325 str
= talloc_strdup(msg
, addr
->address
);
329 expire_time
= ldb_timestring(msg
, addr
->expire_time
);
330 if (!expire_time
) return -1;
331 str
= talloc_asprintf(msg
, "%s;winsOwner:%s;expireTime:%s;",
332 addr
->address
, addr
->wins_owner
,
334 talloc_free(expire_time
);
338 val
.data
= discard_const_p(uint8_t, str
);
339 val
.length
= strlen(str
);
341 return ldb_msg_add_value(msg
, attr_name
, &val
);
344 struct winsdb_addr
**winsdb_addr_list_make(TALLOC_CTX
*mem_ctx
)
346 struct winsdb_addr
**addresses
;
348 addresses
= talloc_array(mem_ctx
, struct winsdb_addr
*, 1);
349 if (!addresses
) return NULL
;
356 struct winsdb_addr
**winsdb_addr_list_add(struct winsdb_addr
**addresses
, const char *address
,
357 const char *wins_owner
, time_t expire_time
)
359 size_t len
= winsdb_addr_list_length(addresses
);
361 addresses
= talloc_realloc(addresses
, addresses
, struct winsdb_addr
*, len
+ 2);
362 if (!addresses
) return NULL
;
364 addresses
[len
] = talloc(addresses
, struct winsdb_addr
);
365 if (!addresses
[len
]) {
366 talloc_free(addresses
);
370 addresses
[len
]->address
= talloc_strdup(addresses
[len
], address
);
371 if (!addresses
[len
]->address
) {
372 talloc_free(addresses
);
376 addresses
[len
]->wins_owner
= talloc_strdup(addresses
[len
], wins_owner
);
377 if (!addresses
[len
]->wins_owner
) {
378 talloc_free(addresses
);
382 addresses
[len
]->expire_time
= expire_time
;
384 addresses
[len
+1] = NULL
;
389 void winsdb_addr_list_remove(struct winsdb_addr
**addresses
, const char *address
)
393 for (i
=0; addresses
[i
]; i
++) {
394 if (strcmp(addresses
[i
]->address
, address
) == 0) {
398 if (!addresses
[i
]) return;
400 for (; addresses
[i
]; i
++) {
401 addresses
[i
] = addresses
[i
+1];
407 struct winsdb_addr
*winsdb_addr_list_check(struct winsdb_addr
**addresses
, const char *address
)
411 for (i
=0; addresses
[i
]; i
++) {
412 if (strcmp(addresses
[i
]->address
, address
) == 0) {
420 size_t winsdb_addr_list_length(struct winsdb_addr
**addresses
)
423 for (i
=0; addresses
[i
]; i
++);
427 const char **winsdb_addr_string_list(TALLOC_CTX
*mem_ctx
, struct winsdb_addr
**addresses
)
429 size_t len
= winsdb_addr_list_length(addresses
);
430 const char **str_list
=NULL
;
433 for (i
=0; i
< len
; i
++) {
434 str_list
= str_list_add(str_list
, addresses
[i
]->address
);
439 talloc_steal(mem_ctx
, str_list
);
444 load a WINS entry from the database
446 NTSTATUS
winsdb_lookup(struct winsdb_handle
*h
,
447 struct nbt_name
*name
,
449 struct winsdb_record
**_rec
)
452 struct ldb_result
*res
= NULL
;
454 struct winsdb_record
*rec
;
455 struct ldb_context
*wins_db
= h
->ldb
;
456 TALLOC_CTX
*tmp_ctx
= talloc_new(mem_ctx
);
457 time_t now
= time(NULL
);
459 /* find the record in the WINS database */
460 ret
= ldb_search(wins_db
, winsdb_dn(tmp_ctx
, name
), LDB_SCOPE_BASE
,
463 if (ret
!= LDB_SUCCESS
|| res
->count
> 1) {
464 status
= NT_STATUS_INTERNAL_DB_CORRUPTION
;
466 } else if (res
->count
== 0) {
467 status
= NT_STATUS_OBJECT_NAME_NOT_FOUND
;
471 talloc_steal(tmp_ctx
, res
);
473 status
= winsdb_record(h
, res
->msgs
[0], tmp_ctx
, now
, &rec
);
474 if (!NT_STATUS_IS_OK(status
)) goto failed
;
476 talloc_steal(mem_ctx
, rec
);
477 talloc_free(tmp_ctx
);
482 talloc_free(tmp_ctx
);
486 NTSTATUS
winsdb_record(struct winsdb_handle
*h
, struct ldb_message
*msg
, TALLOC_CTX
*mem_ctx
, time_t now
, struct winsdb_record
**_rec
)
489 struct winsdb_record
*rec
;
490 struct ldb_message_element
*el
;
491 struct nbt_name
*name
;
492 uint32_t i
, j
, num_values
;
494 rec
= talloc(mem_ctx
, struct winsdb_record
);
496 status
= NT_STATUS_NO_MEMORY
;
500 status
= winsdb_nbt_name(rec
, msg
->dn
, &name
);
501 if (!NT_STATUS_IS_OK(status
)) goto failed
;
503 if (strlen(name
->name
) > 15) {
504 status
= NT_STATUS_INTERNAL_DB_CORRUPTION
;
507 if (name
->scope
&& strlen(name
->scope
) > 238) {
508 status
= NT_STATUS_INTERNAL_DB_CORRUPTION
;
512 /* parse it into a more convenient winsdb_record structure */
514 rec
->type
= ldb_msg_find_int(msg
, "recordType", WREPL_TYPE_UNIQUE
);
515 rec
->state
= ldb_msg_find_int(msg
, "recordState", WREPL_STATE_RELEASED
);
516 rec
->node
= ldb_msg_find_int(msg
, "nodeType", WREPL_NODE_B
);
517 rec
->is_static
= ldb_msg_find_int(msg
, "isStatic", 0);
518 rec
->expire_time
= ldb_string_to_time(ldb_msg_find_string(msg
, "expireTime", NULL
));
519 rec
->version
= ldb_msg_find_uint64(msg
, "versionID", 0);
520 rec
->wins_owner
= ldb_msg_find_string(msg
, "winsOwner", NULL
);
521 rec
->registered_by
= ldb_msg_find_string(msg
, "registeredBy", NULL
);
522 talloc_steal(rec
, rec
->wins_owner
);
523 talloc_steal(rec
, rec
->registered_by
);
525 if (!rec
->wins_owner
|| strcmp(rec
->wins_owner
, "0.0.0.0") == 0) {
526 rec
->wins_owner
= h
->local_owner
;
529 el
= ldb_msg_find_element(msg
, "address");
531 num_values
= el
->num_values
;
536 if (rec
->type
== WREPL_TYPE_UNIQUE
|| rec
->type
== WREPL_TYPE_GROUP
) {
537 if (num_values
!= 1) {
538 status
= NT_STATUS_INTERNAL_DB_CORRUPTION
;
542 if (rec
->state
== WREPL_STATE_ACTIVE
) {
543 if (num_values
< 1) {
544 status
= NT_STATUS_INTERNAL_DB_CORRUPTION
;
548 if (num_values
> 25) {
549 status
= NT_STATUS_INTERNAL_DB_CORRUPTION
;
553 /* see if it has already expired */
554 if (!rec
->is_static
&&
555 rec
->expire_time
<= now
&&
556 rec
->state
== WREPL_STATE_ACTIVE
) {
557 DEBUG(5,("WINS: expiring name %s (expired at %s)\n",
558 nbt_name_string(mem_ctx
, rec
->name
), timestring(mem_ctx
, rec
->expire_time
)));
559 rec
->state
= WREPL_STATE_RELEASED
;
562 rec
->addresses
= talloc_array(rec
, struct winsdb_addr
*, num_values
+1);
563 if (rec
->addresses
== NULL
) {
564 status
= NT_STATUS_NO_MEMORY
;
568 for (i
=0,j
=0;i
<num_values
;i
++) {
569 status
= winsdb_addr_decode(h
, rec
, &el
->values
[i
], rec
->addresses
, &rec
->addresses
[j
]);
570 if (!NT_STATUS_IS_OK(status
)) goto failed
;
573 * the record isn't static and is active
574 * then don't add the address if it's expired
576 if (!rec
->is_static
&&
577 rec
->addresses
[j
]->expire_time
<= now
&&
578 rec
->state
== WREPL_STATE_ACTIVE
) {
579 DEBUG(5,("WINS: expiring name addr %s of %s (expired at %s)\n",
580 rec
->addresses
[j
]->address
, nbt_name_string(rec
->addresses
[j
], rec
->name
),
581 timestring(rec
->addresses
[j
], rec
->addresses
[j
]->expire_time
)));
582 talloc_free(rec
->addresses
[j
]);
583 rec
->addresses
[j
] = NULL
;
588 rec
->addresses
[j
] = NULL
;
591 if (rec
->is_static
&& rec
->state
== WREPL_STATE_ACTIVE
) {
592 rec
->expire_time
= get_time_t_max();
593 for (i
=0;rec
->addresses
[i
];i
++) {
594 rec
->addresses
[i
]->expire_time
= rec
->expire_time
;
598 if (rec
->state
== WREPL_STATE_ACTIVE
) {
599 if (num_values
< 1) {
600 DEBUG(5,("WINS: expiring name %s (because it has no active addresses)\n",
601 nbt_name_string(mem_ctx
, rec
->name
)));
602 rec
->state
= WREPL_STATE_RELEASED
;
609 if (NT_STATUS_EQUAL(NT_STATUS_INTERNAL_DB_CORRUPTION
, status
)) {
610 DEBUG(1,("winsdb_record: corrupted record: %s\n", ldb_dn_linearize(rec
, msg
->dn
)));
617 form a ldb_message from a winsdb_record
619 struct ldb_message
*winsdb_message(struct ldb_context
*ldb
,
620 struct winsdb_record
*rec
, TALLOC_CTX
*mem_ctx
)
624 const char *expire_time
;
625 struct ldb_message
*msg
= ldb_msg_new(mem_ctx
);
626 if (msg
== NULL
) goto failed
;
628 /* make sure we don't put in corrupted records */
629 addr_count
= winsdb_addr_list_length(rec
->addresses
);
630 if (rec
->state
== WREPL_STATE_ACTIVE
&& addr_count
== 0) {
631 rec
->state
= WREPL_STATE_RELEASED
;
633 if (rec
->type
== WREPL_TYPE_UNIQUE
&& addr_count
> 1) {
634 rec
->type
= WREPL_TYPE_MHOMED
;
637 expire_time
= ldb_timestring(msg
, rec
->expire_time
);
642 msg
->dn
= winsdb_dn(msg
, rec
->name
);
643 if (msg
->dn
== NULL
) goto failed
;
644 ret
|= ldb_msg_add_fmt(msg
, "type", "0x%02X", rec
->name
->type
);
645 if (rec
->name
->name
&& *rec
->name
->name
) {
646 ret
|= ldb_msg_add_string(msg
, "name", rec
->name
->name
);
648 if (rec
->name
->scope
&& *rec
->name
->scope
) {
649 ret
|= ldb_msg_add_string(msg
, "scope", rec
->name
->scope
);
651 ret
|= ldb_msg_add_fmt(msg
, "objectClass", "winsRecord");
652 ret
|= ldb_msg_add_fmt(msg
, "recordType", "%u", rec
->type
);
653 ret
|= ldb_msg_add_fmt(msg
, "recordState", "%u", rec
->state
);
654 ret
|= ldb_msg_add_fmt(msg
, "nodeType", "%u", rec
->node
);
655 ret
|= ldb_msg_add_fmt(msg
, "isStatic", "%u", rec
->is_static
);
656 ret
|= ldb_msg_add_empty(msg
, "expireTime", 0);
657 if (!(rec
->is_static
&& rec
->state
== WREPL_STATE_ACTIVE
)) {
658 ret
|= ldb_msg_add_string(msg
, "expireTime", expire_time
);
660 ret
|= ldb_msg_add_fmt(msg
, "versionID", "%llu", (long long)rec
->version
);
661 ret
|= ldb_msg_add_string(msg
, "winsOwner", rec
->wins_owner
);
662 ret
|= ldb_msg_add_empty(msg
, "address", 0);
663 for (i
=0;rec
->addresses
[i
];i
++) {
664 ret
|= ldb_msg_add_winsdb_addr(msg
, rec
, "address", rec
->addresses
[i
]);
666 ret
|= ldb_msg_add_empty(msg
, "registeredBy", 0);
667 if (rec
->registered_by
) {
668 ret
|= ldb_msg_add_string(msg
, "registeredBy", rec
->registered_by
);
669 if (ret
!= 0) goto failed
;
679 save a WINS record into the database
681 uint8_t winsdb_add(struct winsdb_handle
*h
, struct winsdb_record
*rec
, uint32_t flags
)
683 struct ldb_message
*msg
;
684 struct ldb_context
*wins_db
= h
->ldb
;
685 TALLOC_CTX
*tmp_ctx
= talloc_new(wins_db
);
689 trans
= ldb_transaction_start(wins_db
);
690 if (trans
!= LDB_SUCCESS
) goto failed
;
692 if (flags
& WINSDB_FLAG_ALLOC_VERSION
) {
693 /* passing '0' means auto-allocate a new one */
694 rec
->version
= winsdb_set_maxVersion(h
, 0);
695 if (rec
->version
== 0) goto failed
;
697 if (flags
& WINSDB_FLAG_TAKE_OWNERSHIP
) {
698 rec
->wins_owner
= h
->local_owner
;
701 msg
= winsdb_message(wins_db
, rec
, tmp_ctx
);
702 if (msg
== NULL
) goto failed
;
703 ret
= ldb_add(wins_db
, msg
);
704 if (ret
!= 0) goto failed
;
706 trans
= ldb_transaction_commit(wins_db
);
707 if (trans
!= LDB_SUCCESS
) goto failed
;
709 wins_hook(h
, rec
, WINS_HOOK_ADD
);
711 talloc_free(tmp_ctx
);
715 if (trans
== LDB_SUCCESS
) ldb_transaction_cancel(wins_db
);
716 talloc_free(tmp_ctx
);
717 return NBT_RCODE_SVR
;
722 modify a WINS record in the database
724 uint8_t winsdb_modify(struct winsdb_handle
*h
, struct winsdb_record
*rec
, uint32_t flags
)
726 struct ldb_message
*msg
;
727 struct ldb_context
*wins_db
= h
->ldb
;
728 TALLOC_CTX
*tmp_ctx
= talloc_new(wins_db
);
733 trans
= ldb_transaction_start(wins_db
);
734 if (trans
!= LDB_SUCCESS
) goto failed
;
736 if (flags
& WINSDB_FLAG_ALLOC_VERSION
) {
737 /* passing '0' means auto-allocate a new one */
738 rec
->version
= winsdb_set_maxVersion(h
, 0);
739 if (rec
->version
== 0) goto failed
;
741 if (flags
& WINSDB_FLAG_TAKE_OWNERSHIP
) {
742 rec
->wins_owner
= h
->local_owner
;
745 msg
= winsdb_message(wins_db
, rec
, tmp_ctx
);
746 if (msg
== NULL
) goto failed
;
748 for (i
=0;i
<msg
->num_elements
;i
++) {
749 msg
->elements
[i
].flags
= LDB_FLAG_MOD_REPLACE
;
752 ret
= ldb_modify(wins_db
, msg
);
753 if (ret
!= 0) goto failed
;
755 trans
= ldb_transaction_commit(wins_db
);
756 if (trans
!= LDB_SUCCESS
) goto failed
;
758 wins_hook(h
, rec
, WINS_HOOK_MODIFY
);
760 talloc_free(tmp_ctx
);
764 if (trans
== LDB_SUCCESS
) ldb_transaction_cancel(wins_db
);
765 talloc_free(tmp_ctx
);
766 return NBT_RCODE_SVR
;
771 delete a WINS record from the database
773 uint8_t winsdb_delete(struct winsdb_handle
*h
, struct winsdb_record
*rec
)
775 struct ldb_context
*wins_db
= h
->ldb
;
776 TALLOC_CTX
*tmp_ctx
= talloc_new(wins_db
);
777 const struct ldb_dn
*dn
;
781 trans
= ldb_transaction_start(wins_db
);
782 if (trans
!= LDB_SUCCESS
) goto failed
;
784 dn
= winsdb_dn(tmp_ctx
, rec
->name
);
785 if (dn
== NULL
) goto failed
;
787 ret
= ldb_delete(wins_db
, dn
);
788 if (ret
!= 0) goto failed
;
790 trans
= ldb_transaction_commit(wins_db
);
791 if (trans
!= LDB_SUCCESS
) goto failed
;
793 wins_hook(h
, rec
, WINS_HOOK_DELETE
);
795 talloc_free(tmp_ctx
);
799 if (trans
== LDB_SUCCESS
) ldb_transaction_cancel(wins_db
);
800 talloc_free(tmp_ctx
);
801 return NBT_RCODE_SVR
;
804 static BOOL
winsdb_check_or_add_module_list(struct winsdb_handle
*h
)
809 struct ldb_result
*res
= NULL
;
810 struct ldb_message
*msg
= NULL
;
811 TALLOC_CTX
*tmp_ctx
= talloc_new(h
);
812 unsigned int flags
= 0;
814 trans
= ldb_transaction_start(h
->ldb
);
815 if (trans
!= LDB_SUCCESS
) goto failed
;
817 /* check if we have a special @MODULES record already */
818 dn
= ldb_dn_explode(tmp_ctx
, "@MODULES");
819 if (!dn
) goto failed
;
821 /* find the record in the WINS database */
822 ret
= ldb_search(h
->ldb
, dn
, LDB_SCOPE_BASE
, NULL
, NULL
, &res
);
823 if (ret
!= LDB_SUCCESS
) goto failed
;
824 talloc_steal(tmp_ctx
, res
);
826 if (res
->count
> 0) goto skip
;
828 /* if there's no record, add one */
829 msg
= ldb_msg_new(tmp_ctx
);
830 if (!msg
) goto failed
;
833 ret
= ldb_msg_add_string(msg
, "@LIST", "wins_ldb");
834 if (ret
!= 0) goto failed
;
836 ret
= ldb_add(h
->ldb
, msg
);
837 if (ret
!= 0) goto failed
;
839 trans
= ldb_transaction_commit(h
->ldb
);
840 if (trans
!= LDB_SUCCESS
) goto failed
;
842 /* close and reopen the database, with the modules */
843 trans
= LDB_ERR_OTHER
;
847 if (lp_parm_bool(-1,"winsdb", "nosync", False
)) {
848 flags
|= LDB_FLG_NOSYNC
;
851 h
->ldb
= ldb_wrap_connect(h
, lock_path(h
, lp_wins_url()),
852 NULL
, NULL
, flags
, NULL
);
853 if (!h
->ldb
) goto failed
;
855 talloc_free(tmp_ctx
);
859 if (trans
== LDB_SUCCESS
) ldb_transaction_cancel(h
->ldb
);
860 talloc_free(tmp_ctx
);
864 if (trans
== LDB_SUCCESS
) ldb_transaction_cancel(h
->ldb
);
865 talloc_free(tmp_ctx
);
869 struct winsdb_handle
*winsdb_connect(TALLOC_CTX
*mem_ctx
, enum winsdb_handle_caller caller
)
871 struct winsdb_handle
*h
= NULL
;
873 unsigned int flags
= 0;
877 h
= talloc(mem_ctx
, struct winsdb_handle
);
880 if (lp_parm_bool(-1,"winsdb", "nosync", False
)) {
881 flags
|= LDB_FLG_NOSYNC
;
884 h
->ldb
= ldb_wrap_connect(h
, lock_path(h
, lp_wins_url()),
885 NULL
, NULL
, flags
, NULL
);
886 if (!h
->ldb
) goto failed
;
890 owner
= lp_parm_string(-1, "winsdb", "local_owner");
892 owner
= iface_n_ip(0);
895 h
->local_owner
= talloc_strdup(h
, owner
);
896 if (!h
->local_owner
) goto failed
;
898 /* make sure the module list is available and used */
899 ret
= winsdb_check_or_add_module_list(h
);
900 if (!ret
) goto failed
;
902 ldb_err
= ldb_set_opaque(h
->ldb
, "winsdb_handle", h
);
903 if (ldb_err
!= LDB_SUCCESS
) goto failed
;