1/* 2 ldb database library 3 4 Copyright (C) Andrew Tridgell 2004 5 6 ** NOTE! The following LGPL license applies to the ldb 7 ** library. This does NOT imply that all of Samba is released 8 ** under the LGPL 9 10 This library is free software; you can redistribute it and/or 11 modify it under the terms of the GNU Lesser General Public 12 License as published by the Free Software Foundation; either 13 version 3 of the License, or (at your option) any later version. 14 15 This library is distributed in the hope that it will be useful, 16 but WITHOUT ANY WARRANTY; without even the implied warranty of 17 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 18 Lesser General Public License for more details. 19 20 You should have received a copy of the GNU Lesser General Public 21 License along with this library; if not, see <http://www.gnu.org/licenses/>. 22*/ 23 24/* 25 * Name: ldb 26 * 27 * Component: ldbadd 28 * 29 * Description: utility to add records - modelled on ldapadd 30 * 31 * Author: Andrew Tridgell 32 */ 33 34#include "ldb.h" 35#include "tools/cmdline.h" 36 37static int failures; 38 39static void usage(void) 40{ 41 printf("Usage: ldbadd <options> <ldif...>\n"); 42 printf("Adds records to a ldb, reading ldif the specified list of files\n\n"); 43 ldb_cmdline_help("ldbadd", stdout); 44 exit(1); 45} 46 47 48/* 49 add records from an opened file 50*/ 51static int process_file(struct ldb_context *ldb, FILE *f, int *count) 52{ 53 struct ldb_ldif *ldif; 54 int ret = LDB_SUCCESS; 55 56 while ((ldif = ldb_ldif_read_file(ldb, f))) { 57 if (ldif->changetype != LDB_CHANGETYPE_ADD && 58 ldif->changetype != LDB_CHANGETYPE_NONE) { 59 fprintf(stderr, "Only CHANGETYPE_ADD records allowed\n"); 60 break; 61 } 62 63 ldif->msg = ldb_msg_canonicalize(ldb, ldif->msg); 64 65 ret = ldb_add(ldb, ldif->msg); 66 if (ret != LDB_SUCCESS) { 67 fprintf(stderr, "ERR: \"%s\" on DN %s\n", 68 ldb_errstring(ldb), ldb_dn_get_linearized(ldif->msg->dn)); 69 failures++; 70 } else { 71 (*count)++; 72 } 73 ldb_ldif_read_free(ldb, ldif); 74 } 75 76 return ret; 77} 78 79 80 81int main(int argc, const char **argv) 82{ 83 struct ldb_context *ldb; 84 int i, ret=0, count=0; 85 struct ldb_cmdline *options; 86 87 ldb = ldb_init(NULL, NULL); 88 89 options = ldb_cmdline_process(ldb, argc, argv, usage); 90 91 if (ldb_transaction_start(ldb) != 0) { 92 printf("Failed to start transaction: %s\n", ldb_errstring(ldb)); 93 exit(1); 94 } 95 96 if (options->argc == 0) { 97 ret = process_file(ldb, stdin, &count); 98 } else { 99 for (i=0;i<options->argc;i++) { 100 const char *fname = options->argv[i]; 101 FILE *f; 102 f = fopen(fname, "r"); 103 if (!f) { 104 perror(fname); 105 exit(1); 106 } 107 ret = process_file(ldb, f, &count); 108 fclose(f); 109 } 110 } 111 112 if (count != 0) { 113 if (ldb_transaction_commit(ldb) != 0) { 114 printf("Failed to commit transaction: %s\n", ldb_errstring(ldb)); 115 exit(1); 116 } 117 } else { 118 ldb_transaction_cancel(ldb); 119 } 120 121 talloc_free(ldb); 122 123 printf("Added %d records with %d failures\n", count, failures); 124 125 return ret; 126} 127