mongoc_client_pool_t *
mongoc_client_pool_new (const mongoc_uri_t *uri)
{
mongoc_topology_t *topology;
mongoc_client_pool_t *pool;
const bson_t *b;
bson_iter_t iter;
ENTRY;
BSON_ASSERT (uri);
#ifndef MONGOC_ENABLE_SSL
if (mongoc_uri_get_ssl (uri)) {
MONGOC_ERROR ("Can't create SSL client pool,"
" SSL not enabled in this build.");
return NULL;
}
#endif
pool = (mongoc_client_pool_t *)bson_malloc0(sizeof *pool);
mongoc_mutex_init(&pool->mutex);
_mongoc_queue_init(&pool->queue);
pool->uri = mongoc_uri_copy(uri);
pool->min_pool_size = 0;
pool->max_pool_size = 100;
pool->size = 0;
pool->topology_scanner_started = false;
topology = mongoc_topology_new(uri, false);
pool->topology = topology;
mongoc_client_metadata_init (&pool->topology->scanner->ismaster_metadata);
pool->error_api_version = MONGOC_ERROR_API_VERSION_LEGACY;
b = mongoc_uri_get_options(pool->uri);
if (bson_iter_init_find_case(&iter, b, "minpoolsize")) {
if (BSON_ITER_HOLDS_INT32(&iter)) {
pool->min_pool_size = BSON_MAX(0, bson_iter_int32(&iter));
}
}
if (bson_iter_init_find_case(&iter, b, "maxpoolsize")) {
if (BSON_ITER_HOLDS_INT32(&iter)) {
pool->max_pool_size = BSON_MAX(1, bson_iter_int32(&iter));
}
}
mongoc_counter_client_pools_active_inc();
RETURN(pool);
}
mongoc_client_t *
_mongoc_client_new_from_uri (const mongoc_uri_t *uri, mongoc_topology_t *topology)
{
mongoc_client_t *client;
const mongoc_read_prefs_t *read_prefs;
const mongoc_read_concern_t *read_concern;
const mongoc_write_concern_t *write_concern;
BSON_ASSERT (uri);
#ifndef MONGOC_ENABLE_SSL
if (mongoc_uri_get_ssl (uri)) {
MONGOC_ERROR ("Can't create SSL client, SSL not enabled in this build.");
return NULL;
}
#endif
client = (mongoc_client_t *)bson_malloc0(sizeof *client);
client->uri = mongoc_uri_copy (uri);
client->request_id = rand ();
client->initiator = mongoc_client_default_stream_initiator;
client->initiator_data = client;
client->topology = topology;
write_concern = mongoc_uri_get_write_concern (client->uri);
client->write_concern = mongoc_write_concern_copy (write_concern);
read_concern = mongoc_uri_get_read_concern (client->uri);
client->read_concern = mongoc_read_concern_copy (read_concern);
read_prefs = mongoc_uri_get_read_prefs_t (client->uri);
client->read_prefs = mongoc_read_prefs_copy (read_prefs);
mongoc_cluster_init (&client->cluster, client->uri, client);
#ifdef MONGOC_ENABLE_SSL
client->use_ssl = false;
if (mongoc_uri_get_ssl (client->uri)) {
/* sets use_ssl = true */
mongoc_client_set_ssl_opts (client, mongoc_ssl_opt_get_default ());
}
#endif
mongoc_counter_clients_active_inc ();
return client;
}
mongoc_client_t *
_mongoc_client_new_from_uri (const mongoc_uri_t *uri, mongoc_topology_t *topology)
{
mongoc_client_t *client;
const mongoc_read_prefs_t *read_prefs;
const mongoc_write_concern_t *write_concern;
#ifdef MONGOC_ENABLE_SSL
const bson_t *options;
bson_iter_t iter;
#endif
BSON_ASSERT (uri);
client = (mongoc_client_t *)bson_malloc0(sizeof *client);
client->uri = mongoc_uri_copy (uri);
client->request_id = rand ();
client->initiator = mongoc_client_default_stream_initiator;
client->initiator_data = client;
client->topology = topology;
write_concern = mongoc_uri_get_write_concern (client->uri);
client->write_concern = mongoc_write_concern_copy (write_concern);
read_prefs = mongoc_uri_get_read_prefs_t (client->uri);
client->read_prefs = mongoc_read_prefs_copy (read_prefs);
mongoc_cluster_init (&client->cluster, client->uri, client);
#ifdef MONGOC_ENABLE_SSL
options = mongoc_uri_get_options (client->uri);
if (bson_iter_init_find (&iter, options, "ssl") &&
BSON_ITER_HOLDS_BOOL (&iter) &&
bson_iter_bool (&iter)) {
mongoc_client_set_ssl_opts (client, mongoc_ssl_opt_get_default ());
}
#endif
mongoc_counter_clients_active_inc ();
return client;
}
static mongoc_uri_t *
_get_uri (mock_server_t *server, read_pref_test_type_t test_type)
{
mongoc_uri_t *uri;
uri = mongoc_uri_copy (mock_server_get_uri (server));
switch (test_type) {
case READ_PREF_TEST_PRIMARY:
case READ_PREF_TEST_SECONDARY:
mongoc_uri_set_option_as_utf8 (uri, "replicaSet", "rs");
break;
case READ_PREF_TEST_STANDALONE:
case READ_PREF_TEST_MONGOS:
default:
break;
}
return uri;
}
mongoc_client_pool_t *
mongoc_client_pool_new (const mongoc_uri_t *uri)
{
mongoc_topology_t *topology;
mongoc_client_pool_t *pool;
const bson_t *b;
bson_iter_t iter;
ENTRY;
BSON_ASSERT (uri);
pool = (mongoc_client_pool_t *)bson_malloc0(sizeof *pool);
mongoc_mutex_init(&pool->mutex);
_mongoc_queue_init(&pool->queue);
pool->uri = mongoc_uri_copy(uri);
pool->min_pool_size = 0;
pool->max_pool_size = 100;
pool->size = 0;
topology = mongoc_topology_new(uri, false);
pool->topology = topology;
b = mongoc_uri_get_options(pool->uri);
if (bson_iter_init_find_case(&iter, b, "minpoolsize")) {
if (BSON_ITER_HOLDS_INT32(&iter)) {
pool->min_pool_size = BSON_MAX(0, bson_iter_int32(&iter));
}
}
if (bson_iter_init_find_case(&iter, b, "maxpoolsize")) {
if (BSON_ITER_HOLDS_INT32(&iter)) {
pool->max_pool_size = BSON_MAX(1, bson_iter_int32(&iter));
}
}
mongoc_counter_client_pools_active_inc();
RETURN(pool);
}
static void
_test_heartbeat_events (bool pooled,
bool succeeded)
{
context_t context;
mock_server_t *server;
mongoc_uri_t *uri;
mongoc_client_t *client;
mongoc_client_pool_t *pool = NULL;
future_t *future;
request_t *request;
char *expected_json;
bson_error_t error;
context_init (&context);
/* auto-respond to "foo" command */
server = mock_server_new ();
mock_server_run (server);
mock_server_autoresponds (server, responder, NULL, NULL);
uri = mongoc_uri_copy (mock_server_get_uri (server));
mongoc_uri_set_option_as_int32 (uri, "serverSelectionTimeoutMS", 400);
if (pooled) {
pool = mongoc_client_pool_new (uri);
pool_set_heartbeat_event_callbacks (pool, &context);
client = mongoc_client_pool_pop (pool);
} else {
client = mongoc_client_new_from_uri (uri);
client_set_heartbeat_event_callbacks (client, &context);
}
/* trigger "ismaster" handshake */
future = future_client_command_simple (client, "admin",
tmp_bson ("{'foo': 1}"),
NULL, NULL, &error);
/* topology scanner calls ismaster once */
request = mock_server_receives_ismaster (server);
if (succeeded) {
mock_server_replies_ok_and_destroys (request);
} else {
mock_server_hangs_up (request);
request_destroy (request);
}
/* pooled client opens new socket, handshakes it by calling ismaster again */
if (pooled && succeeded) {
request = mock_server_receives_ismaster (server);
mock_server_replies_ok_and_destroys (request);
}
if (succeeded) {
/* "foo" command succeeds */
ASSERT_OR_PRINT (future_get_bool (future), error);
} else {
ASSERT (!future_get_bool (future));
}
if (pooled) {
mongoc_client_pool_push (pool, client);
mongoc_client_pool_destroy (pool);
} else {
mongoc_client_destroy (client);
}
/* even if pooled, only topology scanner sends events, so we get one pair */
if (succeeded) {
expected_json = bson_strdup_printf (
"{'0': {'heartbeat_started_event': {'host': '%s'}},"
" '1': {'heartbeat_succeeded_event': {'host': '%s'}}}",
mock_server_get_host_and_port (server),
mock_server_get_host_and_port (server));
} else {
expected_json = bson_strdup_printf (
"{'0': {'heartbeat_started_event': {'host': '%s'}},"
" '1': {'heartbeat_failed_event': {'host': '%s'}}}",
mock_server_get_host_and_port (server),
mock_server_get_host_and_port (server));
}
check_json_apm_events (&context.events, tmp_bson (expected_json));
future_destroy (future);
bson_free (expected_json);
mongoc_uri_destroy (uri);
mock_server_destroy (server);
context_destroy (&context);
}
static void
test_mongoc_metadata_append_success (void)
{
mock_server_t *server;
mongoc_uri_t *uri;
mongoc_client_t *client;
mongoc_client_pool_t *pool;
request_t *request;
const bson_t *request_doc;
bson_iter_t iter;
bson_iter_t md_iter;
bson_iter_t inner_iter;
const char *val;
const char *driver_name = "php driver";
const char *driver_version = "version abc";
const char *platform = "./configure -nottoomanyflags";
char big_string [METADATA_MAX_SIZE];
memset (big_string, 'a', METADATA_MAX_SIZE - 1);
big_string [METADATA_MAX_SIZE - 1] = '\0';
_reset_metadata ();
/* Make sure setting the metadata works */
ASSERT (mongoc_metadata_append (driver_name, driver_version, platform));
server = mock_server_new ();
mock_server_run (server);
uri = mongoc_uri_copy (mock_server_get_uri (server));
mongoc_uri_set_option_as_int32 (uri, "heartbeatFrequencyMS", 500);
pool = mongoc_client_pool_new (uri);
/* Force topology scanner to start */
client = mongoc_client_pool_pop (pool);
request = mock_server_receives_ismaster (server);
ASSERT (request);
request_doc = request_get_doc (request, 0);
ASSERT (request_doc);
ASSERT (bson_has_field (request_doc, "isMaster"));
ASSERT (bson_has_field (request_doc, METADATA_FIELD));
ASSERT (bson_iter_init_find (&iter, request_doc, METADATA_FIELD));
ASSERT (bson_iter_recurse (&iter, &md_iter));
/* Make sure driver.name and driver.version and platform are all right */
ASSERT (bson_iter_find (&md_iter, "driver"));
ASSERT (BSON_ITER_HOLDS_DOCUMENT (&md_iter));
ASSERT (bson_iter_recurse (&md_iter, &inner_iter));
ASSERT (bson_iter_find (&inner_iter, "name"));
ASSERT (BSON_ITER_HOLDS_UTF8 (&inner_iter));
val = bson_iter_utf8 (&inner_iter, NULL);
ASSERT (val);
ASSERT (strstr (val, driver_name) != NULL);
ASSERT (bson_iter_find (&inner_iter, "version"));
ASSERT (BSON_ITER_HOLDS_UTF8 (&inner_iter));
val = bson_iter_utf8 (&inner_iter, NULL);
ASSERT (val);
ASSERT (strstr (val, driver_version));
/* Check os type not empty */
ASSERT (bson_iter_find (&md_iter, "os"));
ASSERT (BSON_ITER_HOLDS_DOCUMENT (&md_iter));
ASSERT (bson_iter_recurse (&md_iter, &inner_iter));
ASSERT (bson_iter_find (&inner_iter, "type"));
ASSERT (BSON_ITER_HOLDS_UTF8 (&inner_iter));
val = bson_iter_utf8 (&inner_iter, NULL);
ASSERT (val);
ASSERT (strlen (val) > 0);
/* Not checking os_name, as the spec says it can be NULL. */
/* Check platform field ok */
ASSERT (bson_iter_find (&md_iter, "platform"));
ASSERT (BSON_ITER_HOLDS_UTF8 (&md_iter));
val = bson_iter_utf8 (&md_iter, NULL);
ASSERT (val);
ASSERT (strstr (val, platform) != NULL);
mock_server_replies_simple (request,
"{'ok': 1, 'ismaster': true}");
request_destroy (request);
/* Cleanup */
mongoc_client_pool_push (pool, client);
mongoc_client_pool_destroy (pool);
mongoc_uri_destroy (uri);
mock_server_destroy (server);
_reset_metadata ();
}
/* Test the case where we can't prevent the metadata doc being too big
* and so we just don't send it */
static void
test_mongoc_metadata_cannot_send (void)
{
mock_server_t *server;
mongoc_uri_t *uri;
mongoc_client_t *client;
mongoc_client_pool_t *pool;
request_t *request;
const char *const server_reply = "{'ok': 1, 'ismaster': true}";
const bson_t *request_doc;
char big_string[METADATA_MAX_SIZE];
mongoc_metadata_t *md;
_reset_metadata ();
/* Mess with our global metadata struct so the metadata doc will be
* way too big */
memset (big_string, 'a', METADATA_MAX_SIZE - 1);
big_string[METADATA_MAX_SIZE - 1] = '\0';
md = _mongoc_metadata_get ();
bson_free (md->os_name);
md->os_name = bson_strdup (big_string);
server = mock_server_new ();
mock_server_run (server);
uri = mongoc_uri_copy (mock_server_get_uri (server));
mongoc_uri_set_option_as_int32 (uri, "heartbeatFrequencyMS", 500);
pool = mongoc_client_pool_new (uri);
/* Pop a client to trigger the topology scanner */
client = mongoc_client_pool_pop (pool);
request = mock_server_receives_ismaster (server);
/* Make sure the isMaster request DOESN'T have a metadata field: */
ASSERT (request);
request_doc = request_get_doc (request, 0);
ASSERT (request_doc);
ASSERT (bson_has_field (request_doc, "isMaster"));
ASSERT (!bson_has_field (request_doc, METADATA_FIELD));
mock_server_replies_simple (request, server_reply);
request_destroy (request);
/* Cause failure on client side */
request = mock_server_receives_ismaster (server);
ASSERT (request);
mock_server_hangs_up (request);
request_destroy (request);
/* Make sure the isMaster request still DOESN'T have a metadata field
* on subsequent heartbeats. */
request = mock_server_receives_ismaster (server);
ASSERT (request);
request_doc = request_get_doc (request, 0);
ASSERT (request_doc);
ASSERT (bson_has_field (request_doc, "isMaster"));
ASSERT (!bson_has_field (request_doc, METADATA_FIELD));
mock_server_replies_simple (request, server_reply);
request_destroy (request);
/* cleanup */
mongoc_client_pool_push (pool, client);
mongoc_client_pool_destroy (pool);
mongoc_uri_destroy (uri);
mock_server_destroy (server);
/* Reset again so the next tests don't have a metadata doc which
* is too big */
_reset_metadata ();
}
/*
* Append to the platform field a huge string
* Make sure that it gets truncated
*/
static void
test_mongoc_metadata_too_big (void)
{
mongoc_client_t *client;
mock_server_t *server;
mongoc_uri_t *uri;
future_t *future;
request_t *request;
const bson_t *ismaster_doc;
bson_iter_t iter;
enum { BUFFER_SIZE = METADATA_MAX_SIZE };
char big_string[BUFFER_SIZE];
uint32_t len;
const uint8_t *dummy;
server = mock_server_new ();
mock_server_run (server);
_reset_metadata ();
memset (big_string, 'a', BUFFER_SIZE - 1);
big_string[BUFFER_SIZE - 1] = '\0';
ASSERT (mongoc_metadata_append (NULL, NULL, big_string));
uri = mongoc_uri_copy (mock_server_get_uri (server));
client = mongoc_client_new_from_uri (uri);
ASSERT (mongoc_client_set_appname (client, "my app"));
/* Send a ping, mock server deals with it */
future = future_client_command_simple (client,
"admin",
tmp_bson ("{'ping': 1}"),
NULL,
NULL,
NULL);
request = mock_server_receives_ismaster (server);
/* Make sure the isMaster request has a metadata field, and it's not huge */
ASSERT (request);
ismaster_doc = request_get_doc (request, 0);
ASSERT (ismaster_doc);
ASSERT (bson_has_field (ismaster_doc, "isMaster"));
ASSERT (bson_has_field (ismaster_doc, METADATA_FIELD));
/* isMaster with metadata isn't too big */
bson_iter_init_find (&iter,
ismaster_doc,
METADATA_FIELD);
ASSERT (BSON_ITER_HOLDS_DOCUMENT (&iter));
bson_iter_document (&iter, &len, &dummy);
/* Should have truncated the platform field so it fits exactly */
ASSERT (len == METADATA_MAX_SIZE);
mock_server_replies_simple (request, "{'ok': 1}");
request_destroy (request);
request = mock_server_receives_command (server, "admin",
MONGOC_QUERY_SLAVE_OK,
"{'ping': 1}");
mock_server_replies_simple (request, "{'ok': 1}");
ASSERT (future_get_bool (future));
future_destroy (future);
request_destroy (request);
mongoc_client_destroy (client);
mongoc_uri_destroy (uri);
mock_server_destroy (server);
/* So later tests don't have "aaaaa..." as the md platform string */
_reset_metadata ();
}
