bson_ext 1.3.1 → 1.4.0

data/ext/cmongo/c-driver/src/mongo.h

@@ -0,0 +1,260 @@
+/* mongo.h */
+
+/*    Copyright 2009, 2010 10gen Inc.
+ *
+ *    Licensed under the Apache License, Version 2.0 (the "License");
+ *    you may not use this file except in compliance with the License.
+ *    You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *    Unless required by applicable law or agreed to in writing, software
+ *    distributed under the License is distributed on an "AS IS" BASIS,
+ *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *    See the License for the specific language governing permissions and
+ *    limitations under the License.
+ */
+
+#ifndef _MONGO_H_
+#define _MONGO_H_
+
+#include "mongo_except.h"
+#include "bson.h"
+
+#ifdef _WIN32
+#include <windows.h>
+#include <winsock.h>
+#define mongo_close_socket(sock) ( closesocket(sock) )
+typedef int socklen_t;
+#else
+#include <arpa/inet.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netdb.h>
+#include <netinet/in.h>
+#include <netinet/tcp.h>
+#define mongo_close_socket(sock) ( close(sock) )
+#endif
+
+#if defined(_XOPEN_SOURCE) || defined(_POSIX_SOURCE) || _POSIX_C_SOURCE >= 1
+#define _MONGO_USE_GETADDRINFO
+#endif
+
+MONGO_EXTERN_C_START
+
+typedef struct mongo_host_port {
+    char host[255];
+    int port;
+    struct mongo_host_port* next;
+} mongo_host_port;
+
+typedef struct {
+    mongo_host_port* seeds; /*< The list of seed nodes provided by the user. */
+    mongo_host_port* hosts; /*< The list of host and ports reported by the replica set */
+    char* name;             /*< The name of the replica set. */
+    bson_bool_t primary_connected; /*< Whether we've managed to connect to a primary node. */
+} mongo_replset;
+
+typedef struct {
+    mongo_host_port* primary;
+    mongo_replset* replset;
+    int sock;
+    bson_bool_t connected;
+    mongo_exception_context exception;
+} mongo_connection;
+
+#pragma pack(1)
+typedef struct {
+    int len;
+    int id;
+    int responseTo;
+    int op;
+} mongo_header;
+
+typedef struct {
+    mongo_header head;
+    char data;
+} mongo_message;
+
+typedef struct {
+    int flag; /* non-zero on failure */
+    int64_t cursorID;
+    int start;
+    int num;
+} mongo_reply_fields;
+
+typedef struct {
+    mongo_header head;
+    mongo_reply_fields fields;
+    char objs;
+} mongo_reply;
+#pragma pack()
+
+typedef struct {
+    mongo_reply * mm; /* message is owned by cursor */
+    mongo_connection * conn; /* connection is *not* owned by cursor */
+    const char* ns; /* owned by cursor */
+    bson current;
+} mongo_cursor;
+
+enum mongo_operations {
+    mongo_op_msg = 1000,    /* generic msg command followed by a string */
+    mongo_op_update = 2001, /* update object */
+    mongo_op_insert = 2002,
+    mongo_op_query = 2004,
+    mongo_op_get_more = 2005,
+    mongo_op_delete = 2006,
+    mongo_op_kill_cursors = 2007
+};
+
+
+/*
+ * CONNECTIONS
+ */
+typedef enum {
+    mongo_conn_success = 0,
+    mongo_conn_bad_arg,
+    mongo_conn_no_socket,
+    mongo_conn_fail,
+    mongo_conn_not_master, /* leaves conn connected to slave */
+    mongo_conn_bad_set_name, /* The provided replica set name doesn't match the existing replica set */
+    mongo_conn_cannot_find_primary
+} mongo_conn_return;
+
+/*
+ * Connect to a single MongoDB server.
+ *
+ * @param conn a mongo_connection object.
+ * @param host a numerical network address or a network hostname.
+ * @param port the port to connect to.
+ *
+ * @return mongo_conn_return
+ */
+mongo_conn_return mongo_connect( mongo_connection * conn , const char* host, int port );
+
+/* 
+ * Initialize a connection object for connecting with a replica set.
+ *
+ * @param conn a mongo_connection object.
+ * @param name the name of the replica set to connect to.
+ * */
+void mongo_replset_init_conn( mongo_connection* conn, const char* name );
+
+/*
+ * Add a seed node to the connection object.
+ *
+ * You must specify at least one seed node before connecting to a replica set.
+ *
+ * @param conn a mongo_connection object.
+ * @param host a numerical network address or a network hostname.
+ * @param port the port to connect to.
+ */
+int mongo_replset_add_seed( mongo_connection* conn, const char* host, int port );
+
+/*
+ * Connect to a replica set.
+ *
+ * Before passing a connection object to this method, you must already have called
+ * mongo_replset_init_conn and mongo_replset_add_seed.
+ *
+ * @param conn a mongo_connection object.
+ *
+ * @return mongo_conn_return
+ */
+mongo_conn_return mongo_replset_connect( mongo_connection* conn );
+
+/*
+ * Try reconnecting to the server using the existing connection settings.
+ *
+ * This method will disconnect the current socket. If you've authentication,
+ * you'll need to re-authenticate after calling this function.
+ *
+ * @param conn
+ *
+ * @return mongo_conn_return
+ */
+mongo_conn_return mongo_reconnect( mongo_connection * conn );
+
+/*
+ * Close the current connection to the server.
+ */
+bson_bool_t mongo_disconnect( mongo_connection * conn );
+
+/*
+ * Close any existing connection to the server and free all allocated
+ * memory associated with the conn object.
+ *
+ * You must always call this method when finished with the connection object.
+ *
+ * @param conn
+ *
+ * @return bson_bool_t
+ */
+bson_bool_t mongo_destroy( mongo_connection * conn );
+
+/* ----------------------------
+   CORE METHODS - insert update remove query getmore
+   ------------------------------ */
+
+void mongo_insert( mongo_connection * conn , const char * ns , bson * data );
+void mongo_insert_batch( mongo_connection * conn , const char * ns , bson ** data , int num );
+
+static const int MONGO_UPDATE_UPSERT = 0x1;
+static const int MONGO_UPDATE_MULTI = 0x2;
+void mongo_update(mongo_connection* conn, const char* ns, const bson* cond, const bson* op, int flags);
+
+void mongo_remove(mongo_connection* conn, const char* ns, const bson* cond);
+
+mongo_cursor* mongo_find(mongo_connection* conn, const char* ns, bson* query, bson* fields ,int nToReturn ,int nToSkip, int options);
+bson_bool_t mongo_cursor_next(mongo_cursor* cursor);
+void mongo_cursor_destroy(mongo_cursor* cursor);
+
+/* out can be NULL if you don't care about results. useful for commands */
+bson_bool_t mongo_find_one(mongo_connection* conn, const char* ns, bson* query, bson* fields, bson* out);
+
+int64_t mongo_count(mongo_connection* conn, const char* db, const char* coll, bson* query);
+
+/* ----------------------------
+   HIGHER LEVEL - indexes - command helpers eval
+   ------------------------------ */
+
+/* Returns true on success */
+/* WARNING: Unlike other drivers these do not cache results */
+
+static const int MONGO_INDEX_UNIQUE = 0x1;
+static const int MONGO_INDEX_DROP_DUPS = 0x2;
+bson_bool_t mongo_create_index(mongo_connection * conn, const char * ns, bson * key, int options, bson * out);
+bson_bool_t mongo_create_simple_index(mongo_connection * conn, const char * ns, const char* field, int options, bson * out);
+
+/* ----------------------------
+   COMMANDS
+   ------------------------------ */
+
+bson_bool_t mongo_run_command(mongo_connection * conn, const char * db, bson * command, bson * out);
+
+/* for simple commands with a single k-v pair */
+bson_bool_t mongo_simple_int_command(mongo_connection * conn, const char * db, const char* cmd,         int arg, bson * out);
+bson_bool_t mongo_simple_str_command(mongo_connection * conn, const char * db, const char* cmd, const char* arg, bson * out);
+
+bson_bool_t mongo_cmd_drop_db(mongo_connection * conn, const char * db);
+bson_bool_t mongo_cmd_drop_collection(mongo_connection * conn, const char * db, const char * collection, bson * out);
+
+void mongo_cmd_add_user(mongo_connection* conn, const char* db, const char* user, const char* pass);
+bson_bool_t mongo_cmd_authenticate(mongo_connection* conn, const char* db, const char* user, const char* pass);
+
+/* return value is master status */
+bson_bool_t mongo_cmd_ismaster(mongo_connection * conn, bson * out);
+
+/* true return indicates error */
+bson_bool_t mongo_cmd_get_last_error(mongo_connection * conn, const char * db, bson * out);
+bson_bool_t mongo_cmd_get_prev_error(mongo_connection * conn, const char * db, bson * out);
+void        mongo_cmd_reset_error(mongo_connection * conn, const char * db);
+
+/* ----------------------------
+   UTILS
+   ------------------------------ */
+
+MONGO_EXTERN_C_END
+
+
+#endif

