/*
* 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.
*/
#ifndef MONGO_LINK_H
#define MONGO_LINK_H
#include "perl_mongo.h"
#ifdef WIN32
#include <winsock2.h>
#define socklen_t int
#else
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <netinet/tcp.h>
#include <fcntl.h>
#include <netdb.h>
#endif
#include <errno.h>
#ifdef MONGO_SSL
#include <openssl/rand.h>
#include <openssl/ssl.h>
#include <openssl/err.h>
#endif
#ifdef MONGO_SASL
#include <gsasl.h>
#endif
// db ops
#define OP_REPLY 1
#define OP_MSG 1000
#define OP_UPDATE 2001
#define OP_INSERT 2002
#define OP_GET_BY_OID 2003
#define OP_QUERY 2004
#define OP_GET_MORE 2005
#define OP_DELETE 2006
#define OP_KILL_CURSORS 2007
// cursor flags
#define CURSOR_NOT_FOUND 1
#define CURSOR_ERR 2
#define MSG_HEADER_SIZE 16
#define REPLY_HEADER_SIZE (MSG_HEADER_SIZE+20)
#define INITIAL_BUF_SIZE 4096
// should only be 4MB, can be 64MB with big docs
#define MAX_RESPONSE_LEN 67108864
#define DEFAULT_CHUNK_SIZE (256*1024)
// if _id field should be added
#define PREP 1
#define NO_PREP 0
#define CREATE_MSG_HEADER(rid, rto, opcode) \
header.length = 0; \
header.request_id = rid; \
header.response_to = rto; \
header.op = opcode;
#define CREATE_RESPONSE_HEADER(buf, ns, rto, opcode) \
sv_setiv(request_id, SvIV(request_id)+1); \
CREATE_MSG_HEADER(SvIV(request_id), rto, opcode); \
APPEND_HEADER_NS(buf, ns, 0);
#define CREATE_HEADER_WITH_OPTS(buf, ns, opcode, opts) \
sv_setiv(request_id, SvIV(request_id)+1); \
CREATE_MSG_HEADER(SvIV(request_id), 0, opcode); \
APPEND_HEADER_NS(buf, ns, opts);
#define CREATE_HEADER(buf, ns, opcode) \
CREATE_RESPONSE_HEADER(buf, ns, 0, opcode);
#define APPEND_HEADER(buf, opts) buf.pos += INT_32; \
perl_mongo_serialize_int(&buf, header.request_id); \
perl_mongo_serialize_int(&buf, header.response_to); \
perl_mongo_serialize_int(&buf, header.op); \
perl_mongo_serialize_int(&buf, opts);
#define APPEND_HEADER_NS(buf, ns, opts) \
APPEND_HEADER(buf, opts); \
perl_mongo_serialize_string(&buf, ns, strlen(ns));
#define CREATE_BUF(size) \
Newx(buf.start, size, char); \
buf.pos = buf.start; \
buf.end = buf.start + size;
typedef struct {
int length;
int request_id;
int response_to;
int op;
} mongo_msg_header;
/*
* a connection to the database
*
* host is hostname
* port is port number
* socket is the actual socket the connection is using
* connected is a boolean indicating if the socket is connected or not
*/
typedef struct _mongo_server {
char *host;
int port;
int socket;
int connected;
} mongo_server;
/*
* auto_reconnect is whether to reconnect on disconnect
* timeout is how long to try to connect before failing
* num is the number of servers in this set
* master is the index of the master server, if there is more than 1 server
* server is an array of pointers to connections
*/
typedef struct {
int auto_reconnect;
int timeout;
int num;
mongo_server *master;
int copy;
bool ssl;
#ifdef MONGO_SSL
SSL *ssl_handle;
SSL_CTX *ssl_context;
#endif
int (*sender)(void* link, const char* buffer, size_t len);
int (*receiver)(void* link, const char* buffer, size_t len);
} mongo_link;
typedef struct {
// response header
mongo_msg_header header;
// response fields
int flag;
int64_t cursor_id;
int start;
// number of results used
int at;
// number results returned
int num;
// results
buffer buf;
int started_iterating;
} mongo_cursor;
int mongo_link_say(SV *self, buffer *buf);
int mongo_link_hear(SV *self);
int perl_mongo_master(SV *self, int auto_reconnect);
void set_disconnected(SV *link_sv);
//ssl
void perl_mongo_connect( SV *client, mongo_link* link);
void non_ssl_connect(mongo_link* link);
#ifdef MONGO_SSL
void tcp_setup(mongo_link* link);
void ssl_connect(mongo_link* link);
void ssl_disconnect (mongo_link *link);
int ssl_send(void* link, const char* buffer, size_t len);
int ssl_recv(void* link, const char* buffer, size_t len);
#endif
int non_ssl_send(void* link, const char* buffer, size_t len);
int non_ssl_recv(void* link, const char* buffer, size_t len);
#endif