libcouchbase 0.2.0 → 0.3.1

Sign up to get free protection for your applications and to get access to all the features.
Files changed (129) hide show
  1. checksums.yaml +4 -4
  2. data/.travis.yml +1 -1
  3. data/README.md +1 -1
  4. data/ext/libcouchbase/CMakeLists.txt +8 -6
  5. data/ext/libcouchbase/README.markdown +2 -2
  6. data/ext/libcouchbase/RELEASE_NOTES.markdown +0 -86
  7. data/ext/libcouchbase/cmake/Modules/ConfigureDtrace.cmake +0 -11
  8. data/ext/libcouchbase/cmake/Modules/GenerateConfigDotH.cmake +0 -2
  9. data/ext/libcouchbase/cmake/Modules/GetLibcouchbaseFlags.cmake +1 -2
  10. data/ext/libcouchbase/cmake/Modules/GetVersionInfo.cmake +3 -3
  11. data/ext/libcouchbase/cmake/config-cmake.h.in +0 -2
  12. data/ext/libcouchbase/cmake/defs.mk.in +2 -0
  13. data/ext/libcouchbase/cmake/source_files.cmake +5 -21
  14. data/ext/libcouchbase/include/libcouchbase/auth.h +0 -10
  15. data/ext/libcouchbase/include/libcouchbase/cntl.h +1 -27
  16. data/ext/libcouchbase/include/libcouchbase/error.h +1 -15
  17. data/ext/libcouchbase/include/libcouchbase/n1ql.h +1 -13
  18. data/ext/libcouchbase/include/libcouchbase/plugins/io/bsdio-inl.c +1 -1
  19. data/ext/libcouchbase/include/libcouchbase/subdoc.h +0 -9
  20. data/ext/libcouchbase/include/libcouchbase/views.h +1 -7
  21. data/ext/libcouchbase/include/libcouchbase/visibility.h +0 -1
  22. data/ext/libcouchbase/include/memcached/protocol_binary.h +1131 -29
  23. data/ext/libcouchbase/include/memcached/vbucket.h +42 -0
  24. data/ext/libcouchbase/packaging/parse-git-describe.pl +1 -1
  25. data/ext/libcouchbase/plugins/io/libev/libev_io_opts.h +2 -3
  26. data/ext/libcouchbase/src/README.md +2 -0
  27. data/ext/libcouchbase/src/auth-priv.h +0 -1
  28. data/ext/libcouchbase/src/auth.cc +4 -10
  29. data/ext/libcouchbase/src/bootstrap.c +269 -0
  30. data/ext/libcouchbase/src/bootstrap.h +39 -50
  31. data/ext/libcouchbase/src/bucketconfig/bc_cccp.cc +117 -84
  32. data/ext/libcouchbase/src/bucketconfig/bc_file.c +347 -0
  33. data/ext/libcouchbase/src/bucketconfig/bc_http.c +630 -0
  34. data/ext/libcouchbase/src/bucketconfig/bc_http.h +25 -50
  35. data/ext/libcouchbase/src/bucketconfig/bc_mcraw.c +150 -0
  36. data/ext/libcouchbase/src/bucketconfig/clconfig.h +386 -407
  37. data/ext/libcouchbase/src/bucketconfig/confmon.c +474 -0
  38. data/ext/libcouchbase/src/cbft.cc +27 -22
  39. data/ext/libcouchbase/src/cntl.cc +19 -30
  40. data/ext/libcouchbase/src/connspec.cc +1 -48
  41. data/ext/libcouchbase/src/connspec.h +0 -27
  42. data/ext/libcouchbase/src/dump.cc +2 -2
  43. data/ext/libcouchbase/src/getconfig.cc +33 -7
  44. data/ext/libcouchbase/src/handler.cc +2 -0
  45. data/ext/libcouchbase/src/hostlist.cc +36 -0
  46. data/ext/libcouchbase/src/hostlist.h +62 -41
  47. data/ext/libcouchbase/src/http/http-priv.h +112 -125
  48. data/ext/libcouchbase/src/http/http.cc +30 -15
  49. data/ext/libcouchbase/src/http/http.h +34 -1
  50. data/ext/libcouchbase/src/http/http_io.cc +26 -22
  51. data/ext/libcouchbase/src/instance.cc +23 -94
  52. data/ext/libcouchbase/src/internal.h +26 -52
  53. data/ext/libcouchbase/src/jsparse/parser.cc +202 -146
  54. data/ext/libcouchbase/src/jsparse/parser.h +98 -91
  55. data/ext/libcouchbase/src/lcbht/lcbht.c +282 -0
  56. data/ext/libcouchbase/src/lcbht/lcbht.h +163 -174
  57. data/ext/libcouchbase/src/lcbio/connect.c +557 -0
  58. data/ext/libcouchbase/src/lcbio/connect.h +2 -9
  59. data/ext/libcouchbase/src/lcbio/ctx.c +1 -1
  60. data/ext/libcouchbase/src/lcbio/iotable.h +16 -61
  61. data/ext/libcouchbase/src/lcbio/ioutils.h +1 -1
  62. data/ext/libcouchbase/src/lcbio/manager.c +2 -2
  63. data/ext/libcouchbase/src/mc/mcreq.h +2 -9
  64. data/ext/libcouchbase/src/mcserver/mcserver.cc +34 -143
  65. data/ext/libcouchbase/src/mcserver/mcserver.h +12 -7
  66. data/ext/libcouchbase/src/mcserver/negotiate.cc +38 -132
  67. data/ext/libcouchbase/src/n1ql/ixmgmt.cc +2 -1
  68. data/ext/libcouchbase/src/n1ql/n1ql.cc +32 -56
  69. data/ext/libcouchbase/src/newconfig.cc +6 -6
  70. data/ext/libcouchbase/src/nodeinfo.cc +2 -2
  71. data/ext/libcouchbase/src/operations/{cbflush.cc → cbflush.c} +15 -7
  72. data/ext/libcouchbase/src/operations/{counter.cc → counter.c} +0 -0
  73. data/ext/libcouchbase/src/operations/durability.cc +26 -6
  74. data/ext/libcouchbase/src/operations/durability_internal.h +3 -6
  75. data/ext/libcouchbase/src/operations/{get.cc → get.c} +26 -24
  76. data/ext/libcouchbase/src/operations/{observe.cc → observe.c} +93 -68
  77. data/ext/libcouchbase/src/operations/{pktfwd.cc → pktfwd.c} +0 -0
  78. data/ext/libcouchbase/src/operations/{remove.cc → remove.c} +0 -0
  79. data/ext/libcouchbase/src/operations/stats.cc +8 -3
  80. data/ext/libcouchbase/src/operations/{store.cc → store.c} +32 -27
  81. data/ext/libcouchbase/src/operations/subdoc.cc +18 -38
  82. data/ext/libcouchbase/src/operations/{touch.cc → touch.c} +0 -0
  83. data/ext/libcouchbase/src/packetutils.c +37 -0
  84. data/ext/libcouchbase/src/packetutils.h +2 -2
  85. data/ext/libcouchbase/src/probes.d +1 -1
  86. data/ext/libcouchbase/src/{retrychk.cc → retrychk.c} +3 -2
  87. data/ext/libcouchbase/src/retryq.cc +4 -4
  88. data/ext/libcouchbase/src/settings.c +0 -3
  89. data/ext/libcouchbase/src/settings.h +0 -5
  90. data/ext/libcouchbase/src/simplestring.c +211 -0
  91. data/ext/libcouchbase/src/simplestring.h +228 -0
  92. data/ext/libcouchbase/src/ssl/ssl_c.c +0 -1
  93. data/ext/libcouchbase/src/ssl/ssl_common.c +0 -2
  94. data/ext/libcouchbase/src/ssl/ssl_e.c +1 -0
  95. data/ext/libcouchbase/src/ssobuf.h +82 -0
  96. data/ext/libcouchbase/src/trace.h +4 -4
  97. data/ext/libcouchbase/src/vbucket/vbucket.c +1 -0
  98. data/ext/libcouchbase/src/views/{docreq.cc → docreq.c} +54 -48
  99. data/ext/libcouchbase/src/views/docreq.h +30 -24
  100. data/ext/libcouchbase/src/views/viewreq.c +358 -0
  101. data/ext/libcouchbase/src/views/viewreq.h +13 -43
  102. data/ext/libcouchbase/tests/basic/t_connstr.cc +50 -89
  103. data/ext/libcouchbase/tests/basic/t_host.cc +75 -67
  104. data/ext/libcouchbase/tests/basic/t_jsparse.cc +78 -27
  105. data/ext/libcouchbase/tests/basic/t_string.cc +112 -0
  106. data/ext/libcouchbase/tests/htparse/t_basic.cc +78 -58
  107. data/ext/libcouchbase/tests/iotests/mock-environment.h +1 -2
  108. data/ext/libcouchbase/tests/iotests/t_confmon.cc +114 -96
  109. data/ext/libcouchbase/tests/mc/t_alloc.cc +9 -9
  110. data/ext/libcouchbase/tools/cbc-pillowfight.cc +1 -1
  111. data/lib/libcouchbase/ext/tasks.rb +6 -2
  112. data/lib/libcouchbase/query_view.rb +1 -1
  113. data/lib/libcouchbase/results_fiber.rb +6 -6
  114. data/lib/libcouchbase/version.rb +1 -1
  115. metadata +26 -26
  116. data/ext/libcouchbase/src/bootstrap.cc +0 -216
  117. data/ext/libcouchbase/src/bucketconfig/bc_file.cc +0 -281
  118. data/ext/libcouchbase/src/bucketconfig/bc_http.cc +0 -528
  119. data/ext/libcouchbase/src/bucketconfig/bc_mcraw.cc +0 -115
  120. data/ext/libcouchbase/src/bucketconfig/confmon.cc +0 -378
  121. data/ext/libcouchbase/src/dns-srv.cc +0 -142
  122. data/ext/libcouchbase/src/errmap.cc +0 -107
  123. data/ext/libcouchbase/src/errmap.h +0 -113
  124. data/ext/libcouchbase/src/lcbht/lcbht.cc +0 -177
  125. data/ext/libcouchbase/src/lcbio/connect.cc +0 -562
  126. data/ext/libcouchbase/src/lcbio/timer-cxx.h +0 -87
  127. data/ext/libcouchbase/src/mctx-helper.h +0 -51
  128. data/ext/libcouchbase/src/views/viewreq.cc +0 -318
  129. data/ext/libcouchbase/tests/iotests/t_errmap.cc +0 -97