data/ext/cmongo/c-driver/src/mongo_except.h

@@ -0,0 +1,143 @@
+/* mongo_except.h */
+
+/*    Copyright 2009 10gen Inc.
+ *
+ *    Licensed under the Apache License, Version 2.0 (the "License");
+ *    you may not use this file except in compliance with the License.
+ *    You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *    Unless required by applicable law or agreed to in writing, software
+ *    distributed under the License is distributed on an "AS IS" BASIS,
+ *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *    See the License for the specific language governing permissions and
+ *    limitations under the License.
+ */
+
+/* This file is based loosely on cexcept (http://www.nicemice.net/cexcept/). I
+ * have modified it to work better with mongo's API.
+ *
+ * The MONGO_TRY, MONGO_CATCH, and MONGO_TROW macros assume that a pointer to
+ * the current connection is available as 'conn'. If you would like to use a
+ * different name, use the _GENERIC version of these macros.
+ *
+ * WARNING: do not return or otherwise jump (excluding MONGO_TRHOW()) out of a
+ * MONGO_TRY block as the nessesary clean-up code will not be called. Jumping
+ * out of the MONGO_CATCH block is OK.
+ */
+
+#ifdef MONGO_CODE_EXAMPLE
+    mongo_connection conn[1]; /* makes conn a ptr to the connection */
+
+    MONGO_TRY{
+        mongo_find_one(...);
+        MONGO_THROW(conn, MONGO_EXCEPT_NETWORK);
+    }MONGO_CATCH{
+        switch(conn->exception->type){
+            case MONGO_EXCEPT_NETWORK:
+                do_something();
+            case MONGO_EXCEPT_FIND_ERR:
+                do_something();
+            default:
+                MONGO_RETHROW();
+        }
+    }
+#endif
+
+ /* ORIGINAL CEXEPT COPYRIGHT:
+cexcept: README 2.0.1 (2008-Jul-23-Wed)
+http://www.nicemice.net/cexcept/
+Adam M. Costello
+http://www.nicemice.net/amc/
+
+The package is both free-as-in-speech and free-as-in-beer:
+
+    Copyright (c) 2000-2008 Adam M. Costello and Cosmin Truta.
+    This package may be modified only if its author and version
+    information is updated accurately, and may be redistributed
+    only if accompanied by this unaltered notice.  Subject to those
+    restrictions, permission is granted to anyone to do anything with
+    this package.  The copyright holders make no guarantees regarding
+    this package, and are not responsible for any damage resulting from
+    its use.
+ */
+
+#ifndef _MONGO_EXCEPT_H_
+#define _MONGO_EXCEPT_H_
+
+#include <setjmp.h>
+
+/* always non-zero */
+typedef enum{
+    MONGO_EXCEPT_NETWORK=1,
+    MONGO_EXCEPT_FIND_ERR
+}mongo_exception_type;
+
+
+typedef struct {
+  jmp_buf base_handler;
+  jmp_buf *penv;
+  int caught;
+  volatile mongo_exception_type type;
+}mongo_exception_context;
+
+#define MONGO_TRY MONGO_TRY_GENERIC(conn)
+#define MONGO_CATCH MONGO_CATCH_GENERIC(conn)
+#define MONGO_THROW(e) MONGO_THROW_GENERIC(conn, e)
+#define MONGO_RETHROW() MONGO_RETHROW_GENERIC(conn)
+
+/* the rest of this file is implementation details */
+
+/* this is done in mongo_connect */
+#define MONGO_INIT_EXCEPTION(exception_ptr) \
+    do{ \
+        mongo_exception_type t; /* exception_ptr won't be available */\
+        (exception_ptr)->penv = &(exception_ptr)->base_handler; \
+        if ((t = setjmp((exception_ptr)->base_handler))) { /* yes, '=' is correct */ \
+            switch(t){ \
+                case MONGO_EXCEPT_NETWORK: bson_fatal_msg(0, "network error"); \
+                case MONGO_EXCEPT_FIND_ERR: bson_fatal_msg(0, "error in find"); \
+                default: bson_fatal_msg(0, "unknown exception"); \
+            } \
+        } \
+    }while(0)
+
+#define MONGO_TRY_GENERIC(connection) \
+  { \
+    jmp_buf *exception__prev, exception__env; \
+    exception__prev = (connection)->exception.penv; \
+    (connection)->exception.penv = &exception__env; \
+    if (setjmp(exception__env) == 0) { \
+      do
+
+#define MONGO_CATCH_GENERIC(connection) \
+      while ((connection)->exception.caught = 0, \
+             (connection)->exception.caught); \
+    } \
+    else { \
+      (connection)->exception.caught = 1; \
+    } \
+    (connection)->exception.penv = exception__prev; \
+  } \
+  if (!(connection)->exception.caught ) { } \
+  else
+
+/* Try ends with do, and Catch begins with while(0) and ends with     */
+/* else, to ensure that Try/Catch syntax is similar to if/else        */
+/* syntax.                                                            */
+/*                                                                    */
+/* The 0 in while(0) is expressed as x=0,x in order to appease        */
+/* compilers that warn about constant expressions inside while().     */
+/* Most compilers should still recognize that the condition is always */
+/* false and avoid generating code for it.                            */
+
+#define MONGO_THROW_GENERIC(connection, type_in) \
+  for (;; longjmp(*(connection)->exception.penv, type_in)) \
+    (connection)->exception.type = type_in
+
+#define MONGO_RETHROW_GENERIC(connection) \
+    MONGO_THROW_GENERIC(connection, (connection)->exception.type)
+
+
+#endif

