sysvshm.c 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442
  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: Christian Cartus <cartus@atrior.de> |
  14. +----------------------------------------------------------------------+
  15. */
  16. #ifdef HAVE_CONFIG_H
  17. #include "config.h"
  18. #endif
  19. #include "php.h"
  20. #ifdef HAVE_SYSVSHM
  21. #include <errno.h>
  22. #include "php_sysvshm.h"
  23. #include "sysvshm_arginfo.h"
  24. #include "ext/standard/info.h"
  25. #include "ext/standard/php_var.h"
  26. #include "zend_smart_str.h"
  27. #include "php_ini.h"
  28. /* SysvSharedMemory class */
  29. zend_class_entry *sysvshm_ce;
  30. static zend_object_handlers sysvshm_object_handlers;
  31. static inline sysvshm_shm *sysvshm_from_obj(zend_object *obj) {
  32. return (sysvshm_shm *)((char *)(obj) - XtOffsetOf(sysvshm_shm, std));
  33. }
  34. #define Z_SYSVSHM_P(zv) sysvshm_from_obj(Z_OBJ_P(zv))
  35. static zend_object *sysvshm_create_object(zend_class_entry *class_type) {
  36. sysvshm_shm *intern = zend_object_alloc(sizeof(sysvshm_shm), class_type);
  37. zend_object_std_init(&intern->std, class_type);
  38. object_properties_init(&intern->std, class_type);
  39. intern->std.handlers = &sysvshm_object_handlers;
  40. return &intern->std;
  41. }
  42. static zend_function *sysvshm_get_constructor(zend_object *object) {
  43. zend_throw_error(NULL, "Cannot directly construct SysvSharedMemory, use shm_attach() instead");
  44. return NULL;
  45. }
  46. static void sysvshm_free_obj(zend_object *object)
  47. {
  48. sysvshm_shm *sysvshm = sysvshm_from_obj(object);
  49. if (sysvshm->ptr) {
  50. shmdt((void *) sysvshm->ptr);
  51. }
  52. zend_object_std_dtor(&sysvshm->std);
  53. }
  54. /* {{{ sysvshm_module_entry */
  55. zend_module_entry sysvshm_module_entry = {
  56. STANDARD_MODULE_HEADER,
  57. "sysvshm",
  58. ext_functions,
  59. PHP_MINIT(sysvshm),
  60. NULL,
  61. NULL,
  62. NULL,
  63. PHP_MINFO(sysvshm),
  64. PHP_SYSVSHM_VERSION,
  65. STANDARD_MODULE_PROPERTIES
  66. };
  67. /* }}} */
  68. #ifdef COMPILE_DL_SYSVSHM
  69. ZEND_GET_MODULE(sysvshm)
  70. #endif
  71. #undef shm_ptr /* undefine AIX-specific macro */
  72. /* TODO: Make this thread-safe. */
  73. sysvshm_module php_sysvshm;
  74. static int php_put_shm_data(sysvshm_chunk_head *ptr, zend_long key, const char *data, zend_long len);
  75. static zend_long php_check_shm_data(sysvshm_chunk_head *ptr, zend_long key);
  76. static int php_remove_shm_data(sysvshm_chunk_head *ptr, zend_long shm_varpos);
  77. /* {{{ PHP_MINIT_FUNCTION */
  78. PHP_MINIT_FUNCTION(sysvshm)
  79. {
  80. sysvshm_ce = register_class_SysvSharedMemory();
  81. sysvshm_ce->create_object = sysvshm_create_object;
  82. memcpy(&sysvshm_object_handlers, &std_object_handlers, sizeof(zend_object_handlers));
  83. sysvshm_object_handlers.offset = XtOffsetOf(sysvshm_shm, std);
  84. sysvshm_object_handlers.free_obj = sysvshm_free_obj;
  85. sysvshm_object_handlers.get_constructor = sysvshm_get_constructor;
  86. sysvshm_object_handlers.clone_obj = NULL;
  87. sysvshm_object_handlers.compare = zend_objects_not_comparable;
  88. if (cfg_get_long("sysvshm.init_mem", &php_sysvshm.init_mem) == FAILURE) {
  89. php_sysvshm.init_mem=10000;
  90. }
  91. return SUCCESS;
  92. }
  93. /* }}} */
  94. /* {{{ PHP_MINFO_FUNCTION */
  95. PHP_MINFO_FUNCTION(sysvshm)
  96. {
  97. php_info_print_table_start();
  98. php_info_print_table_row(2, "sysvshm support", "enabled");
  99. php_info_print_table_end();
  100. }
  101. /* }}} */
  102. /* {{{ Creates or open a shared memory segment */
  103. PHP_FUNCTION(shm_attach)
  104. {
  105. sysvshm_shm *shm_list_ptr;
  106. char *shm_ptr;
  107. sysvshm_chunk_head *chunk_ptr;
  108. zend_long shm_key, shm_id, shm_size, shm_flag = 0666;
  109. bool shm_size_is_null = 1;
  110. if (SUCCESS != zend_parse_parameters(ZEND_NUM_ARGS(), "l|l!l", &shm_key, &shm_size, &shm_size_is_null, &shm_flag)) {
  111. RETURN_THROWS();
  112. }
  113. if (shm_size_is_null) {
  114. shm_size = php_sysvshm.init_mem;
  115. }
  116. if (shm_size < 1) {
  117. zend_argument_value_error(2, "must be greater than 0");
  118. RETURN_THROWS();
  119. }
  120. /* get the id from a specified key or create new shared memory */
  121. if ((shm_id = shmget(shm_key, 0, 0)) < 0) {
  122. if (shm_size < (zend_long)sizeof(sysvshm_chunk_head)) {
  123. php_error_docref(NULL, E_WARNING, "Failed for key 0x" ZEND_XLONG_FMT ": memorysize too small", shm_key);
  124. RETURN_FALSE;
  125. }
  126. if ((shm_id = shmget(shm_key, shm_size, shm_flag | IPC_CREAT | IPC_EXCL)) < 0) {
  127. php_error_docref(NULL, E_WARNING, "Failed for key 0x" ZEND_XLONG_FMT ": %s", shm_key, strerror(errno));
  128. RETURN_FALSE;
  129. }
  130. }
  131. if ((shm_ptr = shmat(shm_id, NULL, 0)) == (void *) -1) {
  132. php_error_docref(NULL, E_WARNING, "Failed for key 0x" ZEND_XLONG_FMT ": %s", shm_key, strerror(errno));
  133. RETURN_FALSE;
  134. }
  135. /* check if shm is already initialized */
  136. chunk_ptr = (sysvshm_chunk_head *) shm_ptr;
  137. if (strcmp((char*) &(chunk_ptr->magic), "PHP_SM") != 0) {
  138. strcpy((char*) &(chunk_ptr->magic), "PHP_SM");
  139. chunk_ptr->start = sizeof(sysvshm_chunk_head);
  140. chunk_ptr->end = chunk_ptr->start;
  141. chunk_ptr->total = shm_size;
  142. chunk_ptr->free = shm_size-chunk_ptr->end;
  143. }
  144. object_init_ex(return_value, sysvshm_ce);
  145. shm_list_ptr = Z_SYSVSHM_P(return_value);
  146. shm_list_ptr->key = shm_key;
  147. shm_list_ptr->id = shm_id;
  148. shm_list_ptr->ptr = chunk_ptr;
  149. }
  150. /* }}} */
  151. /* {{{ Disconnects from shared memory segment */
  152. PHP_FUNCTION(shm_detach)
  153. {
  154. zval *shm_id;
  155. sysvshm_shm *shm_list_ptr;
  156. if (SUCCESS != zend_parse_parameters(ZEND_NUM_ARGS(), "O", &shm_id, sysvshm_ce)) {
  157. RETURN_THROWS();
  158. }
  159. shm_list_ptr = Z_SYSVSHM_P(shm_id);
  160. if (!shm_list_ptr->ptr) {
  161. zend_throw_error(NULL, "Shared memory block has already been destroyed");
  162. RETURN_THROWS();
  163. }
  164. shmdt((void *) shm_list_ptr->ptr);
  165. shm_list_ptr->ptr = NULL;
  166. RETURN_TRUE;
  167. }
  168. /* }}} */
  169. /* {{{ Removes shared memory from Unix systems */
  170. PHP_FUNCTION(shm_remove)
  171. {
  172. zval *shm_id;
  173. sysvshm_shm *shm_list_ptr;
  174. if (SUCCESS != zend_parse_parameters(ZEND_NUM_ARGS(), "O", &shm_id, sysvshm_ce)) {
  175. RETURN_THROWS();
  176. }
  177. shm_list_ptr = Z_SYSVSHM_P(shm_id);
  178. if (!shm_list_ptr->ptr) {
  179. zend_throw_error(NULL, "Shared memory block has already been destroyed");
  180. RETURN_THROWS();
  181. }
  182. if (shmctl(shm_list_ptr->id, IPC_RMID, NULL) < 0) {
  183. 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));
  184. RETURN_FALSE;
  185. }
  186. RETURN_TRUE;
  187. }
  188. /* }}} */
  189. /* {{{ Inserts or updates a variable in shared memory */
  190. PHP_FUNCTION(shm_put_var)
  191. {
  192. zval *shm_id, *arg_var;
  193. int ret;
  194. zend_long shm_key;
  195. sysvshm_shm *shm_list_ptr;
  196. smart_str shm_var = {0};
  197. php_serialize_data_t var_hash;
  198. if (SUCCESS != zend_parse_parameters(ZEND_NUM_ARGS(), "Olz", &shm_id, sysvshm_ce, &shm_key, &arg_var)) {
  199. RETURN_THROWS();
  200. }
  201. shm_list_ptr = Z_SYSVSHM_P(shm_id);
  202. if (!shm_list_ptr->ptr) {
  203. zend_throw_error(NULL, "Shared memory block has already been destroyed");
  204. RETURN_THROWS();
  205. }
  206. /* setup string-variable and serialize */
  207. PHP_VAR_SERIALIZE_INIT(var_hash);
  208. php_var_serialize(&shm_var, arg_var, &var_hash);
  209. PHP_VAR_SERIALIZE_DESTROY(var_hash);
  210. /* insert serialized variable into shared memory */
  211. 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);
  212. /* free string */
  213. smart_str_free(&shm_var);
  214. if (ret == -1) {
  215. php_error_docref(NULL, E_WARNING, "Not enough shared memory left");
  216. RETURN_FALSE;
  217. }
  218. RETURN_TRUE;
  219. }
  220. /* }}} */
  221. /* {{{ Returns a variable from shared memory */
  222. PHP_FUNCTION(shm_get_var)
  223. {
  224. zval *shm_id;
  225. zend_long shm_key;
  226. sysvshm_shm *shm_list_ptr;
  227. char *shm_data;
  228. zend_long shm_varpos;
  229. sysvshm_chunk *shm_var;
  230. php_unserialize_data_t var_hash;
  231. if (SUCCESS != zend_parse_parameters(ZEND_NUM_ARGS(), "Ol", &shm_id, sysvshm_ce, &shm_key)) {
  232. RETURN_THROWS();
  233. }
  234. shm_list_ptr = Z_SYSVSHM_P(shm_id);
  235. if (!shm_list_ptr->ptr) {
  236. zend_throw_error(NULL, "Shared memory block has already been destroyed");
  237. RETURN_THROWS();
  238. }
  239. /* setup string-variable and serialize */
  240. /* get serialized variable from shared memory */
  241. shm_varpos = php_check_shm_data(shm_list_ptr->ptr, shm_key);
  242. if (shm_varpos < 0) {
  243. php_error_docref(NULL, E_WARNING, "Variable key " ZEND_LONG_FMT " doesn't exist", shm_key);
  244. RETURN_FALSE;
  245. }
  246. shm_var = (sysvshm_chunk*) ((char *)shm_list_ptr->ptr + shm_varpos);
  247. shm_data = &shm_var->mem;
  248. PHP_VAR_UNSERIALIZE_INIT(var_hash);
  249. if (php_var_unserialize(return_value, (const unsigned char **) &shm_data, (unsigned char *) shm_data + shm_var->length, &var_hash) != 1) {
  250. php_error_docref(NULL, E_WARNING, "Variable data in shared memory is corrupted");
  251. RETVAL_FALSE;
  252. }
  253. PHP_VAR_UNSERIALIZE_DESTROY(var_hash);
  254. }
  255. /* }}} */
  256. /* {{{ Checks whether a specific entry exists */
  257. PHP_FUNCTION(shm_has_var)
  258. {
  259. zval *shm_id;
  260. zend_long shm_key;
  261. sysvshm_shm *shm_list_ptr;
  262. if (SUCCESS != zend_parse_parameters(ZEND_NUM_ARGS(), "Ol", &shm_id, sysvshm_ce, &shm_key)) {
  263. RETURN_THROWS();
  264. }
  265. shm_list_ptr = Z_SYSVSHM_P(shm_id);
  266. if (!shm_list_ptr->ptr) {
  267. zend_throw_error(NULL, "Shared memory block has already been destroyed");
  268. RETURN_THROWS();
  269. }
  270. RETURN_BOOL(php_check_shm_data(shm_list_ptr->ptr, shm_key) >= 0);
  271. }
  272. /* }}} */
  273. /* {{{ Removes variable from shared memory */
  274. PHP_FUNCTION(shm_remove_var)
  275. {
  276. zval *shm_id;
  277. zend_long shm_key, shm_varpos;
  278. sysvshm_shm *shm_list_ptr;
  279. if (SUCCESS != zend_parse_parameters(ZEND_NUM_ARGS(), "Ol", &shm_id, sysvshm_ce, &shm_key)) {
  280. RETURN_THROWS();
  281. }
  282. shm_list_ptr = Z_SYSVSHM_P(shm_id);
  283. if (!shm_list_ptr->ptr) {
  284. zend_throw_error(NULL, "Shared memory block has already been destroyed");
  285. RETURN_THROWS();
  286. }
  287. shm_varpos = php_check_shm_data(shm_list_ptr->ptr, shm_key);
  288. if (shm_varpos < 0) {
  289. php_error_docref(NULL, E_WARNING, "Variable key " ZEND_LONG_FMT " doesn't exist", shm_key);
  290. RETURN_FALSE;
  291. }
  292. php_remove_shm_data((shm_list_ptr->ptr), shm_varpos);
  293. RETURN_TRUE;
  294. }
  295. /* }}} */
  296. /* {{{ php_put_shm_data
  297. * inserts an ascii-string into shared memory */
  298. static int php_put_shm_data(sysvshm_chunk_head *ptr, zend_long key, const char *data, zend_long len)
  299. {
  300. sysvshm_chunk *shm_var;
  301. zend_long total_size;
  302. zend_long shm_varpos;
  303. total_size = ((zend_long) (len + sizeof(sysvshm_chunk) - 1) / sizeof(zend_long)) * sizeof(zend_long) + sizeof(zend_long); /* zend_long alligment */
  304. if ((shm_varpos = php_check_shm_data(ptr, key)) > 0) {
  305. php_remove_shm_data(ptr, shm_varpos);
  306. }
  307. if (ptr->free < total_size) {
  308. return -1; /* not enough memory */
  309. }
  310. shm_var = (sysvshm_chunk *) ((char *) ptr + ptr->end);
  311. shm_var->key = key;
  312. shm_var->length = len;
  313. shm_var->next = total_size;
  314. memcpy(&(shm_var->mem), data, len);
  315. ptr->end += total_size;
  316. ptr->free -= total_size;
  317. return 0;
  318. }
  319. /* }}} */
  320. /* {{{ php_check_shm_data */
  321. static zend_long php_check_shm_data(sysvshm_chunk_head *ptr, zend_long key)
  322. {
  323. zend_long pos;
  324. sysvshm_chunk *shm_var;
  325. ZEND_ASSERT(ptr);
  326. pos = ptr->start;
  327. for (;;) {
  328. if (pos >= ptr->end) {
  329. return -1;
  330. }
  331. shm_var = (sysvshm_chunk*) ((char *) ptr + pos);
  332. if (shm_var->key == key) {
  333. return pos;
  334. }
  335. pos += shm_var->next;
  336. if (shm_var->next <= 0 || pos < ptr->start) {
  337. return -1;
  338. }
  339. }
  340. return -1;
  341. }
  342. /* }}} */
  343. /* {{{ php_remove_shm_data */
  344. static int php_remove_shm_data(sysvshm_chunk_head *ptr, zend_long shm_varpos)
  345. {
  346. sysvshm_chunk *chunk_ptr, *next_chunk_ptr;
  347. zend_long memcpy_len;
  348. ZEND_ASSERT(ptr);
  349. chunk_ptr = (sysvshm_chunk *) ((char *) ptr + shm_varpos);
  350. next_chunk_ptr = (sysvshm_chunk *) ((char *) ptr + shm_varpos + chunk_ptr->next);
  351. memcpy_len = ptr->end-shm_varpos - chunk_ptr->next;
  352. ptr->free += chunk_ptr->next;
  353. ptr->end -= chunk_ptr->next;
  354. if (memcpy_len > 0) {
  355. memmove(chunk_ptr, next_chunk_ptr, memcpy_len);
  356. }
  357. return 0;
  358. }
  359. /* }}} */
  360. #endif /* HAVE_SYSVSHM */