sysvshm.c 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455
  1. /*
  2. +----------------------------------------------------------------------+
  3. | PHP Version 7 |
  4. +----------------------------------------------------------------------+
  5. | Copyright (c) 1997-2018 The PHP Group |
  6. +----------------------------------------------------------------------+
  7. | This source file is subject to version 3.01 of the PHP license, |
  8. | that is bundled with this package in the file LICENSE, and is |
  9. | available through the world-wide-web at the following url: |
  10. | http://www.php.net/license/3_01.txt |
  11. | If you did not receive a copy of the PHP license and are unable to |
  12. | obtain it through the world-wide-web, please send a note to |
  13. | license@php.net so we can mail you a copy immediately. |
  14. +----------------------------------------------------------------------+
  15. | Author: Christian Cartus <cartus@atrior.de> |
  16. +----------------------------------------------------------------------+
  17. */
  18. /* This has been built and tested on Linux 2.2.14
  19. *
  20. * This has been built and tested on Solaris 2.6.
  21. * It may not compile or execute correctly on other systems.
  22. */
  23. #ifdef HAVE_CONFIG_H
  24. #include "config.h"
  25. #endif
  26. #include "php.h"
  27. #if HAVE_SYSVSHM
  28. #include <errno.h>
  29. #include "php_sysvshm.h"
  30. #include "ext/standard/info.h"
  31. #include "ext/standard/php_var.h"
  32. #include "zend_smart_str.h"
  33. #include "php_ini.h"
  34. /* {{{ arginfo */
  35. ZEND_BEGIN_ARG_INFO_EX(arginfo_shm_attach, 0, 0, 1)
  36. ZEND_ARG_INFO(0, key)
  37. ZEND_ARG_INFO(0, memsize)
  38. ZEND_ARG_INFO(0, perm)
  39. ZEND_END_ARG_INFO()
  40. ZEND_BEGIN_ARG_INFO_EX(arginfo_shm_detach, 0, 0, 1)
  41. ZEND_ARG_INFO(0, shm_identifier)
  42. ZEND_END_ARG_INFO()
  43. ZEND_BEGIN_ARG_INFO_EX(arginfo_shm_has_var, 0, 0, 2)
  44. ZEND_ARG_INFO(0, id)
  45. ZEND_ARG_INFO(0, variable_key)
  46. ZEND_END_ARG_INFO()
  47. ZEND_BEGIN_ARG_INFO_EX(arginfo_shm_remove, 0, 0, 1)
  48. ZEND_ARG_INFO(0, shm_identifier)
  49. ZEND_END_ARG_INFO()
  50. ZEND_BEGIN_ARG_INFO_EX(arginfo_shm_put_var, 0, 0, 3)
  51. ZEND_ARG_INFO(0, shm_identifier)
  52. ZEND_ARG_INFO(0, variable_key)
  53. ZEND_ARG_INFO(0, variable)
  54. ZEND_END_ARG_INFO()
  55. ZEND_BEGIN_ARG_INFO_EX(arginfo_shm_get_var, 0, 0, 2)
  56. ZEND_ARG_INFO(0, id)
  57. ZEND_ARG_INFO(0, variable_key)
  58. ZEND_END_ARG_INFO()
  59. ZEND_BEGIN_ARG_INFO_EX(arginfo_shm_remove_var, 0, 0, 2)
  60. ZEND_ARG_INFO(0, id)
  61. ZEND_ARG_INFO(0, variable_key)
  62. ZEND_END_ARG_INFO()
  63. /* }}} */
  64. /* {{{ sysvshm_functions[]
  65. */
  66. static const zend_function_entry sysvshm_functions[] = {
  67. PHP_FE(shm_attach, arginfo_shm_attach)
  68. PHP_FE(shm_remove, arginfo_shm_detach)
  69. PHP_FE(shm_detach, arginfo_shm_remove)
  70. PHP_FE(shm_put_var, arginfo_shm_put_var)
  71. PHP_FE(shm_has_var, arginfo_shm_has_var)
  72. PHP_FE(shm_get_var, arginfo_shm_get_var)
  73. PHP_FE(shm_remove_var, arginfo_shm_remove_var)
  74. PHP_FE_END
  75. };
  76. /* }}} */
  77. /* {{{ sysvshm_module_entry
  78. */
  79. zend_module_entry sysvshm_module_entry = {
  80. STANDARD_MODULE_HEADER,
  81. "sysvshm",
  82. sysvshm_functions,
  83. PHP_MINIT(sysvshm),
  84. NULL,
  85. NULL,
  86. NULL,
  87. PHP_MINFO(sysvshm),
  88. PHP_SYSVSHM_VERSION,
  89. STANDARD_MODULE_PROPERTIES
  90. };
  91. /* }}} */
  92. #ifdef COMPILE_DL_SYSVSHM
  93. ZEND_GET_MODULE(sysvshm)
  94. #endif
  95. #undef shm_ptr /* undefine AIX-specific macro */
  96. #define SHM_FETCH_RESOURCE(shm_ptr, z_ptr) do { \
  97. if ((shm_ptr = (sysvshm_shm *)zend_fetch_resource(Z_RES_P(z_ptr), PHP_SHM_RSRC_NAME, php_sysvshm.le_shm)) == NULL) { \
  98. RETURN_FALSE; \
  99. } \
  100. } while (0)
  101. THREAD_LS sysvshm_module php_sysvshm;
  102. static int php_put_shm_data(sysvshm_chunk_head *ptr, zend_long key, const char *data, zend_long len);
  103. static zend_long php_check_shm_data(sysvshm_chunk_head *ptr, zend_long key);
  104. static int php_remove_shm_data(sysvshm_chunk_head *ptr, zend_long shm_varpos);
  105. /* {{{ php_release_sysvshm
  106. */
  107. static void php_release_sysvshm(zend_resource *rsrc)
  108. {
  109. sysvshm_shm *shm_ptr = (sysvshm_shm *) rsrc->ptr;
  110. shmdt((void *) shm_ptr->ptr);
  111. efree(shm_ptr);
  112. }
  113. /* }}} */
  114. /* {{{ PHP_MINIT_FUNCTION
  115. */
  116. PHP_MINIT_FUNCTION(sysvshm)
  117. {
  118. php_sysvshm.le_shm = zend_register_list_destructors_ex(php_release_sysvshm, NULL, PHP_SHM_RSRC_NAME, module_number);
  119. if (cfg_get_long("sysvshm.init_mem", &php_sysvshm.init_mem) == FAILURE) {
  120. php_sysvshm.init_mem=10000;
  121. }
  122. return SUCCESS;
  123. }
  124. /* }}} */
  125. /* {{{ PHP_MINFO_FUNCTION
  126. */
  127. PHP_MINFO_FUNCTION(sysvshm)
  128. {
  129. php_info_print_table_start();
  130. php_info_print_table_row(2, "sysvshm support", "enabled");
  131. php_info_print_table_end();
  132. }
  133. /* }}} */
  134. /* {{{ proto int shm_attach(int key [, int memsize [, int perm]])
  135. Creates or open a shared memory segment */
  136. PHP_FUNCTION(shm_attach)
  137. {
  138. sysvshm_shm *shm_list_ptr;
  139. char *shm_ptr;
  140. sysvshm_chunk_head *chunk_ptr;
  141. zend_long shm_key, shm_id, shm_size = php_sysvshm.init_mem, shm_flag = 0666;
  142. if (SUCCESS != zend_parse_parameters(ZEND_NUM_ARGS(), "l|ll", &shm_key, &shm_size, &shm_flag)) {
  143. return;
  144. }
  145. if (shm_size < 1) {
  146. php_error_docref(NULL, E_WARNING, "Segment size must be greater than zero");
  147. RETURN_FALSE;
  148. }
  149. shm_list_ptr = (sysvshm_shm *) emalloc(sizeof(sysvshm_shm));
  150. /* get the id from a specified key or create new shared memory */
  151. if ((shm_id = shmget(shm_key, 0, 0)) < 0) {
  152. if (shm_size < (zend_long)sizeof(sysvshm_chunk_head)) {
  153. php_error_docref(NULL, E_WARNING, "failed for key 0x" ZEND_XLONG_FMT ": memorysize too small", shm_key);
  154. efree(shm_list_ptr);
  155. RETURN_FALSE;
  156. }
  157. if ((shm_id = shmget(shm_key, shm_size, shm_flag | IPC_CREAT | IPC_EXCL)) < 0) {
  158. php_error_docref(NULL, E_WARNING, "failed for key 0x" ZEND_XLONG_FMT ": %s", shm_key, strerror(errno));
  159. efree(shm_list_ptr);
  160. RETURN_FALSE;
  161. }
  162. }
  163. if ((shm_ptr = shmat(shm_id, NULL, 0)) == (void *) -1) {
  164. php_error_docref(NULL, E_WARNING, "failed for key 0x" ZEND_XLONG_FMT ": %s", shm_key, strerror(errno));
  165. efree(shm_list_ptr);
  166. RETURN_FALSE;
  167. }
  168. /* check if shm is already initialized */
  169. chunk_ptr = (sysvshm_chunk_head *) shm_ptr;
  170. if (strcmp((char*) &(chunk_ptr->magic), "PHP_SM") != 0) {
  171. strcpy((char*) &(chunk_ptr->magic), "PHP_SM");
  172. chunk_ptr->start = sizeof(sysvshm_chunk_head);
  173. chunk_ptr->end = chunk_ptr->start;
  174. chunk_ptr->total = shm_size;
  175. chunk_ptr->free = shm_size-chunk_ptr->end;
  176. }
  177. shm_list_ptr->key = shm_key;
  178. shm_list_ptr->id = shm_id;
  179. shm_list_ptr->ptr = chunk_ptr;
  180. RETURN_RES(zend_register_resource(shm_list_ptr, php_sysvshm.le_shm));
  181. }
  182. /* }}} */
  183. /* {{{ proto bool shm_detach(resource shm_identifier)
  184. Disconnects from shared memory segment */
  185. PHP_FUNCTION(shm_detach)
  186. {
  187. zval *shm_id;
  188. sysvshm_shm *shm_list_ptr;
  189. if (SUCCESS != zend_parse_parameters(ZEND_NUM_ARGS(), "r", &shm_id)) {
  190. return;
  191. }
  192. SHM_FETCH_RESOURCE(shm_list_ptr, shm_id);
  193. RETURN_BOOL(SUCCESS == zend_list_close(Z_RES_P(shm_id)));
  194. }
  195. /* }}} */
  196. /* {{{ proto bool shm_remove(resource shm_identifier)
  197. Removes shared memory from Unix systems */
  198. PHP_FUNCTION(shm_remove)
  199. {
  200. zval *shm_id;
  201. sysvshm_shm *shm_list_ptr;
  202. if (SUCCESS != zend_parse_parameters(ZEND_NUM_ARGS(), "r", &shm_id)) {
  203. return;
  204. }
  205. SHM_FETCH_RESOURCE(shm_list_ptr, shm_id);
  206. if (shmctl(shm_list_ptr->id, IPC_RMID, NULL) < 0) {
  207. php_error_docref(NULL, E_WARNING, "failed for key 0x%x, id " ZEND_LONG_FMT ": %s", shm_list_ptr->key, Z_LVAL_P(shm_id), strerror(errno));
  208. RETURN_FALSE;
  209. }
  210. RETURN_TRUE;
  211. }
  212. /* }}} */
  213. /* {{{ proto bool shm_put_var(resource shm_identifier, int variable_key, mixed variable)
  214. Inserts or updates a variable in shared memory */
  215. PHP_FUNCTION(shm_put_var)
  216. {
  217. zval *shm_id, *arg_var;
  218. int ret;
  219. zend_long shm_key;
  220. sysvshm_shm *shm_list_ptr;
  221. smart_str shm_var = {0};
  222. php_serialize_data_t var_hash;
  223. if (SUCCESS != zend_parse_parameters(ZEND_NUM_ARGS(), "rlz", &shm_id, &shm_key, &arg_var)) {
  224. return;
  225. }
  226. /* setup string-variable and serialize */
  227. PHP_VAR_SERIALIZE_INIT(var_hash);
  228. php_var_serialize(&shm_var, arg_var, &var_hash);
  229. PHP_VAR_SERIALIZE_DESTROY(var_hash);
  230. shm_list_ptr = zend_fetch_resource(Z_RES_P(shm_id), PHP_SHM_RSRC_NAME, php_sysvshm.le_shm);
  231. if (!shm_list_ptr) {
  232. smart_str_free(&shm_var);
  233. RETURN_FALSE;
  234. }
  235. /* insert serialized variable into shared memory */
  236. ret = php_put_shm_data(shm_list_ptr->ptr, shm_key, shm_var.s? ZSTR_VAL(shm_var.s) : NULL, shm_var.s? ZSTR_LEN(shm_var.s) : 0);
  237. /* free string */
  238. smart_str_free(&shm_var);
  239. if (ret == -1) {
  240. php_error_docref(NULL, E_WARNING, "not enough shared memory left");
  241. RETURN_FALSE;
  242. }
  243. RETURN_TRUE;
  244. }
  245. /* }}} */
  246. /* {{{ proto mixed shm_get_var(resource id, int variable_key)
  247. Returns a variable from shared memory */
  248. PHP_FUNCTION(shm_get_var)
  249. {
  250. zval *shm_id;
  251. zend_long shm_key;
  252. sysvshm_shm *shm_list_ptr;
  253. char *shm_data;
  254. zend_long shm_varpos;
  255. sysvshm_chunk *shm_var;
  256. php_unserialize_data_t var_hash;
  257. if (SUCCESS != zend_parse_parameters(ZEND_NUM_ARGS(), "rl", &shm_id, &shm_key)) {
  258. return;
  259. }
  260. SHM_FETCH_RESOURCE(shm_list_ptr, shm_id);
  261. /* setup string-variable and serialize */
  262. /* get serialized variable from shared memory */
  263. shm_varpos = php_check_shm_data((shm_list_ptr->ptr), shm_key);
  264. if (shm_varpos < 0) {
  265. php_error_docref(NULL, E_WARNING, "variable key " ZEND_LONG_FMT " doesn't exist", shm_key);
  266. RETURN_FALSE;
  267. }
  268. shm_var = (sysvshm_chunk*) ((char *)shm_list_ptr->ptr + shm_varpos);
  269. shm_data = &shm_var->mem;
  270. PHP_VAR_UNSERIALIZE_INIT(var_hash);
  271. if (php_var_unserialize(return_value, (const unsigned char **) &shm_data, (unsigned char *) shm_data + shm_var->length, &var_hash) != 1) {
  272. php_error_docref(NULL, E_WARNING, "variable data in shared memory is corrupted");
  273. RETVAL_FALSE;
  274. }
  275. PHP_VAR_UNSERIALIZE_DESTROY(var_hash);
  276. }
  277. /* }}} */
  278. /* {{{ proto bool shm_has_var(resource id, int variable_key)
  279. Checks whether a specific entry exists */
  280. PHP_FUNCTION(shm_has_var)
  281. {
  282. zval *shm_id;
  283. zend_long shm_key;
  284. sysvshm_shm *shm_list_ptr;
  285. if (SUCCESS != zend_parse_parameters(ZEND_NUM_ARGS(), "rl", &shm_id, &shm_key)) {
  286. return;
  287. }
  288. SHM_FETCH_RESOURCE(shm_list_ptr, shm_id);
  289. RETURN_BOOL(php_check_shm_data(shm_list_ptr->ptr, shm_key) >= 0);
  290. }
  291. /* }}} */
  292. /* {{{ proto bool shm_remove_var(resource id, int variable_key)
  293. Removes variable from shared memory */
  294. PHP_FUNCTION(shm_remove_var)
  295. {
  296. zval *shm_id;
  297. zend_long shm_key, shm_varpos;
  298. sysvshm_shm *shm_list_ptr;
  299. if (SUCCESS != zend_parse_parameters(ZEND_NUM_ARGS(), "rl", &shm_id, &shm_key)) {
  300. return;
  301. }
  302. SHM_FETCH_RESOURCE(shm_list_ptr, shm_id);
  303. shm_varpos = php_check_shm_data((shm_list_ptr->ptr), shm_key);
  304. if (shm_varpos < 0) {
  305. php_error_docref(NULL, E_WARNING, "variable key " ZEND_LONG_FMT " doesn't exist", shm_key);
  306. RETURN_FALSE;
  307. }
  308. php_remove_shm_data((shm_list_ptr->ptr), shm_varpos);
  309. RETURN_TRUE;
  310. }
  311. /* }}} */
  312. /* {{{ php_put_shm_data
  313. * inserts an ascii-string into shared memory */
  314. static int php_put_shm_data(sysvshm_chunk_head *ptr, zend_long key, const char *data, zend_long len)
  315. {
  316. sysvshm_chunk *shm_var;
  317. zend_long total_size;
  318. zend_long shm_varpos;
  319. total_size = ((zend_long) (len + sizeof(sysvshm_chunk) - 1) / sizeof(zend_long)) * sizeof(zend_long) + sizeof(zend_long); /* zend_long alligment */
  320. if ((shm_varpos = php_check_shm_data(ptr, key)) > 0) {
  321. php_remove_shm_data(ptr, shm_varpos);
  322. }
  323. if (ptr->free < total_size) {
  324. return -1; /* not enough memory */
  325. }
  326. shm_var = (sysvshm_chunk *) ((char *) ptr + ptr->end);
  327. shm_var->key = key;
  328. shm_var->length = len;
  329. shm_var->next = total_size;
  330. memcpy(&(shm_var->mem), data, len);
  331. ptr->end += total_size;
  332. ptr->free -= total_size;
  333. return 0;
  334. }
  335. /* }}} */
  336. /* {{{ php_check_shm_data
  337. */
  338. static zend_long php_check_shm_data(sysvshm_chunk_head *ptr, zend_long key)
  339. {
  340. zend_long pos;
  341. sysvshm_chunk *shm_var;
  342. pos = ptr->start;
  343. for (;;) {
  344. if (pos >= ptr->end) {
  345. return -1;
  346. }
  347. shm_var = (sysvshm_chunk*) ((char *) ptr + pos);
  348. if (shm_var->key == key) {
  349. return pos;
  350. }
  351. pos += shm_var->next;
  352. if (shm_var->next <= 0 || pos < ptr->start) {
  353. return -1;
  354. }
  355. }
  356. return -1;
  357. }
  358. /* }}} */
  359. /* {{{ php_remove_shm_data
  360. */
  361. static int php_remove_shm_data(sysvshm_chunk_head *ptr, zend_long shm_varpos)
  362. {
  363. sysvshm_chunk *chunk_ptr, *next_chunk_ptr;
  364. zend_long memcpy_len;
  365. chunk_ptr = (sysvshm_chunk *) ((char *) ptr + shm_varpos);
  366. next_chunk_ptr = (sysvshm_chunk *) ((char *) ptr + shm_varpos + chunk_ptr->next);
  367. memcpy_len = ptr->end-shm_varpos - chunk_ptr->next;
  368. ptr->free += chunk_ptr->next;
  369. ptr->end -= chunk_ptr->next;
  370. if (memcpy_len > 0) {
  371. memmove(chunk_ptr, next_chunk_ptr, memcpy_len);
  372. }
  373. return 0;
  374. }
  375. /* }}} */
  376. #endif /* HAVE_SYSVSHM */
  377. /*
  378. * Local variables:
  379. * tab-width: 4
  380. * c-basic-offset: 4
  381. * End:
  382. * vim600: sw=4 ts=4 fdm=marker
  383. * vim<600: sw=4 ts=4
  384. */