dataArg.c 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741
  1. /*
  2. * This file is part of the PikaPython project.
  3. * http://github.com/pikastech/pikapython
  4. *
  5. * MIT License
  6. *
  7. * Copyright (c) 2021 lyon liang6516@outlook.com
  8. *
  9. * Permission is hereby granted, free of charge, to any person obtaining a
  10. * copy of this software and associated documentation files (the "Software"),
  11. * to deal in the Software without restriction, including without limitation
  12. * the rights to use, copy, modify, merge, publish, distribute, sublicense,
  13. * and/or sell copies of the Software, and to permit persons to whom the
  14. * Software is furnished to do so, subject to the following conditions:
  15. *
  16. * The above copyright notice and this permission notice shall be included in
  17. * all copies or substantial portions of the Software.
  18. *
  19. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  20. * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  21. * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
  22. * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  23. * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  24. * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  25. * DEALINGS IN THE SOFTWARE.
  26. */
  27. #include "dataArg.h"
  28. #include "PikaObj.h"
  29. #include "dataArgs.h"
  30. #include "dataMemory.h"
  31. #include "dataString.h"
  32. #include "stdlib.h"
  33. static pika_bool _arg_cache_push(Arg* self, uint32_t size) {
  34. #if !PIKA_ARG_CACHE_ENABLE
  35. return pika_false;
  36. #else
  37. if (pika_false == pika_hook_arg_cache_filter(self)) {
  38. return pika_false;
  39. }
  40. extern PikaMemInfo g_PikaMemInfo;
  41. if (self->heap_size < PIKA_ARG_CACHE_SIZE ||
  42. self->heap_size > 2 * PIKA_ARG_CACHE_SIZE) {
  43. return pika_false;
  44. }
  45. if (PIKA_ARG_CACHE_POOL_SIZE <= g_PikaMemInfo.cache_pool_top) {
  46. return pika_false;
  47. }
  48. g_PikaMemInfo.cache_pool[g_PikaMemInfo.cache_pool_top++] = (uint8_t*)self;
  49. g_PikaMemInfo.heapUsed -= mem_align(sizeof(Arg) + size);
  50. return pika_true;
  51. #endif
  52. }
  53. static Arg* _arg_cache_pop(uint32_t size) {
  54. #if !PIKA_ARG_CACHE_ENABLE
  55. return NULL;
  56. #else
  57. uint32_t req_heap_size = mem_align(sizeof(Arg) + size);
  58. extern PikaMemInfo g_PikaMemInfo;
  59. if (req_heap_size > PIKA_ARG_CACHE_SIZE) {
  60. return NULL;
  61. }
  62. if (!(g_PikaMemInfo.cache_pool_top > 0)) {
  63. return NULL;
  64. }
  65. --g_PikaMemInfo.cache_pool_top;
  66. Arg* self = (Arg*)g_PikaMemInfo.cache_pool[g_PikaMemInfo.cache_pool_top];
  67. g_PikaMemInfo.heapUsed += mem_align(sizeof(Arg) + size);
  68. return self;
  69. #endif
  70. }
  71. uint32_t arg_getTotleSize(Arg* self) {
  72. return arg_totleSize(self);
  73. }
  74. /**
  75. * time33 hash
  76. */
  77. Hash hash_time33EndWith(char* str, char end) {
  78. pika_assert(str != NULL);
  79. if (*str == 0) {
  80. return 5381;
  81. }
  82. Hash hash = 5381;
  83. while (*str && *str != end) {
  84. hash += (hash << 5) + (*str++);
  85. }
  86. return (hash & 0x7FFFFFFF);
  87. }
  88. Hash hash_time33(char* str) {
  89. pika_assert(str != NULL);
  90. if (*str == 0) {
  91. return 5381;
  92. }
  93. Hash hash = 5381;
  94. while (*str) {
  95. hash += (hash << 5) + (*str++);
  96. }
  97. return (hash & 0x7FFFFFFF);
  98. }
  99. static Arg* _arg_set_hash(Arg* self,
  100. Hash nameHash,
  101. ArgType type,
  102. uint8_t* content,
  103. uint32_t size,
  104. Arg* next) {
  105. /* create arg if not exist */
  106. if (NULL == self || self->size < size) {
  107. self = _arg_cache_pop(size);
  108. uint32_t heap_size = sizeof(Arg) + size;
  109. #if PIKA_ARG_CACHE_ENABLE
  110. // if (heap_size < PIKA_ARG_CACHE_SIZE) {
  111. // heap_size = PIKA_ARG_CACHE_SIZE;
  112. // }
  113. extern PikaMemInfo g_PikaMemInfo;
  114. g_PikaMemInfo.alloc_times++;
  115. g_PikaMemInfo.alloc_times_cache++;
  116. #endif
  117. if (NULL == self) {
  118. self = (Arg*)pikaMalloc(heap_size);
  119. #if PIKA_ARG_CACHE_ENABLE
  120. extern PikaMemInfo g_PikaMemInfo;
  121. g_PikaMemInfo.alloc_times_cache--;
  122. self->heap_size = mem_align(heap_size);
  123. #endif
  124. }
  125. self->size = size;
  126. self->flag = 0;
  127. #if PIKA_KERNAL_DEBUG_ENABLE
  128. self->value = (_arg_value*)&self->content;
  129. self->str = (char*)&self->content;
  130. self->bytes = (uint8_t*)&self->content;
  131. self->name = self->_name_buff;
  132. pika_platform_memset(self->_name_buff, 0, PIKA_NAME_BUFF_SIZE);
  133. #endif
  134. arg_setSerialized(self, pika_true);
  135. // arg_setIsKeyword(self, pika_false);
  136. arg_setNext(self, next);
  137. }
  138. self->name_hash = nameHash;
  139. self->type = type;
  140. if (NULL != content) {
  141. pika_platform_memcpy(arg_getContent(self), content, size);
  142. } else {
  143. pika_platform_memset(arg_getContent(self), 0,
  144. align_by(size, sizeof(uint32_t)));
  145. }
  146. pika_assert(self->flag < ARG_FLAG_MAX);
  147. return self;
  148. }
  149. static Arg* arg_create_hash(Hash nameHash,
  150. ArgType type,
  151. uint8_t* content,
  152. uint32_t size,
  153. Arg* next) {
  154. return _arg_set_hash(NULL, nameHash, type, content, size, next);
  155. }
  156. static Arg* arg_create(char* name,
  157. ArgType type,
  158. uint8_t* content,
  159. uint32_t size,
  160. Arg* next) {
  161. Hash nameHash = hash_time33(name);
  162. return arg_create_hash(nameHash, type, content, size, next);
  163. }
  164. Arg* arg_set(Arg* self,
  165. char* name,
  166. ArgType type,
  167. uint8_t* content,
  168. uint32_t size) {
  169. Hash nameHash = hash_time33(name);
  170. self = _arg_set_hash(self, nameHash, type, content, size, NULL);
  171. #if PIKA_KERNAL_DEBUG_ENABLE
  172. strncpy(self->_name_buff, name, PIKA_NAME_BUFF_SIZE);
  173. #endif
  174. return self;
  175. }
  176. void arg_init_stack(Arg* self, uint8_t* buffer, uint32_t size) {
  177. self->_.buffer = buffer;
  178. self->size = size;
  179. #if PIKA_KERNAL_DEBUG_ENABLE
  180. self->value = (_arg_value*)&self->_.buffer;
  181. self->str = (char*)&self->content;
  182. self->bytes = (uint8_t*)&self->content;
  183. #endif
  184. }
  185. uint32_t arg_totleSize(Arg* self) {
  186. return ((Arg*)self)->size + sizeof(Arg);
  187. }
  188. void arg_freeContent(Arg* self) {
  189. pika_assert(NULL != self);
  190. if (_arg_cache_push(self, self->size)) {
  191. return;
  192. }
  193. pikaFree(self, arg_totleSize(self));
  194. return;
  195. }
  196. Arg* arg_setContent(Arg* self, uint8_t* content, uint32_t size) {
  197. if (NULL == self) {
  198. /* malloc */
  199. return arg_create("", ARG_TYPE_NONE, content, size, NULL);
  200. }
  201. /* only copy */
  202. if (arg_getSize(self) >= size) {
  203. pika_platform_memcpy(arg_getContent((Arg*)self), content, size);
  204. return self;
  205. }
  206. /* realloc */
  207. Hash nameHash = arg_getNameHash(self);
  208. ArgType type = arg_getType(self);
  209. Arg* next = arg_getNext(self);
  210. Arg* newContent = arg_create_hash(nameHash, type, content, size, next);
  211. arg_freeContent(self);
  212. return newContent;
  213. }
  214. Arg* arg_setNameHash(Arg* self, Hash nameHash) {
  215. if (NULL == self) {
  216. return arg_create_hash(nameHash, ARG_TYPE_NONE, NULL, 0, NULL);
  217. }
  218. Arg* arg = (Arg*)self;
  219. arg->name_hash = nameHash;
  220. return self;
  221. }
  222. Arg* arg_setName(Arg* self, char* name) {
  223. pika_assert(NULL != name);
  224. self = arg_setNameHash(self, hash_time33(name));
  225. #if PIKA_KERNAL_DEBUG_ENABLE
  226. strncpy(self->_name_buff, name, PIKA_NAME_BUFF_SIZE);
  227. #endif
  228. return self;
  229. }
  230. Arg* arg_setBytes(Arg* self, char* name, uint8_t* src, size_t size) {
  231. self = arg_newContent(size + sizeof(size_t) + 1);
  232. if (NULL == self) {
  233. return NULL;
  234. }
  235. self = arg_setName(self, name);
  236. pika_assert(NULL != self);
  237. arg_setType(self, ARG_TYPE_BYTES);
  238. void* dir = arg_getContent(self);
  239. /* set content all to 0 */
  240. pika_platform_memset(dir, 0, size + sizeof(size_t) + 1);
  241. /* setsize */
  242. pika_platform_memcpy(dir, &size, sizeof(size_t));
  243. /* set init value */
  244. if (NULL != src) {
  245. pika_platform_memcpy((void*)((uintptr_t)dir + sizeof(size_t)), src,
  246. size);
  247. }
  248. pika_assert(self->flag < ARG_FLAG_MAX);
  249. return self;
  250. }
  251. Arg* arg_newContent(uint32_t size) {
  252. Arg* newContent = arg_create("", ARG_TYPE_NONE, NULL, size, NULL);
  253. return newContent;
  254. }
  255. uint8_t* arg_getBytes(Arg* self) {
  256. return arg_getContent(self) + sizeof(size_t);
  257. }
  258. Arg* arg_toStrArg(Arg* arg) {
  259. ArgType type = arg_getType(arg);
  260. char* buff = (char*)pikaMalloc(PIKA_SPRINTF_BUFF_SIZE);
  261. Arg* result = NULL;
  262. if (type == ARG_TYPE_BYTES) {
  263. char buff_item[16] = {0};
  264. size_t bytes_size = arg_getBytesSize(arg);
  265. uint8_t* bytes = arg_getBytes(arg);
  266. Arg* str_arg = arg_newStr("b\'");
  267. for (size_t i = 0; i < bytes_size; i++) {
  268. pika_snprintf(buff_item, 16, "\\x%02x", bytes[i]);
  269. char* str_item = (char*)buff_item;
  270. str_arg = arg_strAppend(str_arg, str_item);
  271. }
  272. str_arg = arg_strAppend(str_arg, "\'");
  273. result = str_arg;
  274. goto __exit;
  275. }
  276. if (type == ARG_TYPE_INT) {
  277. #if PIKA_PRINT_LLD_ENABLE
  278. pika_snprintf(buff, PIKA_SPRINTF_BUFF_SIZE, "%lld",
  279. (long long int)arg_getInt(arg));
  280. #else
  281. pika_platform_snprintf(buff, PIKA_SPRINTF_BUFF_SIZE, "%d",
  282. (int)arg_getInt(arg));
  283. #endif
  284. result = arg_newStr(buff);
  285. goto __exit;
  286. }
  287. if (type == ARG_TYPE_BOOL) {
  288. if (arg_getBool(arg)) {
  289. result = arg_newStr("True");
  290. goto __exit;
  291. }
  292. result = arg_newStr("False");
  293. goto __exit;
  294. }
  295. if (type == ARG_TYPE_FLOAT) {
  296. pika_snprintf(buff, PIKA_SPRINTF_BUFF_SIZE, "%f", arg_getFloat(arg));
  297. result = arg_newStr(buff);
  298. goto __exit;
  299. }
  300. if (type == ARG_TYPE_STRING) {
  301. result = arg_newStr(arg_getStr(arg));
  302. goto __exit;
  303. }
  304. if (type == ARG_TYPE_POINTER) {
  305. pika_snprintf(buff, PIKA_SPRINTF_BUFF_SIZE, "%p", arg_getPtr(arg));
  306. result = arg_newStr(buff);
  307. goto __exit;
  308. }
  309. if (argType_isCallable(type)) {
  310. /* support basic type */
  311. if (_argType_or(type, ARG_TYPE_METHOD_NATIVE,
  312. ARG_TYPE_METHOD_NATIVE_ACTIVE)) {
  313. MethodProp* method_store = (MethodProp*)arg_getContent(arg);
  314. if (strEqu(method_store->name, "int") ||
  315. strEqu(method_store->name, "bool") ||
  316. strEqu(method_store->name, "float") ||
  317. strEqu(method_store->name, "str") ||
  318. strEqu(method_store->name, "bytes") ||
  319. strEqu(method_store->name, "list") ||
  320. strEqu(method_store->name, "dict") ||
  321. strEqu(method_store->name, "tuple")) {
  322. pika_snprintf(buff, PIKA_SPRINTF_BUFF_SIZE, "<class '%s'>",
  323. method_store->name);
  324. result = arg_newStr(buff);
  325. goto __exit;
  326. }
  327. pika_snprintf(buff, PIKA_SPRINTF_BUFF_SIZE,
  328. "<built-in function %s>", method_store->name);
  329. result = arg_newStr(buff);
  330. goto __exit;
  331. }
  332. if (argType_isConstructor(type)) {
  333. pika_snprintf(buff, PIKA_SPRINTF_BUFF_SIZE, "<class 'object'>");
  334. result = arg_newStr(buff);
  335. goto __exit;
  336. }
  337. pika_snprintf(buff, PIKA_SPRINTF_BUFF_SIZE, "<class 'function'>");
  338. result = arg_newStr(buff);
  339. goto __exit;
  340. }
  341. if (type == ARG_TYPE_NONE) {
  342. result = arg_newStr("None");
  343. goto __exit;
  344. }
  345. if (argType_isObject(type)) {
  346. result = arg_newStr(obj_toStr(arg_getPtr(arg)));
  347. goto __exit;
  348. }
  349. if (type == ARG_TYPE_OBJECT_META) {
  350. pika_snprintf(buff, PIKA_SPRINTF_BUFF_SIZE, "<meta object at %p>",
  351. arg_getPtr(arg));
  352. result = arg_newStr(buff);
  353. goto __exit;
  354. }
  355. __exit:
  356. pikaFree(buff, PIKA_SPRINTF_BUFF_SIZE);
  357. return result;
  358. }
  359. void arg_print(Arg* self, pika_bool in_REPL, char* end) {
  360. /* use arg_toStrArg() */
  361. Arg* str_arg = arg_toStrArg(self);
  362. if (NULL == str_arg) {
  363. return;
  364. }
  365. if (in_REPL && arg_getType(self) == ARG_TYPE_STRING) {
  366. pika_platform_printf("'%s'%s", arg_getStr(str_arg), end);
  367. } else {
  368. pika_platform_printf("%s%s", arg_getStr(str_arg), end);
  369. }
  370. arg_deinit(str_arg);
  371. return;
  372. }
  373. size_t arg_getBytesSize(Arg* self) {
  374. size_t mem_size = 0;
  375. void* content = (void*)arg_getContent(self);
  376. if (NULL == content) {
  377. return 0;
  378. }
  379. pika_platform_memcpy(&mem_size, content, sizeof(size_t));
  380. return mem_size;
  381. }
  382. Arg* arg_setStruct(Arg* self,
  383. char* name,
  384. void* struct_ptr,
  385. uint32_t struct_size) {
  386. if (NULL == struct_ptr) {
  387. return NULL;
  388. }
  389. return arg_set(self, name, ARG_TYPE_STRUCT, (uint8_t*)struct_ptr,
  390. struct_size);
  391. }
  392. Arg* arg_setHeapStruct(Arg* self,
  393. char* name,
  394. void* struct_ptr,
  395. uint32_t struct_size,
  396. void* struct_deinit_fun) {
  397. if (NULL == struct_ptr) {
  398. return NULL;
  399. }
  400. Arg* struct_arg =
  401. arg_setContent(NULL, (uint8_t*)&struct_deinit_fun, sizeof(void*));
  402. struct_arg = arg_append(struct_arg, (uint8_t*)struct_ptr, struct_size);
  403. pika_assert(NULL != struct_arg);
  404. arg_setType(struct_arg, ARG_TYPE_STRUCT_HEAP);
  405. struct_arg = arg_setName(struct_arg, name);
  406. return struct_arg;
  407. }
  408. void* arg_getHeapStructDeinitFun(Arg* self) {
  409. void* deinit_fun = NULL;
  410. pika_platform_memcpy(&deinit_fun, arg_getContent(self), sizeof(void*));
  411. return deinit_fun;
  412. }
  413. Arg* arg_setInt(Arg* self, char* name, int64_t val) {
  414. return arg_set(self, name, ARG_TYPE_INT, (uint8_t*)&val, sizeof(val));
  415. }
  416. Arg* arg_setBool(Arg* self, char* name, pika_bool val) {
  417. return arg_set(self, name, ARG_TYPE_BOOL, (uint8_t*)&val, sizeof(val));
  418. }
  419. Arg* arg_setNone(Arg* self) {
  420. return arg_set(self, "", ARG_TYPE_NONE, NULL, 0);
  421. }
  422. Arg* arg_setFloat(Arg* self, char* name, pika_float val) {
  423. return arg_set(self, name, ARG_TYPE_FLOAT, (uint8_t*)&val, sizeof(val));
  424. }
  425. pika_float arg_getFloat(Arg* self) {
  426. if (NULL == arg_getContent(self)) {
  427. return -999.999;
  428. }
  429. return *(pika_float*)arg_getContent(self);
  430. }
  431. Arg* arg_setPtr(Arg* self, char* name, ArgType type, void* pointer) {
  432. return arg_set(self, name, type, (uint8_t*)&pointer, sizeof(uintptr_t));
  433. }
  434. Arg* arg_setStrN(Arg* self, char* name, char* string, size_t len) {
  435. if (NULL == string) {
  436. return NULL;
  437. }
  438. Arg* ret = arg_set(self, name, ARG_TYPE_STRING, NULL, len + 1);
  439. pika_platform_memcpy(arg_getContent(ret), string, len);
  440. return ret;
  441. }
  442. Arg* arg_setStr(Arg* self, char* name, char* string) {
  443. if (NULL == string) {
  444. return NULL;
  445. }
  446. return arg_set(self, name, ARG_TYPE_STRING, (uint8_t*)string,
  447. strGetSize(string) + 1);
  448. }
  449. int64_t arg_getInt(Arg* self) {
  450. pika_assert(NULL != self);
  451. if (NULL == arg_getContent(self)) {
  452. return _PIKA_INT_ERR;
  453. }
  454. return *(int64_t*)arg_getContent(self);
  455. }
  456. pika_bool arg_getBool(Arg* self) {
  457. pika_assert(NULL != self);
  458. if (NULL == arg_getContent(self)) {
  459. return _PIKA_BOOL_ERR;
  460. }
  461. ArgType type = arg_getType(self);
  462. if (type == ARG_TYPE_BOOL) {
  463. return *(pika_bool*)arg_getContent(self);
  464. }
  465. if (type == ARG_TYPE_INT) {
  466. return (pika_bool)arg_getInt(self);
  467. }
  468. if (type == ARG_TYPE_FLOAT) {
  469. return (pika_bool)arg_getFloat(self);
  470. }
  471. return _PIKA_BOOL_ERR;
  472. }
  473. void* arg_getPtr(Arg* self) {
  474. if (arg_getType(self) == ARG_TYPE_NONE) {
  475. return NULL;
  476. }
  477. if (NULL == arg_getContent(self)) {
  478. return NULL;
  479. }
  480. return *(void**)arg_getContent(self);
  481. }
  482. char* arg_getStr(Arg* self) {
  483. return (char*)arg_getContent(self);
  484. }
  485. uint32_t arg_getContentSize(Arg* self) {
  486. return arg_getSize(self);
  487. }
  488. Arg* New_arg(void* voidPointer) {
  489. return NULL;
  490. }
  491. void arg_refcntInc(Arg* self) {
  492. ArgType arg_type = arg_getType(self);
  493. if (ARG_TYPE_OBJECT != arg_type) {
  494. return;
  495. }
  496. if (arg_getIsWeakRef(self)) {
  497. return;
  498. }
  499. obj_refcntInc((PikaObj*)arg_getPtr(self));
  500. }
  501. void arg_refcntDec(Arg* self) {
  502. ArgType arg_type = arg_getType(self);
  503. if (ARG_TYPE_OBJECT != arg_type) {
  504. return;
  505. }
  506. if (arg_getIsWeakRef(self)) {
  507. return;
  508. }
  509. obj_refcntDec((PikaObj*)arg_getPtr(self));
  510. }
  511. Arg* arg_copy_content(Arg* arg_dict, Arg* arg_src) {
  512. arg_dict = arg_setContent(arg_dict, arg_getContent(arg_src),
  513. arg_getContentSize(arg_src));
  514. arg_dict = arg_setNameHash(arg_dict, arg_getNameHash(arg_src));
  515. pika_assert(NULL != arg_dict);
  516. arg_setType(arg_dict, arg_getType(arg_src));
  517. arg_setIsKeyword(arg_dict, arg_getIsKeyword(arg_src));
  518. arg_setIsWeakRef(arg_dict, arg_getIsWeakRef(arg_src));
  519. return arg_dict;
  520. }
  521. Arg* arg_copy(Arg* arg_src) {
  522. if (NULL == arg_src) {
  523. return NULL;
  524. }
  525. pika_assert(arg_src->flag < ARG_FLAG_MAX);
  526. if (ARG_TYPE_OBJECT == arg_getType(arg_src)) {
  527. arg_refcntInc(arg_src);
  528. }
  529. #if 0
  530. if (argType_isObjectMethodActive(arg_getType(arg_src))) {
  531. PikaObj* hostObj = methodArg_getHostObj(arg_src);
  532. if (NULL != hostObj) {
  533. obj_refcntInc(hostObj);
  534. }
  535. }
  536. #endif
  537. Arg* arg_dict = New_arg(NULL);
  538. arg_dict = arg_copy_content(arg_dict, arg_src);
  539. return arg_dict;
  540. }
  541. Arg* arg_copy_noalloc(Arg* arg_src, Arg* arg_dict) {
  542. if (NULL == arg_src) {
  543. return NULL;
  544. }
  545. if (NULL == arg_dict) {
  546. return arg_copy(arg_src);
  547. }
  548. /* size is too big to be copied by noalloc */
  549. if (arg_getSize(arg_src) > arg_getSize(arg_dict)) {
  550. return arg_copy(arg_src);
  551. }
  552. arg_refcntInc(arg_src);
  553. arg_setSerialized(arg_dict, pika_false);
  554. arg_dict = arg_copy_content(arg_dict, arg_src);
  555. return arg_dict;
  556. }
  557. Arg* arg_append(Arg* self, void* new_content, size_t new_size) {
  558. uint8_t* old_content = arg_getContent(self);
  559. size_t old_size = arg_getContentSize(self);
  560. Arg* new_arg = NULL;
  561. #if PIKA_ARG_CACHE_ENABLE
  562. /* create arg_out */
  563. if (self->heap_size > mem_align(sizeof(Arg) + old_size + new_size)) {
  564. new_arg = self;
  565. new_arg->size = old_size + new_size;
  566. extern PikaMemInfo g_PikaMemInfo;
  567. g_PikaMemInfo.heapUsed += mem_align(sizeof(Arg) + old_size + new_size) -
  568. mem_align(sizeof(Arg) + old_size);
  569. }
  570. #endif
  571. if (NULL == new_arg) {
  572. new_arg = arg_setContent(NULL, NULL, old_size + new_size);
  573. }
  574. pika_assert(NULL != new_arg);
  575. arg_setType(new_arg, arg_getType(self));
  576. arg_setNameHash(new_arg, arg_getNameHash(self));
  577. if (self != new_arg) {
  578. /* copy old content */
  579. pika_platform_memcpy(arg_getContent(new_arg), old_content, old_size);
  580. }
  581. /* copy new content */
  582. pika_platform_memcpy(arg_getContent(new_arg) + old_size, new_content,
  583. new_size);
  584. if (self != new_arg) {
  585. arg_deinit(self);
  586. }
  587. return new_arg;
  588. }
  589. void* arg_getHeapStruct(Arg* self) {
  590. return arg_getContent(self) + sizeof(void*);
  591. }
  592. void arg_deinitHeap(Arg* self) {
  593. if (arg_getIsWeakRef(self)) {
  594. return;
  595. }
  596. ArgType type = arg_getType(self);
  597. /* deinit heap struct */
  598. if (type == ARG_TYPE_STRUCT_HEAP) {
  599. /* deinit heap strcut */
  600. StructDeinitFun struct_deinit_fun =
  601. (StructDeinitFun)arg_getHeapStructDeinitFun(self);
  602. struct_deinit_fun(arg_getHeapStruct(self));
  603. return;
  604. }
  605. /* deinit sub object */
  606. if (ARG_TYPE_OBJECT == type) {
  607. PikaObj* subObj = arg_getPtr(self);
  608. obj_GC(subObj);
  609. return;
  610. }
  611. #if 0
  612. if (argType_isObjectMethodActive(arg_getType(self))) {
  613. PikaObj* hostObj = methodArg_getHostObj(self);
  614. if (NULL != hostObj) {
  615. obj_GC(hostObj);
  616. }
  617. }
  618. #endif
  619. }
  620. Arg* arg_loadFile(Arg* self, char* filename) {
  621. size_t file_size = 0;
  622. FILE* input_file = pika_platform_fopen(filename, "rb");
  623. if (NULL == input_file) {
  624. return NULL;
  625. }
  626. // Move the file pointer to the end of the file
  627. pika_platform_fseek(input_file, 0, SEEK_END);
  628. // Get the current file pointer position which is the file size
  629. file_size = pika_platform_ftell(input_file);
  630. // Move the file pointer back to the beginning of the file
  631. pika_platform_fseek(input_file, 0, SEEK_SET);
  632. // Allocate buffer based on file size
  633. Arg* res = arg_newBytes(NULL, file_size);
  634. char* file_buff = (char*)arg_getBytes(res);
  635. if (file_buff == NULL) {
  636. pika_platform_fclose(input_file);
  637. return NULL;
  638. }
  639. pika_platform_memset(file_buff, 0, file_size);
  640. // Read the file into the buffer
  641. if (pika_platform_fread(file_buff, 1, file_size, input_file) != file_size) {
  642. // Handle read error or file size mismatch
  643. arg_deinit(res);
  644. pika_platform_fclose(input_file);
  645. return NULL;
  646. }
  647. // Clean up
  648. pika_platform_fclose(input_file);
  649. return res;
  650. }
  651. void arg_deinit(Arg* self) {
  652. pika_assert(NULL != self);
  653. /* deinit arg pointed heap */
  654. arg_deinitHeap(self);
  655. if (!arg_isSerialized(self)) {
  656. return;
  657. }
  658. /* free the ref */
  659. arg_freeContent(self);
  660. }
  661. pika_bool arg_isEqual(Arg* self, Arg* other) {
  662. if (NULL == self || NULL == other) {
  663. return pika_false;
  664. }
  665. if (arg_getType(self) != arg_getType(other)) {
  666. return pika_false;
  667. }
  668. if (arg_getType(self) == ARG_TYPE_OBJECT) {
  669. if (arg_getPtr(self) != arg_getPtr(other)) {
  670. return pika_false;
  671. }
  672. }
  673. if (arg_getType(self) == ARG_TYPE_STRING) {
  674. if (strEqu(arg_getStr(self), arg_getStr(other))) {
  675. return pika_true;
  676. }
  677. }
  678. if (0 != pika_platform_memcmp(arg_getContent(self), arg_getContent(other),
  679. arg_getContentSize(self))) {
  680. return pika_false;
  681. }
  682. return pika_true;
  683. }