static mcon_collection *mongo_filter_candidates_by_replicaset_name(mongo_con_manager *manager, mcon_collection *candidates, mongo_servers *servers)
{
int i;
mcon_collection *filtered;
char *candidate_hash;
char *candidate_replsetname;
mongo_manager_log(manager, MLOG_RS, MLOG_FINE, "limiting to servers with same replicaset name");
filtered = mcon_init_collection(sizeof(mongo_connection*));
for (i = 0; i < candidates->count; i++) {
candidate_hash = ((mongo_connection *) candidates->data[i])->hash;
mongo_server_split_hash(candidate_hash, NULL, NULL, (char**) &candidate_replsetname, NULL, NULL, NULL, NULL);
/* Filter out all servers that don't have the replicaset name the same
* as what we have in the server definition struct. But only when the
* repl_set_name in the server definition struct is actually *set*. If
* not, we allow all connections. This make sure we can sort of handle
* [ replicaset => true ], although it would not support one PHP worker
* process connecting to multiple replicasets correctly. */
if (candidate_replsetname) {
if (!servers->repl_set_name || strcmp(candidate_replsetname, servers->repl_set_name) == 0) {
mongo_print_connection_info(manager, (mongo_connection *) candidates->data[i], MLOG_FINE);
mcon_collection_add(filtered, (mongo_connection *) candidates->data[i]);
}
free(candidate_replsetname);
}
}
mongo_manager_log(manager, MLOG_RS, MLOG_FINE, "limiting to servers with same replicaset name: done");
return filtered;
}
static mcon_collection *mongo_filter_candidates_by_credentials(mongo_con_manager *manager, mcon_collection *candidates, mongo_servers *servers)
{
int i;
char *db, *username, *auth_hash, *hashed = NULL;
mcon_collection *filtered;
mongo_manager_log(manager, MLOG_RS, MLOG_FINE, "limiting by credentials");
filtered = mcon_init_collection(sizeof(mongo_connection*));
for (i = 0; i < candidates->count; i++) {
db = username = auth_hash = hashed = NULL;
mongo_server_split_hash(((mongo_connection *) candidates->data[i])->hash, NULL, NULL, NULL, &db, &username, &auth_hash, NULL);
if (servers->server[0]->username && servers->server[0]->password && servers->server[0]->db) {
if (strcmp(db, servers->server[0]->db) != 0) {
mongo_manager_log(manager, MLOG_RS, MLOG_FINE, "- skipping '%s', database didn't match ('%s' vs '%s')", ((mongo_connection *) candidates->data[i])->hash, db, servers->server[0]->db);
goto skip;
}
if (strcmp(username, servers->server[0]->username) != 0) {
mongo_manager_log(manager, MLOG_RS, MLOG_FINE, "- skipping '%s', username didn't match ('%s' vs '%s')", ((mongo_connection *) candidates->data[i])->hash, username, servers->server[0]->username);
goto skip;
}
hashed = mongo_server_create_hashed_password(username, servers->server[0]->password);
if (strcmp(auth_hash, hashed) != 0) {
mongo_manager_log(manager, MLOG_RS, MLOG_FINE, "- skipping '%s', authentication hash didn't match ('%s' vs '%s')", ((mongo_connection *) candidates->data[i])->hash, auth_hash, hashed);
goto skip;
}
}
mcon_collection_add(filtered, (mongo_connection *) candidates->data[i]);
mongo_print_connection_info(manager, (mongo_connection *) candidates->data[i], MLOG_FINE);
skip:
if (db) {
free(db);
}
if (username) {
free(username);
}
if (auth_hash) {
free(auth_hash);
}
if (hashed) {
free(hashed);
}
}
mongo_manager_log(manager, MLOG_RS, MLOG_FINE, "limiting by credentials: done");
return filtered;
}
int main(void)
{
char *hash1, *hash2;
char *host, *db, *username, *auth_hash;
int port, pid;
mongo_server_def def1 = { "whisky", 13000, NULL, NULL, NULL };
mongo_server_def def2 = { "whisky", 13000, "phpunit", "derick", "not!" };
hash1 = mongo_server_create_hash(&def1);
mongo_server_split_hash(hash1, &host, &port, &db, &username, &auth_hash, &pid);
printf("HASH: %s; host: %s, port: %d, db: %s, username: %s, auth_hash: %s, pid: %d\n",
hash1, host, port, db, username, auth_hash, pid);
free(host);
free(hash1);
hash2 = mongo_server_create_hash(&def2);
mongo_server_split_hash(hash2, &host, &port, &db, &username, &auth_hash, &pid);
printf("HASH: %s; host: %s, port: %d, db: %s, username: %s, auth_hash: %s, pid: %d\n",
hash2, host, port, db, username, auth_hash, pid);
free(host); free(db); free(username); free(auth_hash);
host = db = username = auth_hash = NULL; port = pid = 0;
mongo_server_split_hash(hash2, &host, &port, NULL, &username, &auth_hash, &pid);
printf("HASH: %s; host: %s, port: %d, db: %s, username: %s, auth_hash: %s, pid: %d\n",
hash2, host, port, db, username, auth_hash, pid);
free(host); free(username); free(auth_hash);
host = db = username = auth_hash = NULL; port = pid = 0;
mongo_server_split_hash(hash2, &host, &port, &db, NULL, &auth_hash, &pid);
printf("HASH: %s; host: %s, port: %d, db: %s, username: %s, auth_hash: %s, pid: %d\n",
hash2, host, port, db, username, auth_hash, pid);
free(host); free(db); free(auth_hash);
host = db = username = auth_hash = NULL; port = pid = 0;
mongo_server_split_hash(hash2, &host, &port, &db, NULL, NULL, &pid);
printf("HASH: %s; host: %s, port: %d, db: %s, username: %s, auth_hash: %s, pid: %d\n",
hash2, host, port, db, username, auth_hash, pid);
free(host); free(db);
host = db = username = auth_hash = NULL; port = pid = 0;
mongo_server_split_hash(hash2, &host, &port, NULL, &username, NULL, &pid);
printf("HASH: %s; host: %s, port: %d, db: %s, username: %s, auth_hash: %s, pid: %d\n",
hash2, host, port, db, username, auth_hash, pid);
free(host); free(username);
free(hash2);
return 0;
}
