int
main (int argc,
char *argv[])
{
const char *default_uristr = "mongodb://localhost/test";
char *uristr;
const char *database_name;
mongoc_uri_t *uri;
mongoc_client_t *client;
mongoc_database_t *db;
mongoc_collection_t *collection;
mongoc_init ();
uristr = getenv ("MONGODB_URI");
uristr = uristr ? uristr : (char*)default_uristr;
uri = mongoc_uri_new (uristr);
client = mongoc_client_new_from_uri (uri);
database_name = mongoc_uri_get_database (uri);
db = mongoc_client_get_database (client, database_name);
collection = mongoc_database_get_collection (db, "test");
test_suite (db, collection);
mongoc_collection_destroy (collection);
mongoc_database_destroy (db);
mongoc_client_destroy (client);
mongoc_uri_destroy (uri);
mongoc_cleanup ();
return 0;
}
int64_t
execute (int argc,
char *argv[])
{
int64_t count = 0;
const char *uristr = MONGODB_DEFAULT_URI;
const char *database_name;
mongoc_uri_t *uri;
mongoc_client_t *client;
mongoc_database_t *db;
bson_t bson_schema;
bson_error_t error;
int argi;
uristr = getenv ("MONGODB_URI");
uri = mongoc_uri_new (uristr);
client = mongoc_client_new (uristr);
database_name = mongoc_uri_get_database (uri);
db = mongoc_client_get_database (client, database_name);
bson_init_from_json_file (&bson_schema, schema_file) || WARN_ERROR;
for (argi = 0; argi < argc; argi++) {
fprintf (stderr, "[%d/%d] %s\n", argi + 1, argc, argv[argi]);
count += load_table (db, argv[argi], &bson_schema);
}
bson_destroy (&bson_schema);
mongoc_database_destroy (db);
mongoc_client_destroy (client);
mongoc_uri_destroy (uri);
return count;
}
int mongodb_module_init(struct state_conf *conf, UNUSED char **fields, UNUSED int fieldlens)
{
char *uri_str = NULL;
buffer_fill = 0;
const char *db;
if (conf->output_args) {
uri_str = conf->output_args;
}
mongoc_init();
mongoc_log_set_handler(mongodb_module_log, NULL);
mongoc_uri_t *uri = mongoc_uri_new(uri_str);
if (uri == NULL) {
log_fatal("mongodb-module", "URI %s not valid!", uri_str);
}
client = mongoc_client_new_from_uri(uri);
db = mongoc_uri_get_database(uri);
collection = mongoc_client_get_collection(client, db ? db : strdup("zmap_output"),
conf->output_filename ? conf->output_filename : strdup("zmap_output"));
bulk = mongoc_collection_create_bulk_operation(collection,false,NULL);
return EXIT_SUCCESS;
}
mongoc_database_t *
mongoc_client_get_default_database (mongoc_client_t *client)
{
const char *db;
BSON_ASSERT (client);
db = mongoc_uri_get_database (client->uri);
if (db) {
return mongoc_client_get_database (client, db);
}
return NULL;
}
static MONGO * get_mongodb_connection(void) {
persistent_users_db_t * pud = get_persistent_users_db();
MONGO * mydbconnection = (MONGO *) pthread_getspecific(connection_key);
if (!mydbconnection) {
mongoc_init();
mongoc_log_set_handler(&mongo_logger, NULL);
mydbconnection = (MONGO *) turn_malloc(sizeof(MONGO));
mydbconnection->uri = mongoc_uri_new(pud->userdb);
if (!mydbconnection->uri) {
TURN_LOG_FUNC(
TURN_LOG_LEVEL_ERROR,
"Cannot open parse MongoDB URI <%s>, connection string format error\n",
pud->userdb);
MongoFree(mydbconnection);
mydbconnection = NULL;
} else {
mydbconnection->client = mongoc_client_new_from_uri(
mydbconnection->uri);
if (!mydbconnection->client) {
TURN_LOG_FUNC(TURN_LOG_LEVEL_ERROR,
"Cannot initialize MongoDB connection\n");
MongoFree(mydbconnection);
mydbconnection = NULL;
} else {
mydbconnection->database = mongoc_uri_get_database(
mydbconnection->uri);
if (!mydbconnection->database)
mydbconnection->database = MONGO_DEFAULT_DB;
if(mydbconnection) {
(void) pthread_setspecific(connection_key, mydbconnection);
}
TURN_LOG_FUNC(
TURN_LOG_LEVEL_INFO,
"Opened MongoDB URI <%s>\n",
pud->userdb);
}
}
}
return mydbconnection;
}
std::string uri::database() const {
return mongoc_uri_get_database(_impl->uri_t);
}
static void
test_mongoc_uri_new (void)
{
const mongoc_host_list_t *hosts;
const bson_t *options;
const bson_t *credentials;
bson_t properties;
mongoc_uri_t *uri;
bson_iter_t iter;
bson_iter_t child;
/* bad uris */
ASSERT(!mongoc_uri_new("mongodb://"));
ASSERT(!mongoc_uri_new("mongodb://::"));
ASSERT(!mongoc_uri_new("mongodb://*****:*****@localhost:27017/foo/?authSource=abcd");
ASSERT(uri);
ASSERT_CMPSTR(mongoc_uri_get_username(uri), "christian");
ASSERT_CMPSTR(mongoc_uri_get_password(uri), "secret");
ASSERT_CMPSTR(mongoc_uri_get_auth_source(uri), "abcd");
mongoc_uri_destroy(uri);
/* should use the default auth source and mechanism */
uri = mongoc_uri_new("mongodb://*****:*****@localhost:27017");
ASSERT(uri);
ASSERT_CMPSTR(mongoc_uri_get_auth_source(uri), "admin");
ASSERT(!mongoc_uri_get_auth_mechanism(uri));
mongoc_uri_destroy(uri);
/* should use the db when no authSource is specified */
uri = mongoc_uri_new("mongodb://*****:*****@localhost/foo");
ASSERT(uri);
ASSERT_CMPSTR(mongoc_uri_get_auth_source(uri), "foo");
mongoc_uri_destroy(uri);
/* should recognize an empty password */
uri = mongoc_uri_new("mongodb://samantha:@localhost");
ASSERT(uri);
ASSERT_CMPSTR(mongoc_uri_get_username(uri), "samantha");
ASSERT_CMPSTR(mongoc_uri_get_password(uri), "");
mongoc_uri_destroy(uri);
/* should recognize no password */
uri = mongoc_uri_new("mongodb://christian@localhost:27017");
ASSERT(uri);
ASSERT_CMPSTR(mongoc_uri_get_username(uri), "christian");
ASSERT(!mongoc_uri_get_password(uri));
mongoc_uri_destroy(uri);
/* should recognize a url escaped character in the username */
uri = mongoc_uri_new("mongodb://christian%40realm:pwd@localhost:27017");
ASSERT(uri);
ASSERT_CMPSTR(mongoc_uri_get_username(uri), "christian@realm");
mongoc_uri_destroy(uri);
/* while you shouldn't do this, lets test for it */
uri = mongoc_uri_new("mongodb://christian%40realm@localhost:27017/db%2ename");
ASSERT(uri);
ASSERT_CMPSTR(mongoc_uri_get_database(uri), "db.name");
mongoc_uri_destroy(uri);
uri = mongoc_uri_new("mongodb://christian%40realm@localhost:27017/db%2Ename");
ASSERT(uri);
ASSERT_CMPSTR(mongoc_uri_get_database(uri), "db.name");
mongoc_uri_destroy(uri);
uri = mongoc_uri_new("mongodb://christian%40realm@localhost:27017/?abcd=%20");
ASSERT(uri);
options = mongoc_uri_get_options(uri);
ASSERT(options);
ASSERT(bson_iter_init_find(&iter, options, "abcd"));
ASSERT(BSON_ITER_HOLDS_UTF8(&iter));
ASSERT_CMPSTR(bson_iter_utf8(&iter, NULL), " ");
mongoc_uri_destroy(uri);
uri = mongoc_uri_new("mongodb://christian%40realm@[::6]:27017/?abcd=%20");
ASSERT(uri);
options = mongoc_uri_get_options(uri);
ASSERT(options);
ASSERT(bson_iter_init_find(&iter, options, "abcd"));
ASSERT(BSON_ITER_HOLDS_UTF8(&iter));
ASSERT_CMPSTR(bson_iter_utf8(&iter, NULL), " ");
mongoc_uri_destroy(uri);
/* GSSAPI-specific options */
/* should recognize the GSSAPI mechanism, and use $external as source */
uri = mongoc_uri_new("mongodb://user%40DOMAIN.COM:password@localhost/?authMechanism=GSSAPI");
ASSERT(uri);
ASSERT_CMPSTR(mongoc_uri_get_auth_mechanism(uri), "GSSAPI");
/*ASSERT_CMPSTR(mongoc_uri_get_auth_source(uri), "$external");*/
mongoc_uri_destroy(uri);
/* use $external as source when db is specified */
uri = mongoc_uri_new("mongodb://user%40DOMAIN.COM:password@localhost/foo/?authMechanism=GSSAPI");
ASSERT(uri);
ASSERT_CMPSTR(mongoc_uri_get_auth_source(uri), "$external");
mongoc_uri_destroy(uri);
/* should not accept authSource other than $external */
ASSERT(!mongoc_uri_new("mongodb://user%40DOMAIN.COM:password@localhost/foo/?authMechanism=GSSAPI&authSource=bar"));
/* should accept authMechanismProperties */
uri = mongoc_uri_new("mongodb://user%40DOMAIN.COM:password@localhost/?authMechanism=GSSAPI"
"&authMechanismProperties=SERVICE_NAME:other,CANONICALIZE_HOST_NAME:true");
ASSERT(uri);
credentials = mongoc_uri_get_credentials(uri);
ASSERT(credentials);
ASSERT(mongoc_uri_get_mechanism_properties(uri, &properties));
assert (bson_iter_init_find_case (&iter, &properties, "SERVICE_NAME") &&
BSON_ITER_HOLDS_UTF8 (&iter) &&
(0 == strcmp (bson_iter_utf8 (&iter, NULL), "other")));
assert (bson_iter_init_find_case (&iter, &properties, "CANONICALIZE_HOST_NAME") &&
BSON_ITER_HOLDS_UTF8 (&iter) &&
(0 == strcmp (bson_iter_utf8 (&iter, NULL), "true")));
mongoc_uri_destroy(uri);
/* reverse order of arguments to ensure parsing still succeeds */
uri = mongoc_uri_new("mongodb://user@localhost/"
"?authMechanismProperties=SERVICE_NAME:other"
"&authMechanism=GSSAPI");
ASSERT(uri);
mongoc_uri_destroy(uri);
/* deprecated gssapiServiceName option */
uri = mongoc_uri_new("mongodb://christian%40realm.cc@localhost:27017/?authMechanism=GSSAPI&gssapiServiceName=blah");
ASSERT(uri);
options = mongoc_uri_get_options(uri);
ASSERT(options);
assert (0 == strcmp (mongoc_uri_get_auth_mechanism (uri), "GSSAPI"));
assert (0 == strcmp (mongoc_uri_get_username (uri), "*****@*****.**"));
assert (bson_iter_init_find_case (&iter, options, "gssapiServiceName") &&
BSON_ITER_HOLDS_UTF8 (&iter) &&
(0 == strcmp (bson_iter_utf8 (&iter, NULL), "blah")));
mongoc_uri_destroy(uri);
/* MONGODB-CR */
/* should recognize this mechanism */
uri = mongoc_uri_new("mongodb://user@localhost/?authMechanism=MONGODB-CR");
ASSERT(uri);
ASSERT_CMPSTR(mongoc_uri_get_auth_mechanism(uri), "MONGODB-CR");
mongoc_uri_destroy(uri);
/* X509 */
/* should recognize this mechanism, and use $external as the source */
uri = mongoc_uri_new("mongodb://user@localhost/?authMechanism=MONGODB-X509");
ASSERT(uri);
ASSERT_CMPSTR(mongoc_uri_get_auth_mechanism(uri), "MONGODB-X509");
/*ASSERT_CMPSTR(mongoc_uri_get_auth_source(uri), "$external");*/
mongoc_uri_destroy(uri);
/* use $external as source when db is specified */
uri = mongoc_uri_new("mongodb://CN%3DmyName%2COU%3DmyOrgUnit%2CO%3DmyOrg%2CL%3DmyLocality"
"%2CST%3DmyState%2CC%3DmyCountry@localhost/foo/?authMechanism=MONGODB-X509");
ASSERT(uri);
ASSERT_CMPSTR(mongoc_uri_get_auth_source(uri), "$external");
mongoc_uri_destroy(uri);
/* should not accept authSource other than $external */
ASSERT(!mongoc_uri_new("mongodb://CN%3DmyName%2COU%3DmyOrgUnit%2CO%3DmyOrg%2CL%3DmyLocality"
"%2CST%3DmyState%2CC%3DmyCountry@localhost/foo/?authMechanism=MONGODB-X509&authSource=bar"));
/* should recognize the encoded username */
uri = mongoc_uri_new("mongodb://CN%3DmyName%2COU%3DmyOrgUnit%2CO%3DmyOrg%2CL%3DmyLocality"
"%2CST%3DmyState%2CC%3DmyCountry@localhost/?authMechanism=MONGODB-X509");
ASSERT(uri);
ASSERT_CMPSTR(mongoc_uri_get_username(uri), "CN=myName,OU=myOrgUnit,O=myOrg,L=myLocality,ST=myState,C=myCountry");
mongoc_uri_destroy(uri);
/* PLAIN */
/* should recognize this mechanism */
uri = mongoc_uri_new("mongodb://user@localhost/?authMechanism=PLAIN");
ASSERT(uri);
ASSERT_CMPSTR(mongoc_uri_get_auth_mechanism(uri), "PLAIN");
mongoc_uri_destroy(uri);
/* SCRAM-SHA1 */
/* should recognize this mechanism */
uri = mongoc_uri_new("mongodb://user@localhost/?authMechanism=SCRAM-SHA1");
ASSERT(uri);
ASSERT_CMPSTR(mongoc_uri_get_auth_mechanism(uri), "SCRAM-SHA1");
mongoc_uri_destroy(uri);
}
static void
test_mongoc_uri_new (void)
{
const mongoc_host_list_t *hosts;
const bson_t *options;
mongoc_uri_t *uri;
bson_iter_t iter;
bson_iter_t child;
ASSERT(!mongoc_uri_new("mongodb://"));
ASSERT(!mongoc_uri_new("mongodb://::"));
ASSERT(!mongoc_uri_new("mongodb://*****:*****@localhost:27017?authSource=abcd");
ASSERT(uri);
ASSERT_CMPSTR(mongoc_uri_get_username(uri), "christian");
ASSERT_CMPSTR(mongoc_uri_get_password(uri), "secret");
ASSERT_CMPSTR(mongoc_uri_get_auth_source(uri), "abcd");
mongoc_uri_destroy(uri);
uri = mongoc_uri_new("mongodb://*****:*****@localhost:27017");
ASSERT(uri);
ASSERT_CMPSTR(mongoc_uri_get_auth_source(uri), "admin");
mongoc_uri_destroy(uri);
uri = mongoc_uri_new("mongodb://christian@localhost:27017");
ASSERT(uri);
ASSERT_CMPSTR(mongoc_uri_get_username(uri), "christian");
mongoc_uri_destroy(uri);
uri = mongoc_uri_new("mongodb://christian%40realm@localhost:27017");
ASSERT(uri);
ASSERT_CMPSTR(mongoc_uri_get_username(uri), "christian@realm");
mongoc_uri_destroy(uri);
/* while you shouldn't do this, lets test for it */
uri = mongoc_uri_new("mongodb://christian%40realm@localhost:27017/db%2ename");
ASSERT(uri);
ASSERT_CMPSTR(mongoc_uri_get_database(uri), "db.name");
mongoc_uri_destroy(uri);
uri = mongoc_uri_new("mongodb://christian%40realm@localhost:27017/db%2Ename");
ASSERT(uri);
ASSERT_CMPSTR(mongoc_uri_get_database(uri), "db.name");
mongoc_uri_destroy(uri);
uri = mongoc_uri_new("mongodb://christian%40realm@localhost:27017/?abcd=%20");
ASSERT(uri);
options = mongoc_uri_get_options(uri);
ASSERT(options);
ASSERT(bson_iter_init_find(&iter, options, "abcd"));
ASSERT(BSON_ITER_HOLDS_UTF8(&iter));
ASSERT_CMPSTR(bson_iter_utf8(&iter, NULL), " ");
mongoc_uri_destroy(uri);
uri = mongoc_uri_new("mongodb://christian%40realm.cc@localhost:27017/?authmechanism=GSSAPI&gssapiservicename=blah");
ASSERT(uri);
options = mongoc_uri_get_options(uri);
ASSERT(options);
assert (0 == strcmp (mongoc_uri_get_auth_mechanism (uri), "GSSAPI"));
assert (0 == strcmp (mongoc_uri_get_username (uri), "*****@*****.**"));
assert (bson_iter_init_find_case (&iter, options, "gssapiservicename") &&
BSON_ITER_HOLDS_UTF8 (&iter) &&
(0 == strcmp (bson_iter_utf8 (&iter, NULL), "blah")));
mongoc_uri_destroy(uri);
uri = mongoc_uri_new("mongodb://christian%40realm@[::6]:27017/?abcd=%20");
ASSERT(uri);
options = mongoc_uri_get_options(uri);
ASSERT(options);
ASSERT(bson_iter_init_find(&iter, options, "abcd"));
ASSERT(BSON_ITER_HOLDS_UTF8(&iter));
ASSERT_CMPSTR(bson_iter_utf8(&iter, NULL), " ");
mongoc_uri_destroy(uri);
}