@@ -19,12 +19,15 @@
19
19
  #define LCB_HTTP_H
20
20
 
21
21
  #include <libcouchbase/couchbase.h>
22
- #include "contrib/http_parser/http_parser.h"
23
- #include <list>
24
- #include <string>
25
-
22
+ #include "simplestring.h"
23
+ #include "sllist.h"
26
24
  struct lcb_settings_st;
27
25
 
26
+ #ifdef __cplusplus
27
+ extern "C" {
28
+ #endif
29
+
30
+
28
31
  /**
29
32
  * @file
30
33
  * HTTP Response parsing.
@@ -36,175 +39,161 @@ struct lcb_settings_st;
36
39
  * body.
37
40
  */
38
41
 
39
- namespace lcb {
40
- namespace htparse {
41
-
42
-
43
- struct MimeHeader {
44
- std::string key;
45
- std::string value;
46
- };
47
-
48
- struct Response {
49
- void clear() {
50
- status = 0;
51
- state = 0;
52
- headers.clear();
53
- body.clear();
54
- }
55
-
56
- /**
57
- * Get a header value for a key
58
- * @param response The response
59
- * @param key The key to look up
60
- * @return A string containing the value. If the header has no value then the
61
- * empty string will be returned. If the header does not exist NULL will be
62
- * returned.
63
- */
64
- const MimeHeader* get_header(const std::string& key) const;
65
-
66
- /**
67
- * Get a header value for a key
68
- * @param key The key to look up
69
- * @return A string containing the value. If the header has no value then the
70
- * empty string will be returned. If the header does not exist NULL will be
71
- * returned.
72
- */
73
- const char *get_header_value(const std::string& key) const {
74
- const MimeHeader *header = get_header(key);
75
- if (header) {
76
- return header->value.c_str();
77
- }
78
- return NULL;
79
- }
80
-
42
+ /** Response state */
43
+ typedef enum {
44
+ LCBHT_S_HTSTATUS = 1 << 0, /**< Have HTTP status */
45
+ LCBHT_S_HEADER = 1 << 1, /**< Have HTTP header */
46
+ LCBHT_S_BODY = 1 << 2, /**< Have HTTP body */
47
+ LCBHT_S_DONE = 1 << 3, /**< Have a full message */
48
+
49
+ /**Have a parse error. Note this is not the same as a HTTP error */
50
+ LCBHT_S_ERROR = 1 << 4
51
+ } lcbht_RESPSTATE;
52
+
53
+ typedef struct {
54
+ sllist_node slnode; /**< Next header in list */
55
+ const char *key;
56
+ const char *value;
57
+ lcb_string buf_; /**< Storage for the key and value */
58
+ } lcbht_MIMEHDR;
59
+
60
+ typedef struct {
81
61
  unsigned short status; /**< HTTP Status code */
82
- unsigned state;
83
- typedef std::list<MimeHeader> HeaderList;
84
- HeaderList headers;
85
- std::string body; /**< Body */
86
- };
87
-
88
- class Parser : private http_parser {
89
- public:
90
- /**
91
- * Initialize the parser object
92
- * @param settings the settings structure used for logging
93
- */
94
- Parser(lcb_settings_st*);
95
- ~Parser();
96
-
97
- /** Response state */
98
- enum State {
99
- S_NONE = 0,
100
- S_HTSTATUS = 1 << 0, /**< Have HTTP status */
101
- S_HEADER = 1 << 1, /**< Have HTTP header */
102
- S_BODY = 1 << 2, /**< Have HTTP body */
103
- S_DONE = 1 << 3, /**< Have a full message */
104
-
105
- /**Have a parse error. Note this is not the same as a HTTP error */
106
- S_ERROR = 1 << 4
107
- };
108
-
109
- /**
110
- * Parse incoming data into a message
111
- * @param data Pointer to new data
112
- * @param ndata Size of the data
113
- *
114
- * @return The current state of the parser. If `state & LCBHT_S_DONE` then
115
- * the current response should be handled before continuing.
116
- * If `state & LCBHT_S_ERROR` then there was an error parsing the contents
117
- * as it violated the HTTP protocol.
118
- */
119
- unsigned parse(const void *data, size_t ndata);
120
-
121
- /**
122
- * Parse incoming data without buffering
123
- * @param data The data to parse
124
- * @param ndata Length of the data
125
- * @param[out] nused How much of the data was actually consumed
126
- * @param[out] nbody Size of the body pointer
127
- * @param[out] pbody a pointer for the body
128
- *
129
- * @return See lcbht_set_bufmode for the meaning of this value
130
- *
131
- * @note It is not an error if `pbody` is NULL. It may mean that the parse state
132
- * is still within the headers and there is no body to parse yet.
133
- *
134
- * This function is intended to be used in a loop, until there is no input
135
- * remaining. The use of the `nused` pointer is to determine by how much the
136
- * `data` pointer should be incremented (and the `ndata` decremented) for the
137
- * next call. When this function returns with a non-error status, `pbody`
138
- * will contain a pointer to a buffer of data (but see note above) which can
139
- * then be processed by the application.
140
- *
141
- * @code{.c++}
142
- * char **body, *input;
143
- * unsigned inlen = get_input_len(), nused, bodylen;
144
- * unsigned res;
145
- * do {
146
- * res = parser->parse_ex(input, inlen, &nused, &nbody, &body);
147
- * if (res & Parser::S_ERROR) {
148
- * // handle error
149
- * break;
150
- * }
151
- * if (nbody) {
152
- * // handle body
153
- * }
154
- * input += nused;
155
- * inlen -= nused;
156
- * } while (!(res & Parser::S_DONE));
157
- * @endcode
158
- */
159
- unsigned parse_ex(const void *data, unsigned ndata,
160
- unsigned* nused, unsigned *nbody, const char **pbody);
161
-
162
- /**
163
- * Obtain the current response being processed.
164
- * @return a reference to a response object. The response object is only valid
165
- * until the next call into another parser API
166
- */
167
- Response& get_cur_response() {
168
- return resp;
169
- }
170
-
171
- /**
172
- * Determine whether HTTP/1.1 keepalive is enabled on the connection
173
- * @return true if keepalive is enabled, false otherwise.
174
- */
175
- bool can_keepalive() const;
176
-
177
- void reset();
178
-
179
- // Callbacks:
180
- inline int on_hdr_key(const char *, size_t);
181
- inline int on_hdr_value(const char *, size_t);
182
- inline int on_hdr_done();
183
- inline int on_body(const char *, size_t);
184
- inline int on_msg_done();
185
-
186
- static Parser* from_htp(http_parser *p) {
187
- return static_cast<Parser*>(p);
188
- }
189
-
190
- private:
191
- Response resp;
192
- lcb_settings_st *settings;
193
-
194
- enum last_call_type {
195
- CB_NONE, CB_HDR_KEY, CB_HDR_VALUE,
196
- CB_HDR_DONE, CB_BODY, CB_MSG_DONE
197
- };
198
- last_call_type lastcall;
199
-
200
- /* For parse_ex */
201
- const char *last_body;
202
- unsigned last_bodylen;
203
-
204
- bool paused;
205
- bool is_ex;
206
- };
207
-
208
- } // namespace htparse
209
- } // namespace lcb
62
+ lcbht_RESPSTATE state;
63
+ sllist_root headers; /**< List of response headers */
64
+ lcb_string body; /**< Body */
65
+ } lcbht_RESPONSE;
66
+
67
+ typedef struct lcbht_PARSER *lcbht_pPARSER;
68
+
69
+ /**
70
+ * Initialize the parser object
71
+ * @param settings the settings structure used for logging
72
+ * @return a new parser object
73
+ */
74
+ lcbht_pPARSER
75
+ lcbht_new(struct lcb_settings_st *settings);
76
+
77
+ /** Free the parser object */
78
+ void
79
+ lcbht_free(lcbht_pPARSER);
80
+
81
+ void
82
+ lcbht_reset(lcbht_pPARSER);
83
+
84
+ /**
85
+ * Parse incoming data into a message
86
+ * @param parser The parser
87
+ * @param data Pointer to new data
88
+ * @param ndata Size of the data
89
+ *
90
+ * @return The current state of the parser. If `state & LCBHT_S_DONE` then
91
+ * the current response should be handled before continuing.
92
+ * If `state & LCBHT_S_ERROR` then there was an error parsing the contents
93
+ * as it violated the HTTP protocol.
94
+ */
95
+ lcbht_RESPSTATE
96
+ lcbht_parse(lcbht_pPARSER parser, const void *data, unsigned ndata);
97
+
98
+ /**
99
+ * Parse incoming data without buffering
100
+ * @param parser The parser to use
101
+ * @param data The data to parse
102
+ * @param ndata Length of the data
103
+ * @param[out] nused How much of the data was actually consumed
104
+ * @param[out] nbody Size of the body pointer
105
+ * @param[out] pbody a pointer for the body
106
+ *
107
+ * @return See lcbht_set_bufmode for the meaning of this value
108
+ *
109
+ * @note It is not an error if `pbody` is NULL. It may mean that the parse state
110
+ * is still within the headers and there is no body to parse yet.
111
+ *
112
+ * This function is intended to be used in a loop, until there is no input
113
+ * remaining. The use of the `nused` pointer is to determine by how much the
114
+ * `data` pointer should be incremented (and the `ndata` decremented) for the
115
+ * next call. When this function returns with a non-error status, `pbody`
116
+ * will contain a pointer to a buffer of data (but see note above) which can
117
+ * then be processed by the application.
118
+ *
119
+ * @code{.c}
120
+ * char **body, *input;
121
+ * unsigned inlen = get_input_len(), nused, bodylen;
122
+ * lcbht_RESPSTATE res;
123
+ * do {
124
+ * res = lcbht_parse_ex(parser, input, inlen, &nused, &nbody, &body);
125
+ * if (res & LCBHT_S_ERROR) {
126
+ * // handle error
127
+ * break;
128
+ * }
129
+ * if (nbody) {
130
+ * // handle body
131
+ * }
132
+ * input += nused;
133
+ * inlen -= nused;
134
+ * } while (!(res & LCBHT_S_DONE));
135
+ * @endcode
136
+ */
137
+ lcbht_RESPSTATE
138
+ lcbht_parse_ex(lcbht_pPARSER parser, const void *data, unsigned ndata,
139
+ unsigned *nused, unsigned *nbody, const char **pbody);
140
+
141
+
142
+ /**
143
+ * Obtain the current response being processed.
144
+ * @param parser The parser
145
+ * @return a pointer to a response object. The response object is only valid
146
+ * until the next call into another parser API
147
+ */
148
+ lcbht_RESPONSE *
149
+ lcbht_get_response(lcbht_pPARSER parser);
150
+
151
+ /**
152
+ * Determine whether HTTP/1.1 keepalive is enabled on the connection
153
+ * @param parser The parser
154
+ * @return true if keepalive is enabled, false otherwise.
155
+ */
156
+ int
157
+ lcbht_can_keepalive(lcbht_pPARSER parser);
158
+
159
+ /**
160
+ * Clear the response object
161
+ * @param resp the response to clear
162
+ */
163
+ void
164
+ lcbht_clear_response(lcbht_RESPONSE *resp);
165
+
166
+ /**
167
+ * Get a header value for a key
168
+ * @param response The response
169
+ * @param key The key to look up
170
+ * @return A string containing the value. If the header has no value then the
171
+ * empty string will be returned. If the header does not exist NULL will be
172
+ * returned.
173
+ */
174
+ const char *
175
+ lcbht_get_resphdr(const lcbht_RESPONSE *response, const char *key);
176
+
177
+ /**
178
+ * Return a list of headers
179
+ * @param response The response
180
+ * @return A list of headers. Iterate over this value like so:
181
+ * @code{.c}
182
+ * char **hdrlist = lcbht_make_resphdrlist(response);
183
+ * for (char **cur = hdrlist; *cur; cur += 2) {
184
+ * char *key = cur[0];
185
+ * char *value = cur[1];
186
+ * // do something
187
+ * free(key);
188
+ * free(value);
189
+ * }
190
+ * free(hdrlist);
191
+ * @endcode
192
+ */
193
+ char **
194
+ lcbht_make_resphdrlist(lcbht_RESPONSE *response);
195
+
196
+ #ifdef __cplusplus
197
+ }
198
+ #endif
210
199
  #endif
