int
main (int argc, char *argv[])
{
const char *uri_str = "mongodb://localhost:27017";
mongoc_client_t *client;
mongoc_database_t *database;
mongoc_collection_t *collection;
bson_t *command, reply, *insert;
bson_error_t error;
char *str;
bool retval;
/*
* Required to initialize libmongoc's internals
*/
mongoc_init ();
/*
* Optionally get MongoDB URI from command line
*/
if (argc > 1) {
uri_str = argv[1];
}
/*
* Create a new client instance
*/
client = mongoc_client_new (uri_str);
/*
* Register the application name so we can track it in the profile logs
* on the server. This can also be done from the URI (see other examples).
*/
mongoc_client_set_appname (client, "connect-example");
/*
* Get a handle on the database "db_name" and collection "coll_name"
*/
database = mongoc_client_get_database (client, "db_name");
collection = mongoc_client_get_collection (client, "db_name", "coll_name");
/*
* Do work. This example pings the database, prints the result as JSON and
* performs an insert
*/
command = BCON_NEW ("ping", BCON_INT32 (1));
retval = mongoc_client_command_simple (
client, "admin", command, NULL, &reply, &error);
if (!retval) {
fprintf (stderr, "%s\n", error.message);
return EXIT_FAILURE;
}
str = bson_as_json (&reply, NULL);
printf ("%s\n", str);
insert = BCON_NEW ("hello", BCON_UTF8 ("world"));
if (!mongoc_collection_insert (
collection, MONGOC_INSERT_NONE, insert, NULL, &error)) {
fprintf (stderr, "%s\n", error.message);
}
bson_destroy (insert);
bson_destroy (&reply);
bson_destroy (command);
bson_free (str);
/*
* Release our handles and clean up libmongoc
*/
mongoc_collection_destroy (collection);
mongoc_database_destroy (database);
mongoc_client_destroy (client);
mongoc_cleanup ();
return 0;
}
/*
* 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 ();
}
