transports.c 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520
  1. /*
  2. +----------------------------------------------------------------------+
  3. | Copyright (c) The PHP Group |
  4. +----------------------------------------------------------------------+
  5. | This source file is subject to version 3.01 of the PHP license, |
  6. | that is bundled with this package in the file LICENSE, and is |
  7. | available through the world-wide-web at the following url: |
  8. | https://www.php.net/license/3_01.txt |
  9. | If you did not receive a copy of the PHP license and are unable to |
  10. | obtain it through the world-wide-web, please send a note to |
  11. | license@php.net so we can mail you a copy immediately. |
  12. +----------------------------------------------------------------------+
  13. | Author: Wez Furlong <wez@thebrainroom.com> |
  14. +----------------------------------------------------------------------+
  15. */
  16. #include "php.h"
  17. #include "php_streams_int.h"
  18. #include "ext/standard/file.h"
  19. static HashTable xport_hash;
  20. PHPAPI HashTable *php_stream_xport_get_hash(void)
  21. {
  22. return &xport_hash;
  23. }
  24. PHPAPI int php_stream_xport_register(const char *protocol, php_stream_transport_factory factory)
  25. {
  26. zend_string *str = zend_string_init_interned(protocol, strlen(protocol), 1);
  27. zend_hash_update_ptr(&xport_hash, str, factory);
  28. zend_string_release_ex(str, 1);
  29. return SUCCESS;
  30. }
  31. PHPAPI int php_stream_xport_unregister(const char *protocol)
  32. {
  33. return zend_hash_str_del(&xport_hash, protocol, strlen(protocol));
  34. }
  35. #define ERR_REPORT(out_err, fmt, arg) \
  36. if (out_err) { *out_err = strpprintf(0, fmt, arg); } \
  37. else { php_error_docref(NULL, E_WARNING, fmt, arg); }
  38. #define ERR_RETURN(out_err, local_err, fmt) \
  39. if (out_err) { *out_err = local_err; } \
  40. else { php_error_docref(NULL, E_WARNING, fmt, local_err ? ZSTR_VAL(local_err) : "Unspecified error"); \
  41. if (local_err) { zend_string_release_ex(local_err, 0); local_err = NULL; } \
  42. }
  43. PHPAPI php_stream *_php_stream_xport_create(const char *name, size_t namelen, int options,
  44. int flags, const char *persistent_id,
  45. struct timeval *timeout,
  46. php_stream_context *context,
  47. zend_string **error_string,
  48. int *error_code
  49. STREAMS_DC)
  50. {
  51. php_stream *stream = NULL;
  52. php_stream_transport_factory factory = NULL;
  53. const char *p, *protocol = NULL;
  54. size_t n = 0;
  55. bool failed = false;
  56. bool bailout = false;
  57. zend_string *error_text = NULL;
  58. struct timeval default_timeout = { 0, 0 };
  59. default_timeout.tv_sec = FG(default_socket_timeout);
  60. if (timeout == NULL) {
  61. timeout = &default_timeout;
  62. }
  63. /* check for a cached persistent socket */
  64. if (persistent_id) {
  65. switch(php_stream_from_persistent_id(persistent_id, &stream)) {
  66. case PHP_STREAM_PERSISTENT_SUCCESS:
  67. /* use a 0 second timeout when checking if the socket
  68. * has already died */
  69. if (PHP_STREAM_OPTION_RETURN_OK == php_stream_set_option(stream, PHP_STREAM_OPTION_CHECK_LIVENESS, 0, NULL)) {
  70. return stream;
  71. }
  72. /* dead - kill it */
  73. php_stream_pclose(stream);
  74. stream = NULL;
  75. /* fall through */
  76. case PHP_STREAM_PERSISTENT_FAILURE:
  77. default:
  78. /* failed; get a new one */
  79. ;
  80. }
  81. }
  82. for (p = name; isalnum((int)*p) || *p == '+' || *p == '-' || *p == '.'; p++) {
  83. n++;
  84. }
  85. if ((*p == ':') && (n > 1) && !strncmp("://", p, 3)) {
  86. protocol = name;
  87. name = p + 3;
  88. namelen -= n + 3;
  89. } else {
  90. protocol = "tcp";
  91. n = 3;
  92. }
  93. if (protocol) {
  94. if (NULL == (factory = zend_hash_str_find_ptr(&xport_hash, protocol, n))) {
  95. char wrapper_name[32];
  96. if (n >= sizeof(wrapper_name))
  97. n = sizeof(wrapper_name) - 1;
  98. PHP_STRLCPY(wrapper_name, protocol, sizeof(wrapper_name), n);
  99. ERR_REPORT(error_string, "Unable to find the socket transport \"%s\" - did you forget to enable it when you configured PHP?",
  100. wrapper_name);
  101. return NULL;
  102. }
  103. }
  104. if (factory == NULL) {
  105. /* should never happen */
  106. php_error_docref(NULL, E_WARNING, "Could not find a factory !?");
  107. return NULL;
  108. }
  109. stream = (factory)(protocol, n,
  110. (char*)name, namelen, persistent_id, options, flags, timeout,
  111. context STREAMS_REL_CC);
  112. if (stream) {
  113. zend_try {
  114. php_stream_context_set(stream, context);
  115. if ((flags & STREAM_XPORT_SERVER) == 0) {
  116. /* client */
  117. if (flags & (STREAM_XPORT_CONNECT|STREAM_XPORT_CONNECT_ASYNC)) {
  118. if (-1 == php_stream_xport_connect(stream, name, namelen,
  119. flags & STREAM_XPORT_CONNECT_ASYNC ? 1 : 0,
  120. timeout, &error_text, error_code)) {
  121. ERR_RETURN(error_string, error_text, "connect() failed: %s");
  122. failed = true;
  123. }
  124. }
  125. } else {
  126. /* server */
  127. if (flags & STREAM_XPORT_BIND) {
  128. if (0 != php_stream_xport_bind(stream, name, namelen, &error_text)) {
  129. ERR_RETURN(error_string, error_text, "bind() failed: %s");
  130. failed = true;
  131. } else if (flags & STREAM_XPORT_LISTEN) {
  132. zval *zbacklog = NULL;
  133. int backlog = 32;
  134. if (PHP_STREAM_CONTEXT(stream) && (zbacklog = php_stream_context_get_option(PHP_STREAM_CONTEXT(stream), "socket", "backlog")) != NULL) {
  135. backlog = zval_get_long(zbacklog);
  136. }
  137. if (0 != php_stream_xport_listen(stream, backlog, &error_text)) {
  138. ERR_RETURN(error_string, error_text, "listen() failed: %s");
  139. failed = true;
  140. }
  141. }
  142. }
  143. }
  144. } zend_catch {
  145. bailout = true;
  146. } zend_end_try();
  147. }
  148. if (failed || bailout) {
  149. /* failure means that they don't get a stream to play with */
  150. if (persistent_id) {
  151. php_stream_pclose(stream);
  152. } else {
  153. php_stream_close(stream);
  154. }
  155. stream = NULL;
  156. if (bailout) {
  157. zend_bailout();
  158. }
  159. }
  160. return stream;
  161. }
  162. /* Bind the stream to a local address */
  163. PHPAPI int php_stream_xport_bind(php_stream *stream,
  164. const char *name, size_t namelen,
  165. zend_string **error_text
  166. )
  167. {
  168. php_stream_xport_param param;
  169. int ret;
  170. memset(&param, 0, sizeof(param));
  171. param.op = STREAM_XPORT_OP_BIND;
  172. param.inputs.name = (char*)name;
  173. param.inputs.namelen = namelen;
  174. param.want_errortext = error_text ? 1 : 0;
  175. ret = php_stream_set_option(stream, PHP_STREAM_OPTION_XPORT_API, 0, &param);
  176. if (ret == PHP_STREAM_OPTION_RETURN_OK) {
  177. if (error_text) {
  178. *error_text = param.outputs.error_text;
  179. }
  180. return param.outputs.returncode;
  181. }
  182. return ret;
  183. }
  184. /* Connect to a remote address */
  185. PHPAPI int php_stream_xport_connect(php_stream *stream,
  186. const char *name, size_t namelen,
  187. int asynchronous,
  188. struct timeval *timeout,
  189. zend_string **error_text,
  190. int *error_code
  191. )
  192. {
  193. php_stream_xport_param param;
  194. int ret;
  195. memset(&param, 0, sizeof(param));
  196. param.op = asynchronous ? STREAM_XPORT_OP_CONNECT_ASYNC: STREAM_XPORT_OP_CONNECT;
  197. param.inputs.name = (char*)name;
  198. param.inputs.namelen = namelen;
  199. param.inputs.timeout = timeout;
  200. param.want_errortext = error_text ? 1 : 0;
  201. ret = php_stream_set_option(stream, PHP_STREAM_OPTION_XPORT_API, 0, &param);
  202. if (ret == PHP_STREAM_OPTION_RETURN_OK) {
  203. if (error_text) {
  204. *error_text = param.outputs.error_text;
  205. }
  206. if (error_code) {
  207. *error_code = param.outputs.error_code;
  208. }
  209. return param.outputs.returncode;
  210. }
  211. return ret;
  212. }
  213. /* Prepare to listen */
  214. PHPAPI int php_stream_xport_listen(php_stream *stream, int backlog, zend_string **error_text)
  215. {
  216. php_stream_xport_param param;
  217. int ret;
  218. memset(&param, 0, sizeof(param));
  219. param.op = STREAM_XPORT_OP_LISTEN;
  220. param.inputs.backlog = backlog;
  221. param.want_errortext = error_text ? 1 : 0;
  222. ret = php_stream_set_option(stream, PHP_STREAM_OPTION_XPORT_API, 0, &param);
  223. if (ret == PHP_STREAM_OPTION_RETURN_OK) {
  224. if (error_text) {
  225. *error_text = param.outputs.error_text;
  226. }
  227. return param.outputs.returncode;
  228. }
  229. return ret;
  230. }
  231. /* Get the next client and their address (as a string) */
  232. PHPAPI int php_stream_xport_accept(php_stream *stream, php_stream **client,
  233. zend_string **textaddr,
  234. void **addr, socklen_t *addrlen,
  235. struct timeval *timeout,
  236. zend_string **error_text
  237. )
  238. {
  239. php_stream_xport_param param;
  240. int ret;
  241. memset(&param, 0, sizeof(param));
  242. param.op = STREAM_XPORT_OP_ACCEPT;
  243. param.inputs.timeout = timeout;
  244. param.want_addr = addr ? 1 : 0;
  245. param.want_textaddr = textaddr ? 1 : 0;
  246. param.want_errortext = error_text ? 1 : 0;
  247. ret = php_stream_set_option(stream, PHP_STREAM_OPTION_XPORT_API, 0, &param);
  248. if (ret == PHP_STREAM_OPTION_RETURN_OK) {
  249. *client = param.outputs.client;
  250. if (addr) {
  251. *addr = param.outputs.addr;
  252. *addrlen = param.outputs.addrlen;
  253. }
  254. if (textaddr) {
  255. *textaddr = param.outputs.textaddr;
  256. }
  257. if (error_text) {
  258. *error_text = param.outputs.error_text;
  259. }
  260. return param.outputs.returncode;
  261. }
  262. return ret;
  263. }
  264. PHPAPI int php_stream_xport_get_name(php_stream *stream, int want_peer,
  265. zend_string **textaddr,
  266. void **addr, socklen_t *addrlen
  267. )
  268. {
  269. php_stream_xport_param param;
  270. int ret;
  271. memset(&param, 0, sizeof(param));
  272. param.op = want_peer ? STREAM_XPORT_OP_GET_PEER_NAME : STREAM_XPORT_OP_GET_NAME;
  273. param.want_addr = addr ? 1 : 0;
  274. param.want_textaddr = textaddr ? 1 : 0;
  275. ret = php_stream_set_option(stream, PHP_STREAM_OPTION_XPORT_API, 0, &param);
  276. if (ret == PHP_STREAM_OPTION_RETURN_OK) {
  277. if (addr) {
  278. *addr = param.outputs.addr;
  279. *addrlen = param.outputs.addrlen;
  280. }
  281. if (textaddr) {
  282. *textaddr = param.outputs.textaddr;
  283. }
  284. return param.outputs.returncode;
  285. }
  286. return ret;
  287. }
  288. PHPAPI int php_stream_xport_crypto_setup(php_stream *stream, php_stream_xport_crypt_method_t crypto_method, php_stream *session_stream)
  289. {
  290. php_stream_xport_crypto_param param;
  291. int ret;
  292. memset(&param, 0, sizeof(param));
  293. param.op = STREAM_XPORT_CRYPTO_OP_SETUP;
  294. param.inputs.method = crypto_method;
  295. param.inputs.session = session_stream;
  296. ret = php_stream_set_option(stream, PHP_STREAM_OPTION_CRYPTO_API, 0, &param);
  297. if (ret == PHP_STREAM_OPTION_RETURN_OK) {
  298. return param.outputs.returncode;
  299. }
  300. php_error_docref("streams.crypto", E_WARNING, "This stream does not support SSL/crypto");
  301. return ret;
  302. }
  303. PHPAPI int php_stream_xport_crypto_enable(php_stream *stream, int activate)
  304. {
  305. php_stream_xport_crypto_param param;
  306. int ret;
  307. memset(&param, 0, sizeof(param));
  308. param.op = STREAM_XPORT_CRYPTO_OP_ENABLE;
  309. param.inputs.activate = activate;
  310. ret = php_stream_set_option(stream, PHP_STREAM_OPTION_CRYPTO_API, 0, &param);
  311. if (ret == PHP_STREAM_OPTION_RETURN_OK) {
  312. return param.outputs.returncode;
  313. }
  314. php_error_docref("streams.crypto", E_WARNING, "This stream does not support SSL/crypto");
  315. return ret;
  316. }
  317. /* Similar to recv() system call; read data from the stream, optionally
  318. * peeking, optionally retrieving OOB data */
  319. PHPAPI int php_stream_xport_recvfrom(php_stream *stream, char *buf, size_t buflen,
  320. int flags, void **addr, socklen_t *addrlen, zend_string **textaddr
  321. )
  322. {
  323. php_stream_xport_param param;
  324. int ret = 0;
  325. int recvd_len = 0;
  326. #if 0
  327. int oob;
  328. if (flags == 0 && addr == NULL) {
  329. return php_stream_read(stream, buf, buflen);
  330. }
  331. if (stream->readfilters.head) {
  332. php_error_docref(NULL, E_WARNING, "Cannot peek or fetch OOB data from a filtered stream");
  333. return -1;
  334. }
  335. oob = (flags & STREAM_OOB) == STREAM_OOB;
  336. if (!oob && addr == NULL) {
  337. /* must be peeking at regular data; copy content from the buffer
  338. * first, then adjust the pointer/len before handing off to the
  339. * stream */
  340. recvd_len = stream->writepos - stream->readpos;
  341. if (recvd_len > buflen) {
  342. recvd_len = buflen;
  343. }
  344. if (recvd_len) {
  345. memcpy(buf, stream->readbuf, recvd_len);
  346. buf += recvd_len;
  347. buflen -= recvd_len;
  348. }
  349. /* if we filled their buffer, return */
  350. if (buflen == 0) {
  351. return recvd_len;
  352. }
  353. }
  354. #endif
  355. /* otherwise, we are going to bypass the buffer */
  356. memset(&param, 0, sizeof(param));
  357. param.op = STREAM_XPORT_OP_RECV;
  358. param.want_addr = addr ? 1 : 0;
  359. param.want_textaddr = textaddr ? 1 : 0;
  360. param.inputs.buf = buf;
  361. param.inputs.buflen = buflen;
  362. param.inputs.flags = flags;
  363. ret = php_stream_set_option(stream, PHP_STREAM_OPTION_XPORT_API, 0, &param);
  364. if (ret == PHP_STREAM_OPTION_RETURN_OK) {
  365. if (addr) {
  366. *addr = param.outputs.addr;
  367. *addrlen = param.outputs.addrlen;
  368. }
  369. if (textaddr) {
  370. *textaddr = param.outputs.textaddr;
  371. }
  372. return recvd_len + param.outputs.returncode;
  373. }
  374. return recvd_len ? recvd_len : -1;
  375. }
  376. /* Similar to send() system call; send data to the stream, optionally
  377. * sending it as OOB data */
  378. PHPAPI int php_stream_xport_sendto(php_stream *stream, const char *buf, size_t buflen,
  379. int flags, void *addr, socklen_t addrlen)
  380. {
  381. php_stream_xport_param param;
  382. int ret = 0;
  383. int oob;
  384. #if 0
  385. if (flags == 0 && addr == NULL) {
  386. return php_stream_write(stream, buf, buflen);
  387. }
  388. #endif
  389. oob = (flags & STREAM_OOB) == STREAM_OOB;
  390. if ((oob || addr) && stream->writefilters.head) {
  391. php_error_docref(NULL, E_WARNING, "Cannot write OOB data, or data to a targeted address on a filtered stream");
  392. return -1;
  393. }
  394. memset(&param, 0, sizeof(param));
  395. param.op = STREAM_XPORT_OP_SEND;
  396. param.want_addr = addr ? 1 : 0;
  397. param.inputs.buf = (char*)buf;
  398. param.inputs.buflen = buflen;
  399. param.inputs.flags = flags;
  400. param.inputs.addr = addr;
  401. param.inputs.addrlen = addrlen;
  402. ret = php_stream_set_option(stream, PHP_STREAM_OPTION_XPORT_API, 0, &param);
  403. if (ret == PHP_STREAM_OPTION_RETURN_OK) {
  404. return param.outputs.returncode;
  405. }
  406. return -1;
  407. }
  408. /* Similar to shutdown() system call; shut down part of a full-duplex
  409. * connection */
  410. PHPAPI int php_stream_xport_shutdown(php_stream *stream, stream_shutdown_t how)
  411. {
  412. php_stream_xport_param param;
  413. int ret = 0;
  414. memset(&param, 0, sizeof(param));
  415. param.op = STREAM_XPORT_OP_SHUTDOWN;
  416. param.how = how;
  417. ret = php_stream_set_option(stream, PHP_STREAM_OPTION_XPORT_API, 0, &param);
  418. if (ret == PHP_STREAM_OPTION_RETURN_OK) {
  419. return param.outputs.returncode;
  420. }
  421. return -1;
  422. }