@@ -0,0 +1,557 @@
1
+ /* -*- Mode: C; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
2
+ /*
3
+ * Copyright 2014 Couchbase, Inc.
4
+ *
5
+ * Licensed under the Apache License, Version 2.0 (the "License");
6
+ * you may not use this file except in compliance with the License.
7
+ * You may obtain a copy of the License at
8
+ *
9
+ * http://www.apache.org/licenses/LICENSE-2.0
10
+ *
11
+ * Unless required by applicable law or agreed to in writing, software
12
+ * distributed under the License is distributed on an "AS IS" BASIS,
13
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14
+ * See the License for the specific language governing permissions and
15
+ * limitations under the License.
16
+ */
17
+
18
+ #include "config.h"
19
+ #include "connect.h"
20
+ #include "ioutils.h"
21
+ #include "iotable.h"
22
+ #include "settings.h"
23
+ #include "timer-ng.h"
24
+ #include <errno.h>
25
+
26
+ /* win32 lacks EAI_SYSTEM */
27
+ #ifndef EAI_SYSTEM
28
+ #define EAI_SYSTEM 0
29
+ #endif
30
+ #define LOGARGS(conn, lvl) conn->settings, "connection", LCB_LOG_##lvl, __FILE__, __LINE__
31
+ static const lcb_host_t *get_loghost(lcbio_SOCKET *s) {
32
+ static lcb_host_t host = { "NOHOST", "NOPORT" };
33
+ if (!s) { return &host; }
34
+ if (!s->info) { return &host; }
35
+ return &s->info->ep;
36
+ }
37
+
38
+ /** Format string arguments for %p%s:%s */
39
+ #define CSLOGID(sock) get_loghost(sock)->host, get_loghost(sock)->port, (void*)s
40
+ #define CSLOGFMT "<%s:%s> (SOCK=%p) "
41
+
42
+ typedef enum {
43
+ CS_PENDING = 0,
44
+ CS_CANCELLED,
45
+ CS_TIMEDOUT,
46
+ CS_CONNECTED,
47
+ CS_ERROR
48
+ } connect_state;
49
+
50
+ typedef struct lcbio_CONNSTART {
51
+ lcbio_CONNDONE_cb handler;
52
+ lcbio_SOCKET *sock;
53
+ lcbio_OSERR syserr;
54
+ void *arg;
55
+ void *event;
56
+ short ev_active; /* whether the event pointer is active (Event only) */
57
+ short in_uhandler; /* Whether we're inside the user-defined handler */
58
+ struct addrinfo *ai_root;
59
+ struct addrinfo *ai;
60
+ connect_state state;
61
+ lcb_error_t pending;
62
+ lcbio_ASYNC *async;
63
+ char *hoststr;
64
+ } lcbio_CONNSTART;
65
+
66
+ static void
67
+ cs_unwatch(lcbio_CONNSTART *cs)
68
+ {
69
+ lcbio_SOCKET *s = cs->sock;
70
+ if (s && cs->ev_active) {
71
+ lcb_assert(s->u.fd != INVALID_SOCKET);
72
+ IOT_V0EV(s->io).cancel(IOT_ARG(s->io), s->u.fd, cs->event);
73
+ cs->ev_active = 0;
74
+ }
75
+ }
76
+
77
+ /**
78
+ * Handler invoked to deliver final status for a connection. This will invoke
79
+ * the user supplied callback with the relevant status (if it has not been
80
+ * cancelled) and then free the CONNSTART object.
81
+ */
82
+ static void
83
+ cs_handler(void *cookie)
84
+ {
85
+ lcbio_CONNSTART *cs = cookie;
86
+ lcb_error_t err;
87
+ lcbio_SOCKET *s = cs->sock;
88
+
89
+ if (s && cs->event) {
90
+ cs_unwatch(cs);
91
+ IOT_V0EV(s->io).destroy(IOT_ARG(s->io), cs->event);
92
+ }
93
+
94
+ if (cs->state == CS_PENDING) {
95
+ /* state was not changed since initial scheduling */
96
+ err = LCB_ETIMEDOUT;
97
+ } else if (cs->state == CS_CONNECTED) {
98
+ /* clear pending error */
99
+ err = LCB_SUCCESS;
100
+ } else {
101
+ if (s != NULL && cs->pending == LCB_CONNECT_ERROR) {
102
+ err = lcbio_mklcberr(cs->syserr, s->settings);
103
+ } else {
104
+ err = cs->pending;
105
+ }
106
+ }
107
+
108
+ if (cs->state == CS_CANCELLED) {
109
+ /* ignore everything. Clean up resources */
110
+ goto GT_DTOR;
111
+ }
112
+
113
+ if (s) {
114
+ lcbio__load_socknames(s);
115
+ if (err == LCB_SUCCESS) {
116
+ lcb_log(LOGARGS(s, INFO), CSLOGFMT "Connected ", CSLOGID(s));
117
+
118
+ if (s->settings->tcp_nodelay) {
119
+ lcb_error_t ndrc = lcbio_disable_nagle(s);
120
+ if (ndrc != LCB_SUCCESS) {
121
+ lcb_log(LOGARGS(s, INFO), CSLOGFMT "Couldn't set TCP_NODELAY", CSLOGID(s));
122
+ } else {
123
+ lcb_log(LOGARGS(s, DEBUG), CSLOGFMT "Successfuly set TCP_NODELAY", CSLOGID(s));
124
+ }
125
+ }
126
+ } else {
127
+ lcb_log(LOGARGS(s, ERR), CSLOGFMT "Failed: lcb_err=0x%x, os_errno=%u", CSLOGID(s), err, cs->syserr);
128
+ }
129
+ }
130
+
131
+ /** Handler section */
132
+ cs->in_uhandler = 1;
133
+ cs->handler(err == LCB_SUCCESS ? s : NULL, cs->arg, err, cs->syserr);
134
+
135
+ GT_DTOR:
136
+ if (cs->async) {
137
+ lcbio_timer_destroy(cs->async);
138
+ }
139
+ if (cs->sock) {
140
+ lcbio_unref(cs->sock);
141
+ }
142
+ if (cs->ai_root) {
143
+ freeaddrinfo(cs->ai_root);
144
+ }
145
+ free(cs);
146
+ }
147
+
148
+ static void
149
+ cs_state_signal(lcbio_CONNSTART *cs, connect_state state, lcb_error_t err)
150
+ {
151
+ if (cs->state != CS_PENDING) {
152
+ /** State already set */
153
+ return;
154
+ }
155
+
156
+
157
+ if (state == CS_CONNECTED) {
158
+ /* clear last errors if we're successful */
159
+ cs->pending = LCB_SUCCESS;
160
+ } else if (cs->pending == LCB_SUCCESS) {
161
+ /* set error code only if previous code was not a failure */
162
+ cs->pending = err;
163
+ }
164
+
165
+ cs->state = state;
166
+ lcbio_async_signal(cs->async);
167
+ }
168
+
169
+ /** Cancels and mutes any pending event */
170
+ void
171
+ lcbio_connect_cancel(lcbio_pCONNSTART cs)
172
+ {
173
+ if (cs->in_uhandler) {
174
+ /* already inside user-defined handler */
175
+ return;
176
+ }
177
+ cs->state = CS_CANCELLED;
178
+ cs_handler(cs);
179
+ }
180
+
181
+
182
+ static int
183
+ ensure_sock(lcbio_CONNSTART *cs)
184
+ {
185
+ lcbio_SOCKET *s = cs->sock;
186
+ lcbio_TABLE *io = s->io;
187
+ int errtmp = 0;
188
+
189
+ if (cs->ai == NULL) {
190
+ return -1;
191
+ }
192
+
193
+ if (IOT_IS_EVENT(io)) {
194
+ if (s->u.fd != INVALID_SOCKET) {
195
+ /* already have one? */
196
+ return 0;
197
+ }
198
+
199
+ while (s->u.fd == INVALID_SOCKET && cs->ai != NULL) {
200
+ s->u.fd = lcbio_E_ai2sock(io, &cs->ai, &errtmp);
201
+ if (s->u.fd != INVALID_SOCKET) {
202
+ return 0;
203
+ }
204
+ }
205
+ } else {
206
+ if (s->u.sd) {
207
+ return 0;
208
+ }
209
+
210
+ while (s->u.sd == NULL && cs->ai != NULL) {
211
+ s->u.sd = lcbio_C_ai2sock(io, &cs->ai, &errtmp);
212
+ if (s->u.sd) {
213
+ s->u.sd->lcbconn = (void *) cs->sock;
214
+ s->u.sd->parent = IOT_ARG(io);
215
+ return 0;
216
+ }
217
+ }
218
+ }
219
+
220
+ if (cs->ai == NULL) {
221
+ lcbio_mksyserr(IOT_ERRNO(io), &cs->syserr);
222
+ return -1;
223
+ }
224
+ return 0;
225
+ }
226
+
227
+ static void
228
+ destroy_cursock(lcbio_CONNSTART *cs)
229
+ {
230
+ lcbio_SOCKET *s = cs->sock;
231
+ lcbio_TABLE *iot = s->io;
232
+ if (cs->ai) {
233
+ cs->ai = cs->ai->ai_next;
234
+ }
235
+
236
+ if (!cs->ai) {
237
+ return;
238
+ }
239
+
240
+ if (IOT_IS_EVENT(iot)) {
241
+ if (cs->ev_active) {
242
+ lcb_assert(s->u.fd != INVALID_SOCKET);
243
+ IOT_V0EV(iot).cancel(IOT_ARG(iot), s->u.fd, cs->event);
244
+ cs->ev_active = 0;
245
+ }
246
+ IOT_V0IO(iot).close(IOT_ARG(iot), s->u.fd);
247
+ s->u.fd = INVALID_SOCKET;
248
+ } else {
249
+ if (s->u.sd) {
250
+ IOT_V1(iot).close(IOT_ARG(iot), s->u.sd);
251
+ s->u.sd = NULL;
252
+ }
253
+ }
254
+ }
255
+
256
+ static void
257
+ E_connect(lcb_socket_t sock, short events, void *arg)
258
+ {
259
+ lcbio_CONNSTART *cs = arg;
260
+ lcbio_SOCKET *s = cs->sock;
261
+ lcbio_TABLE *io = s->io;
262
+ int retry_once = 0;
263
+ lcbio_CSERR connstatus;
264
+
265
+ (void)sock;
266
+
267
+ lcb_log(LOGARGS(s, TRACE), CSLOGFMT "Got event handler for new connection", CSLOGID(s));
268
+
269
+ GT_NEXTSOCK:
270
+ if (ensure_sock(cs) == -1) {
271
+ cs_state_signal(cs, CS_ERROR, LCB_CONNECT_ERROR);
272
+ return;
273
+ }
274
+
275
+ if (events & LCB_ERROR_EVENT) {
276
+ socklen_t errlen = sizeof(int);
277
+ int sockerr = 0;
278
+ lcb_log(LOGARGS(s, TRACE), CSLOGFMT "Received ERROR_EVENT", CSLOGID(s));
279
+ getsockopt(s->u.fd, SOL_SOCKET, SO_ERROR, (char *)&sockerr, &errlen);
280
+ lcbio_mksyserr(sockerr, &cs->syserr);
281
+ destroy_cursock(cs);
282
+ goto GT_NEXTSOCK;
283
+
284
+ } else {
285
+ int rv = 0;
286
+ struct addrinfo *ai = cs->ai;
287
+
288
+ GT_CONNECT:
289
+ rv = IOT_V0IO(io).connect0(
290
+ IOT_ARG(io), s->u.fd, ai->ai_addr, (unsigned)ai->ai_addrlen);
291
+
292
+ if (rv == 0) {
293
+ cs_unwatch(cs);
294
+ cs_state_signal(cs, CS_CONNECTED, LCB_SUCCESS);
295
+ return;
296
+ }
297
+ }
298
+
299
+ connstatus = lcbio_mkcserr(IOT_ERRNO(io));
300
+ lcbio_mksyserr(IOT_ERRNO(io), &cs->syserr);
301
+
302
+
303
+ switch (connstatus) {
304
+
305
+ case LCBIO_CSERR_INTR:
306
+ goto GT_CONNECT;
307
+
308
+ case LCBIO_CSERR_CONNECTED:
309
+ cs_unwatch(cs);
310
+ cs_state_signal(cs, CS_CONNECTED, LCB_SUCCESS);
311
+ return;
312
+
313
+ case LCBIO_CSERR_BUSY:
314
+ lcb_log(LOGARGS(s, TRACE), CSLOGFMT "Scheduling asynchronous watch for socket.", CSLOGID(s));
315
+ IOT_V0EV(io).watch(
316
+ IOT_ARG(io), s->u.fd, cs->event, LCB_WRITE_EVENT, cs, E_connect);
317
+ cs->ev_active = 1;
318
+ return;
319
+
320
+ case LCBIO_CSERR_EINVAL:
321
+ if (!retry_once) {
322
+ retry_once = 1;
323
+ goto GT_CONNECT;
324
+ }
325
+ /* fallthrough */
326
+
327
+ case LCBIO_CSERR_EFAIL:
328
+ default:
329
+ /* close the current socket and try again */
330
+ lcb_log(LOGARGS(s, TRACE), CSLOGFMT "connect() failed. os_error=%d [%s]", CSLOGID(s), IOT_ERRNO(io), strerror(IOT_ERRNO(io)));
331
+ destroy_cursock(cs);
332
+ goto GT_NEXTSOCK;
333
+ }
334
+ }
335
+
336
+ static void C_connect(lcbio_CONNSTART *cs);
337
+
338
+ static void
339
+ C_conncb(lcb_sockdata_t *sock, int status)
340
+ {
341
+ lcbio_SOCKET *s = (void *)sock->lcbconn;
342
+ lcbio_CONNSTART *cs = (void *)s->ctx;
343
+
344
+ lcb_log(LOGARGS(s, TRACE), CSLOGFMT "Received completion handler. Status=%d. errno=%d", CSLOGID(s), status, IOT_ERRNO(s->io));
345
+
346
+ if (!--s->refcount) {
347
+ lcbio__destroy(s);
348
+ return;
349
+ }
350
+
351
+ if (!status) {
352
+ if (cs->state == CS_PENDING) {
353
+ cs->state = CS_CONNECTED;
354
+ }
355
+ cs_handler(cs);
356
+ } else {
357
+ lcbio_mksyserr(IOT_ERRNO(s->io), &cs->syserr);
358
+ destroy_cursock(cs);
359
+ C_connect(cs);
360
+ }
361
+ }
362
+
363
+ static void
364
+ C_connect(lcbio_CONNSTART *cs)
365
+ {
366
+ int rv;
367
+ lcbio_SOCKET *s = cs->sock;
368
+ int retry_once = 0;
369
+ lcbio_CSERR status;
370
+ lcbio_TABLE *io = s->io;
371
+
372
+ GT_NEXTSOCK:
373
+ if (ensure_sock(cs) != 0) {
374
+ lcbio_mksyserr(IOT_ERRNO(io), &cs->syserr);
375
+ cs_state_signal(cs, CS_ERROR, LCB_CONNECT_ERROR);
376
+ return;
377
+ }
378
+
379
+ GT_CONNECT:
380
+ rv = IOT_V1(io).connect(IOT_ARG(io), s->u.sd, cs->ai->ai_addr,
381
+ (unsigned)cs->ai->ai_addrlen, C_conncb);
382
+ if (rv == 0) {
383
+ lcbio_ref(s);
384
+ return;
385
+ }
386
+
387
+ lcbio_mksyserr(IOT_ERRNO(io), &cs->syserr);
388
+ status = lcbio_mkcserr(IOT_ERRNO(io));
389
+ switch (status) {
390
+
391
+ case LCBIO_CSERR_INTR:
392
+ goto GT_CONNECT;
393
+
394
+ case LCBIO_CSERR_CONNECTED:
395
+ cs_state_signal(cs, CS_CONNECTED, LCB_SUCCESS);
396
+ return;
397
+
398
+ case LCBIO_CSERR_BUSY:
399
+ return;
400
+
401
+ case LCBIO_CSERR_EINVAL:
402
+ if (!retry_once) {
403
+ retry_once = 1;
404
+ goto GT_CONNECT;
405
+ }
406
+ /* fallthrough */
407
+
408
+ case LCBIO_CSERR_EFAIL:
409
+ default:
410
+ destroy_cursock(cs);
411
+ goto GT_NEXTSOCK;
412
+ }
413
+ }
414
+
415
+ struct lcbio_CONNSTART *
416
+ lcbio_connect(lcbio_TABLE *iot, lcb_settings *settings, const lcb_host_t *dest,
417
+ uint32_t timeout, lcbio_CONNDONE_cb handler, void *arg)
418
+ {
419
+ lcbio_SOCKET *s;
420
+ lcbio_CONNSTART *ret;
421
+ struct addrinfo hints;
422
+ int rv;
423
+
424
+ s = calloc(1, sizeof(*s));
425
+ ret = calloc(1, sizeof(*ret));
426
+
427
+ /** Initialize the socket first */
428
+ s->io = iot;
429
+ s->settings = settings;
430
+ s->ctx = ret;
431
+ s->refcount = 1;
432
+ s->info = calloc(1, sizeof(*s->info));
433
+ s->info->ep = *dest;
434
+ lcbio_table_ref(s->io);
435
+ lcb_settings_ref(s->settings);
436
+ lcb_list_init(&s->protos);
437
+
438
+ if (IOT_IS_EVENT(iot)) {
439
+ s->u.fd = INVALID_SOCKET;
440
+ ret->event = IOT_V0EV(iot).create(IOT_ARG(iot));
441
+ }
442
+
443
+ /** Initialize the connstart structure */
444
+ ret->handler = handler;
445
+ ret->arg = arg;
446
+ ret->sock = s;
447
+ ret->async = lcbio_timer_new(iot, ret, cs_handler);
448
+
449
+ lcbio_timer_rearm(ret->async, timeout);
450
+ lcb_log(LOGARGS(s, INFO), CSLOGFMT "Starting. Timeout=%uus", CSLOGID(s), timeout);
451
+
452
+ /** Hostname lookup: */
453
+ memset(&hints, 0, sizeof(hints));
454
+ hints.ai_flags = AI_PASSIVE;
455
+ hints.ai_socktype = SOCK_STREAM;
456
+ if (settings->ipv6 == LCB_IPV6_DISABLED) {
457
+ hints.ai_family = AF_INET;
458
+ } else if (settings->ipv6 == LCB_IPV6_ONLY) {
459
+ hints.ai_family = AF_INET6;
460
+ } else {
461
+ hints.ai_family = AF_UNSPEC;
462
+ }
463
+
464
+ if ((rv = getaddrinfo(dest->host, dest->port, &hints, &ret->ai_root))) {
465
+ const char *errstr = rv != EAI_SYSTEM ? gai_strerror(rv) : "";
466
+ lcb_log(LOGARGS(s, ERR), CSLOGFMT "Couldn't look up %s (%s) [EAI=%d]", CSLOGID(s), dest->host, errstr, rv);
467
+ cs_state_signal(ret, CS_ERROR, LCB_UNKNOWN_HOST);
468
+ } else {
469
+ ret->ai = ret->ai_root;
470
+
471
+ /** Figure out how to connect */
472
+ if (IOT_IS_EVENT(iot)) {
473
+ E_connect(-1, LCB_WRITE_EVENT, ret);
474
+ } else {
475
+ C_connect(ret);
476
+ }
477
+ }
478
+ return ret;
479
+ }
480
+
481
+ lcbio_CONNSTART *
482
+ lcbio_connect_hl(lcbio_TABLE *iot, lcb_settings *settings,
483
+ hostlist_t hl, int rollover, uint32_t timeout,
484
+ lcbio_CONNDONE_cb handler, void *arg)
485
+ {
486
+ const lcb_host_t *cur;
487
+ unsigned ii, hlmax;
488
+ ii = 0;
489
+ hlmax = hostlist_size(hl);
490
+
491
+ while ( (cur = hostlist_shift_next(hl, rollover)) && ii++ < hlmax) {
492
+ lcbio_CONNSTART *ret = lcbio_connect(
493
+ iot, settings, cur, timeout, handler, arg);
494
+ if (ret) {
495
+ return ret;
496
+ }
497
+ }
498
+
499
+ return NULL;
500
+ }
501
+
502
+ lcbio_SOCKET *
503
+ lcbio_wrap_fd(lcbio_pTABLE iot, lcb_settings *settings, lcb_socket_t fd)
504
+ {
505
+ lcbio_SOCKET *ret = calloc(1, sizeof(*ret));
506
+ lcbio_CONNDONE_cb *ci = calloc(1, sizeof(*ci));
507
+
508
+ if (ret == NULL || ci == NULL) {
509
+ free(ret);
510
+ free(ci);
511
+ return NULL;
512
+ }
513
+
514
+ assert(iot->model = LCB_IOMODEL_EVENT);
515
+
516
+ lcb_list_init(&ret->protos);
517
+ ret->settings = settings;
518
+ ret->io = iot;
519
+ ret->refcount = 1;
520
+ ret->u.fd = fd;
521
+
522
+ lcbio_table_ref(ret->io);
523
+ lcb_settings_ref(ret->settings);
524
+ lcbio__load_socknames(ret);
525
+ return ret;
526
+ }
527
+
528
+ void
529
+ lcbio_shutdown(lcbio_SOCKET *s)
530
+ {
531
+ lcbio_TABLE *io = s->io;
532
+
533
+ lcbio__protoctx_delall(s);
534
+ if (IOT_IS_EVENT(io)) {
535
+ if (s->u.fd != INVALID_SOCKET) {
536
+ IOT_V0IO(io).close(IOT_ARG(io), s->u.fd);
537
+ s->u.fd = INVALID_SOCKET;
538
+ }
539
+ } else {
540
+ if (s->u.sd) {
541
+ IOT_V1(io).close(IOT_ARG(io), s->u.sd);
542
+ s->u.sd = NULL;
543
+ }
544
+ }
545
+ }
546
+
547
+ void
548
+ lcbio__destroy(lcbio_SOCKET *s)
549
+ {
550
+ lcbio_shutdown(s);
551
+ if (s->info) {
552
+ free(s->info);
553
+ }
554
+ lcbio_table_unref(s->io);
555
+ lcb_settings_unref(s->settings);
556
+ free(s);
557
+ }