wmi-1.3.16 from opsview.com
This commit is contained in:
@@ -0,0 +1,406 @@
|
||||
/*
|
||||
* Copyright (c) 1997 - 2004 Kungliga Tekniska Högskolan
|
||||
* (Royal Institute of Technology, Stockholm, Sweden).
|
||||
* All rights reserved.
|
||||
*
|
||||
* Redistribution and use in source and binary forms, with or without
|
||||
* modification, are permitted provided that the following conditions
|
||||
* are met:
|
||||
*
|
||||
* 1. Redistributions of source code must retain the above copyright
|
||||
* notice, this list of conditions and the following disclaimer.
|
||||
*
|
||||
* 2. Redistributions in binary form must reproduce the above copyright
|
||||
* notice, this list of conditions and the following disclaimer in the
|
||||
* documentation and/or other materials provided with the distribution.
|
||||
*
|
||||
* 3. Neither the name of the Institute nor the names of its contributors
|
||||
* may be used to endorse or promote products derived from this software
|
||||
* without specific prior written permission.
|
||||
*
|
||||
* THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
|
||||
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
* ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
|
||||
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
||||
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
|
||||
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
||||
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
||||
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
||||
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
||||
* SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#include "hdb_locl.h"
|
||||
|
||||
RCSID("$Id: hdb.c,v 1.62 2006/10/06 16:47:22 lha Exp $");
|
||||
|
||||
#ifdef HAVE_DLFCN_H
|
||||
#include <dlfcn.h>
|
||||
#endif
|
||||
|
||||
struct hdb_method {
|
||||
const char *prefix;
|
||||
krb5_error_code (*create)(krb5_context, HDB **, const char *filename);
|
||||
};
|
||||
|
||||
static struct hdb_method methods[] = {
|
||||
#if HAVE_DB1 || HAVE_DB3
|
||||
{"db:", hdb_db_create},
|
||||
#endif
|
||||
#if HAVE_NDBM
|
||||
{"ndbm:", hdb_ndbm_create},
|
||||
#endif
|
||||
#if defined(OPENLDAP) && !defined(OPENLDAP_MODULE)
|
||||
{"ldap:", hdb_ldap_create},
|
||||
{"ldapi:", hdb_ldapi_create},
|
||||
#endif
|
||||
#ifdef _SAMBA_BUILD_
|
||||
{"ldb:", hdb_ldb_create},
|
||||
#endif
|
||||
#ifdef HAVE_LDB /* Used for integrated samba build */
|
||||
{"ldb:", hdb_ldb_create},
|
||||
#endif
|
||||
{NULL, NULL}
|
||||
};
|
||||
|
||||
#if HAVE_DB1 || HAVE_DB3
|
||||
static struct hdb_method dbmetod = {"", hdb_db_create };
|
||||
#elif defined(HAVE_NDBM)
|
||||
static struct hdb_method dbmetod = {"", hdb_ndbm_create };
|
||||
#endif
|
||||
|
||||
|
||||
krb5_error_code
|
||||
hdb_next_enctype2key(krb5_context context,
|
||||
const hdb_entry *e,
|
||||
krb5_enctype enctype,
|
||||
Key **key)
|
||||
{
|
||||
Key *k;
|
||||
|
||||
for (k = *key ? (*key) + 1 : e->keys.val;
|
||||
k < e->keys.val + e->keys.len;
|
||||
k++)
|
||||
if(k->key.keytype == enctype){
|
||||
*key = k;
|
||||
return 0;
|
||||
}
|
||||
return KRB5_PROG_ETYPE_NOSUPP; /* XXX */
|
||||
}
|
||||
|
||||
krb5_error_code
|
||||
hdb_enctype2key(krb5_context context,
|
||||
hdb_entry *e,
|
||||
krb5_enctype enctype,
|
||||
Key **key)
|
||||
{
|
||||
*key = NULL;
|
||||
return hdb_next_enctype2key(context, e, enctype, key);
|
||||
}
|
||||
|
||||
void
|
||||
hdb_free_key(Key *key)
|
||||
{
|
||||
memset(key->key.keyvalue.data,
|
||||
0,
|
||||
key->key.keyvalue.length);
|
||||
free_Key(key);
|
||||
free(key);
|
||||
}
|
||||
|
||||
|
||||
krb5_error_code
|
||||
hdb_lock(int fd, int operation)
|
||||
{
|
||||
int i, code = 0;
|
||||
|
||||
for(i = 0; i < 3; i++){
|
||||
code = flock(fd, (operation == HDB_RLOCK ? LOCK_SH : LOCK_EX) | LOCK_NB);
|
||||
if(code == 0 || errno != EWOULDBLOCK)
|
||||
break;
|
||||
sleep(1);
|
||||
}
|
||||
if(code == 0)
|
||||
return 0;
|
||||
if(errno == EWOULDBLOCK)
|
||||
return HDB_ERR_DB_INUSE;
|
||||
return HDB_ERR_CANT_LOCK_DB;
|
||||
}
|
||||
|
||||
krb5_error_code
|
||||
hdb_unlock(int fd)
|
||||
{
|
||||
int code;
|
||||
code = flock(fd, LOCK_UN);
|
||||
if(code)
|
||||
return 4711 /* XXX */;
|
||||
return 0;
|
||||
}
|
||||
|
||||
void
|
||||
hdb_free_entry(krb5_context context, hdb_entry_ex *ent)
|
||||
{
|
||||
int i;
|
||||
|
||||
if (ent->free_entry)
|
||||
(*ent->free_entry)(context, ent);
|
||||
|
||||
for(i = 0; i < ent->entry.keys.len; ++i) {
|
||||
Key *k = &ent->entry.keys.val[i];
|
||||
|
||||
memset (k->key.keyvalue.data, 0, k->key.keyvalue.length);
|
||||
}
|
||||
free_hdb_entry(&ent->entry);
|
||||
}
|
||||
|
||||
krb5_error_code
|
||||
hdb_foreach(krb5_context context,
|
||||
HDB *db,
|
||||
unsigned flags,
|
||||
hdb_foreach_func_t func,
|
||||
void *data)
|
||||
{
|
||||
krb5_error_code ret;
|
||||
hdb_entry_ex entry;
|
||||
ret = db->hdb_firstkey(context, db, flags, &entry);
|
||||
while(ret == 0){
|
||||
ret = (*func)(context, db, &entry, data);
|
||||
hdb_free_entry(context, &entry);
|
||||
if(ret == 0)
|
||||
ret = db->hdb_nextkey(context, db, flags, &entry);
|
||||
}
|
||||
if(ret == HDB_ERR_NOENTRY)
|
||||
ret = 0;
|
||||
return ret;
|
||||
}
|
||||
|
||||
krb5_error_code
|
||||
hdb_check_db_format(krb5_context context, HDB *db)
|
||||
{
|
||||
krb5_data tag;
|
||||
krb5_data version;
|
||||
krb5_error_code ret, ret2;
|
||||
unsigned ver;
|
||||
int foo;
|
||||
|
||||
ret = db->hdb_lock(context, db, HDB_RLOCK);
|
||||
if (ret)
|
||||
return ret;
|
||||
|
||||
tag.data = HDB_DB_FORMAT_ENTRY;
|
||||
tag.length = strlen(tag.data);
|
||||
ret = (*db->hdb__get)(context, db, tag, &version);
|
||||
ret2 = db->hdb_unlock(context, db);
|
||||
if(ret)
|
||||
return ret;
|
||||
if (ret2)
|
||||
return ret2;
|
||||
foo = sscanf(version.data, "%u", &ver);
|
||||
krb5_data_free (&version);
|
||||
if (foo != 1)
|
||||
return HDB_ERR_BADVERSION;
|
||||
if(ver != HDB_DB_FORMAT)
|
||||
return HDB_ERR_BADVERSION;
|
||||
return 0;
|
||||
}
|
||||
|
||||
krb5_error_code
|
||||
hdb_init_db(krb5_context context, HDB *db)
|
||||
{
|
||||
krb5_error_code ret, ret2;
|
||||
krb5_data tag;
|
||||
krb5_data version;
|
||||
char ver[32];
|
||||
|
||||
ret = hdb_check_db_format(context, db);
|
||||
if(ret != HDB_ERR_NOENTRY)
|
||||
return ret;
|
||||
|
||||
ret = db->hdb_lock(context, db, HDB_WLOCK);
|
||||
if (ret)
|
||||
return ret;
|
||||
|
||||
tag.data = HDB_DB_FORMAT_ENTRY;
|
||||
tag.length = strlen(tag.data);
|
||||
snprintf(ver, sizeof(ver), "%u", HDB_DB_FORMAT);
|
||||
version.data = ver;
|
||||
version.length = strlen(version.data) + 1; /* zero terminated */
|
||||
ret = (*db->hdb__put)(context, db, 0, tag, version);
|
||||
ret2 = db->hdb_unlock(context, db);
|
||||
if (ret)
|
||||
return ret;
|
||||
return ret2;
|
||||
}
|
||||
|
||||
#ifdef HAVE_DLOPEN
|
||||
|
||||
/*
|
||||
* Load a dynamic backend from /usr/heimdal/lib/hdb_NAME.so,
|
||||
* looking for the hdb_NAME_create symbol.
|
||||
*/
|
||||
|
||||
static const struct hdb_method *
|
||||
find_dynamic_method (krb5_context context,
|
||||
const char *filename,
|
||||
const char **rest)
|
||||
{
|
||||
static struct hdb_method method;
|
||||
struct hdb_so_method *mso;
|
||||
char *prefix, *path, *symbol;
|
||||
const char *p;
|
||||
void *dl;
|
||||
size_t len;
|
||||
|
||||
p = strchr(filename, ':');
|
||||
|
||||
/* if no prefix, don't know what module to load, just ignore it */
|
||||
if (p == NULL)
|
||||
return NULL;
|
||||
|
||||
len = p - filename;
|
||||
*rest = filename + len + 1;
|
||||
|
||||
prefix = strndup(filename, len);
|
||||
if (prefix == NULL)
|
||||
krb5_errx(context, 1, "out of memory");
|
||||
|
||||
if (asprintf(&path, LIBDIR "/hdb_%s.so", prefix) == -1)
|
||||
krb5_errx(context, 1, "out of memory");
|
||||
|
||||
#ifndef RTLD_NOW
|
||||
#define RTLD_NOW 0
|
||||
#endif
|
||||
#ifndef RTLD_GLOBAL
|
||||
#define RTLD_GLOBAL 0
|
||||
#endif
|
||||
|
||||
dl = dlopen(path, RTLD_NOW | RTLD_GLOBAL);
|
||||
if (dl == NULL) {
|
||||
krb5_warnx(context, "error trying to load dynamic module %s: %s\n",
|
||||
path, dlerror());
|
||||
free(prefix);
|
||||
free(path);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
if (asprintf(&symbol, "hdb_%s_interface", prefix) == -1)
|
||||
krb5_errx(context, 1, "out of memory");
|
||||
|
||||
mso = dlsym(dl, symbol);
|
||||
if (mso == NULL) {
|
||||
krb5_warnx(context, "error finding symbol %s in %s: %s\n",
|
||||
symbol, path, dlerror());
|
||||
dlclose(dl);
|
||||
free(symbol);
|
||||
free(prefix);
|
||||
free(path);
|
||||
return NULL;
|
||||
}
|
||||
free(path);
|
||||
free(symbol);
|
||||
|
||||
if (mso->version != HDB_INTERFACE_VERSION) {
|
||||
krb5_warnx(context,
|
||||
"error wrong version in shared module %s "
|
||||
"version: %d should have been %d\n",
|
||||
prefix, mso->version, HDB_INTERFACE_VERSION);
|
||||
dlclose(dl);
|
||||
free(prefix);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
if (mso->create == NULL) {
|
||||
krb5_errx(context, 1,
|
||||
"no entry point function in shared mod %s ",
|
||||
prefix);
|
||||
dlclose(dl);
|
||||
free(prefix);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
method.create = mso->create;
|
||||
method.prefix = prefix;
|
||||
|
||||
return &method;
|
||||
}
|
||||
#endif /* HAVE_DLOPEN */
|
||||
|
||||
/*
|
||||
* find the relevant method for `filename', returning a pointer to the
|
||||
* rest in `rest'.
|
||||
* return NULL if there's no such method.
|
||||
*/
|
||||
|
||||
static const struct hdb_method *
|
||||
find_method (const char *filename, const char **rest)
|
||||
{
|
||||
const struct hdb_method *h;
|
||||
|
||||
for (h = methods; h->prefix != NULL; ++h) {
|
||||
if (strncmp (filename, h->prefix, strlen(h->prefix)) == 0) {
|
||||
*rest = filename + strlen(h->prefix);
|
||||
return h;
|
||||
}
|
||||
}
|
||||
#if defined(HAVE_DB1) || defined(HAVE_DB3) || defined(HAVE_NDBM)
|
||||
if (strncmp(filename, "/", 1) == 0
|
||||
|| strncmp(filename, "./", 2) == 0
|
||||
|| strncmp(filename, "../", 3) == 0)
|
||||
{
|
||||
*rest = filename;
|
||||
return &dbmetod;
|
||||
}
|
||||
#endif
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
krb5_error_code
|
||||
hdb_list_builtin(krb5_context context, char **list)
|
||||
{
|
||||
const struct hdb_method *h;
|
||||
size_t len = 0;
|
||||
char *buf = NULL;
|
||||
|
||||
for (h = methods; h->prefix != NULL; ++h) {
|
||||
if (h->prefix[0] == '\0')
|
||||
continue;
|
||||
len += strlen(h->prefix) + 2;
|
||||
}
|
||||
|
||||
len += 1;
|
||||
buf = malloc(len);
|
||||
if (buf == NULL) {
|
||||
krb5_set_error_string(context, "malloc: out of memory");
|
||||
return ENOMEM;
|
||||
}
|
||||
buf[0] = '\0';
|
||||
|
||||
for (h = methods; h->prefix != NULL; ++h) {
|
||||
if (h != methods)
|
||||
strlcat(buf, ", ", len);
|
||||
strlcat(buf, h->prefix, len);
|
||||
}
|
||||
*list = buf;
|
||||
return 0;
|
||||
}
|
||||
|
||||
krb5_error_code
|
||||
hdb_create(krb5_context context, HDB **db, const char *filename)
|
||||
{
|
||||
const struct hdb_method *h;
|
||||
const char *residual;
|
||||
|
||||
if(filename == NULL)
|
||||
filename = HDB_DEFAULT_DB;
|
||||
krb5_add_et_list(context, initialize_hdb_error_table_r);
|
||||
h = find_method (filename, &residual);
|
||||
#ifdef HAVE_DLOPEN
|
||||
if (h == NULL)
|
||||
h = find_dynamic_method (context, filename, &residual);
|
||||
#endif
|
||||
if (h == NULL)
|
||||
krb5_errx(context, 1, "No database support for %s", filename);
|
||||
return (*h->create)(context, db, residual);
|
||||
}
|
||||
Reference in New Issue
Block a user