data/ext/cmongo/c-driver/src/numbers.c

@@ -0,0 +1,127 @@
+/*    Copyright 2009 10gen Inc.
+ *
+ *    Licensed under the Apache License, Version 2.0 (the "License");
+ *    you may not use this file except in compliance with the License.
+ *    You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *    Unless required by applicable law or agreed to in writing, software
+ *    distributed under the License is distributed on an "AS IS" BASIS,
+ *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *    See the License for the specific language governing permissions and
+ *    limitations under the License.
+ */
+
+/* all the numbers that fit in a 4 byte string */
+const char bson_numstrs[1000][4] = {
+    "0",  "1",  "2",  "3",  "4",  "5",  "6",  "7",  "8",  "9",
+    "10", "11", "12", "13", "14", "15", "16", "17", "18", "19",
+    "20", "21", "22", "23", "24", "25", "26", "27", "28", "29",
+    "30", "31", "32", "33", "34", "35", "36", "37", "38", "39",
+    "40", "41", "42", "43", "44", "45", "46", "47", "48", "49",
+    "50", "51", "52", "53", "54", "55", "56", "57", "58", "59",
+    "60", "61", "62", "63", "64", "65", "66", "67", "68", "69",
+    "70", "71", "72", "73", "74", "75", "76", "77", "78", "79",
+    "80", "81", "82", "83", "84", "85", "86", "87", "88", "89",
+    "90", "91", "92", "93", "94", "95", "96", "97", "98", "99",
+
+    "100", "101", "102", "103", "104", "105", "106", "107", "108", "109",
+    "110", "111", "112", "113", "114", "115", "116", "117", "118", "119",
+    "120", "121", "122", "123", "124", "125", "126", "127", "128", "129",
+    "130", "131", "132", "133", "134", "135", "136", "137", "138", "139",
+    "140", "141", "142", "143", "144", "145", "146", "147", "148", "149",
+    "150", "151", "152", "153", "154", "155", "156", "157", "158", "159",
+    "160", "161", "162", "163", "164", "165", "166", "167", "168", "169",
+    "170", "171", "172", "173", "174", "175", "176", "177", "178", "179",
+    "180", "181", "182", "183", "184", "185", "186", "187", "188", "189",
+    "190", "191", "192", "193", "194", "195", "196", "197", "198", "199",
+
+    "200", "201", "202", "203", "204", "205", "206", "207", "208", "209",
+    "210", "211", "212", "213", "214", "215", "216", "217", "218", "219",
+    "220", "221", "222", "223", "224", "225", "226", "227", "228", "229",
+    "230", "231", "232", "233", "234", "235", "236", "237", "238", "239",
+    "240", "241", "242", "243", "244", "245", "246", "247", "248", "249",
+    "250", "251", "252", "253", "254", "255", "256", "257", "258", "259",
+    "260", "261", "262", "263", "264", "265", "266", "267", "268", "269",
+    "270", "271", "272", "273", "274", "275", "276", "277", "278", "279",
+    "280", "281", "282", "283", "284", "285", "286", "287", "288", "289",
+    "290", "291", "292", "293", "294", "295", "296", "297", "298", "299",
+
+    "300", "301", "302", "303", "304", "305", "306", "307", "308", "309",
+    "310", "311", "312", "313", "314", "315", "316", "317", "318", "319",
+    "320", "321", "322", "323", "324", "325", "326", "327", "328", "329",
+    "330", "331", "332", "333", "334", "335", "336", "337", "338", "339",
+    "340", "341", "342", "343", "344", "345", "346", "347", "348", "349",
+    "350", "351", "352", "353", "354", "355", "356", "357", "358", "359",
+    "360", "361", "362", "363", "364", "365", "366", "367", "368", "369",
+    "370", "371", "372", "373", "374", "375", "376", "377", "378", "379",
+    "380", "381", "382", "383", "384", "385", "386", "387", "388", "389",
+    "390", "391", "392", "393", "394", "395", "396", "397", "398", "399",
+
+    "400", "401", "402", "403", "404", "405", "406", "407", "408", "409",
+    "410", "411", "412", "413", "414", "415", "416", "417", "418", "419",
+    "420", "421", "422", "423", "424", "425", "426", "427", "428", "429",
+    "430", "431", "432", "433", "434", "435", "436", "437", "438", "439",
+    "440", "441", "442", "443", "444", "445", "446", "447", "448", "449",
+    "450", "451", "452", "453", "454", "455", "456", "457", "458", "459",
+    "460", "461", "462", "463", "464", "465", "466", "467", "468", "469",
+    "470", "471", "472", "473", "474", "475", "476", "477", "478", "479",
+    "480", "481", "482", "483", "484", "485", "486", "487", "488", "489",
+    "490", "491", "492", "493", "494", "495", "496", "497", "498", "499",
+
+    "500", "501", "502", "503", "504", "505", "506", "507", "508", "509",
+    "510", "511", "512", "513", "514", "515", "516", "517", "518", "519",
+    "520", "521", "522", "523", "524", "525", "526", "527", "528", "529",
+    "530", "531", "532", "533", "534", "535", "536", "537", "538", "539",
+    "540", "541", "542", "543", "544", "545", "546", "547", "548", "549",
+    "550", "551", "552", "553", "554", "555", "556", "557", "558", "559",
+    "560", "561", "562", "563", "564", "565", "566", "567", "568", "569",
+    "570", "571", "572", "573", "574", "575", "576", "577", "578", "579",
+    "580", "581", "582", "583", "584", "585", "586", "587", "588", "589",
+    "590", "591", "592", "593", "594", "595", "596", "597", "598", "599",
+
+    "600", "601", "602", "603", "604", "605", "606", "607", "608", "609",
+    "610", "611", "612", "613", "614", "615", "616", "617", "618", "619",
+    "620", "621", "622", "623", "624", "625", "626", "627", "628", "629",
+    "630", "631", "632", "633", "634", "635", "636", "637", "638", "639",
+    "640", "641", "642", "643", "644", "645", "646", "647", "648", "649",
+    "650", "651", "652", "653", "654", "655", "656", "657", "658", "659",
+    "660", "661", "662", "663", "664", "665", "666", "667", "668", "669",
+    "670", "671", "672", "673", "674", "675", "676", "677", "678", "679",
+    "680", "681", "682", "683", "684", "685", "686", "687", "688", "689",
+    "690", "691", "692", "693", "694", "695", "696", "697", "698", "699",
+
+    "700", "701", "702", "703", "704", "705", "706", "707", "708", "709",
+    "710", "711", "712", "713", "714", "715", "716", "717", "718", "719",
+    "720", "721", "722", "723", "724", "725", "726", "727", "728", "729",
+    "730", "731", "732", "733", "734", "735", "736", "737", "738", "739",
+    "740", "741", "742", "743", "744", "745", "746", "747", "748", "749",
+    "750", "751", "752", "753", "754", "755", "756", "757", "758", "759",
+    "760", "761", "762", "763", "764", "765", "766", "767", "768", "769",
+    "770", "771", "772", "773", "774", "775", "776", "777", "778", "779",
+    "780", "781", "782", "783", "784", "785", "786", "787", "788", "789",
+    "790", "791", "792", "793", "794", "795", "796", "797", "798", "799",
+
+    "800", "801", "802", "803", "804", "805", "806", "807", "808", "809",
+    "810", "811", "812", "813", "814", "815", "816", "817", "818", "819",
+    "820", "821", "822", "823", "824", "825", "826", "827", "828", "829",
+    "830", "831", "832", "833", "834", "835", "836", "837", "838", "839",
+    "840", "841", "842", "843", "844", "845", "846", "847", "848", "849",
+    "850", "851", "852", "853", "854", "855", "856", "857", "858", "859",
+    "860", "861", "862", "863", "864", "865", "866", "867", "868", "869",
+    "870", "871", "872", "873", "874", "875", "876", "877", "878", "879",
+    "880", "881", "882", "883", "884", "885", "886", "887", "888", "889",
+    "890", "891", "892", "893", "894", "895", "896", "897", "898", "899",
+
+    "900", "901", "902", "903", "904", "905", "906", "907", "908", "909",
+    "910", "911", "912", "913", "914", "915", "916", "917", "918", "919",
+    "920", "921", "922", "923", "924", "925", "926", "927", "928", "929",
+    "930", "931", "932", "933", "934", "935", "936", "937", "938", "939",
+    "940", "941", "942", "943", "944", "945", "946", "947", "948", "949",
+    "950", "951", "952", "953", "954", "955", "956", "957", "958", "959",
+    "960", "961", "962", "963", "964", "965", "966", "967", "968", "969",
+    "970", "971", "972", "973", "974", "975", "976", "977", "978", "979",
+    "980", "981", "982", "983", "984", "985", "986", "987", "988", "989",
+    "990", "991", "992", "993", "994", "995", "996", "997", "998", "999",
+};