lwp_pid.c 24 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028
  1. /*
  2. * Copyright (c) 2006-2021, RT-Thread Development Team
  3. *
  4. * SPDX-License-Identifier: Apache-2.0
  5. *
  6. * Change Logs:
  7. * Date Author Notes
  8. * 2019-10-16 zhangjun first version
  9. * 2021-02-20 lizhirui fix warning
  10. */
  11. #include <rthw.h>
  12. #include <rtthread.h>
  13. #include <dfs_posix.h>
  14. #include "lwp.h"
  15. #include "lwp_pid.h"
  16. #include "lwp_console.h"
  17. #ifdef RT_USING_USERSPACE
  18. #include "lwp_user_mm.h"
  19. #ifdef RT_USING_GDBSERVER
  20. #include <hw_breakpoint.h>
  21. #include <lwp_gdbserver.h>
  22. #endif
  23. #endif
  24. #define DBG_TAG "LWP_PID"
  25. #define DBG_LVL DBG_INFO
  26. #include <rtdbg.h>
  27. #define PID_MAX 10000
  28. #define PID_CT_ASSERT(name, x) \
  29. struct assert_##name {char ary[2 * (x) - 1];}
  30. PID_CT_ASSERT(pid_min_nr, RT_LWP_MAX_NR > 1);
  31. PID_CT_ASSERT(pid_max_nr, RT_LWP_MAX_NR < PID_MAX);
  32. static struct lwp_avl_struct lwp_pid_ary[RT_LWP_MAX_NR];
  33. static struct lwp_avl_struct *lwp_pid_free_head = RT_NULL;
  34. static int lwp_pid_ary_alloced = 0;
  35. static struct lwp_avl_struct *lwp_pid_root = RT_NULL;
  36. static pid_t current_pid = 0;
  37. struct lwp_avl_struct *lwp_get_pid_ary(void)
  38. {
  39. return lwp_pid_ary;
  40. }
  41. static pid_t lwp_pid_get(void)
  42. {
  43. rt_base_t level;
  44. struct lwp_avl_struct *p;
  45. pid_t pid = 0;
  46. level = rt_hw_interrupt_disable();
  47. p = lwp_pid_free_head;
  48. if (p)
  49. {
  50. lwp_pid_free_head = (struct lwp_avl_struct *)p->avl_right;
  51. }
  52. else if (lwp_pid_ary_alloced < RT_LWP_MAX_NR)
  53. {
  54. p = lwp_pid_ary + lwp_pid_ary_alloced;
  55. lwp_pid_ary_alloced++;
  56. }
  57. if (p)
  58. {
  59. int found_noused = 0;
  60. RT_ASSERT(p->data == RT_NULL);
  61. for (pid = current_pid + 1; pid < PID_MAX; pid++)
  62. {
  63. if (!lwp_avl_find(pid, lwp_pid_root))
  64. {
  65. found_noused = 1;
  66. break;
  67. }
  68. }
  69. if (!found_noused)
  70. {
  71. for (pid = 1; pid <= current_pid; pid++)
  72. {
  73. if (!lwp_avl_find(pid, lwp_pid_root))
  74. {
  75. found_noused = 1;
  76. break;
  77. }
  78. }
  79. }
  80. p->avl_key = pid;
  81. lwp_avl_insert(p, &lwp_pid_root);
  82. current_pid = pid;
  83. }
  84. rt_hw_interrupt_enable(level);
  85. return pid;
  86. }
  87. static void lwp_pid_put(pid_t pid)
  88. {
  89. rt_base_t level;
  90. struct lwp_avl_struct *p;
  91. level = rt_hw_interrupt_disable();
  92. p = lwp_avl_find(pid, lwp_pid_root);
  93. if (p)
  94. {
  95. p->data = RT_NULL;
  96. lwp_avl_remove(p, &lwp_pid_root);
  97. p->avl_right = lwp_pid_free_head;
  98. lwp_pid_free_head = p;
  99. }
  100. rt_hw_interrupt_enable(level);
  101. }
  102. static void lwp_pid_set_lwp(pid_t pid, struct rt_lwp *lwp)
  103. {
  104. rt_base_t level;
  105. struct lwp_avl_struct *p;
  106. level = rt_hw_interrupt_disable();
  107. p = lwp_avl_find(pid, lwp_pid_root);
  108. if (p)
  109. {
  110. p->data = lwp;
  111. }
  112. rt_hw_interrupt_enable(level);
  113. }
  114. int libc_stdio_get_console(void);
  115. static void __exit_files(struct rt_lwp *lwp)
  116. {
  117. int fd = lwp->fdt.maxfd - 1;
  118. while (fd >= 0)
  119. {
  120. struct dfs_fd *d;
  121. d = lwp->fdt.fds[fd];
  122. if (d)
  123. {
  124. dfs_file_close(d);
  125. fdt_fd_release(&lwp->fdt, fd);
  126. }
  127. fd--;
  128. }
  129. }
  130. void lwp_user_object_lock_init(struct rt_lwp *lwp)
  131. {
  132. rt_mutex_init(&lwp->object_mutex, "lwp_obj", RT_IPC_FLAG_PRIO);
  133. }
  134. void lwp_user_object_lock_destroy(struct rt_lwp *lwp)
  135. {
  136. rt_mutex_detach(&lwp->object_mutex);
  137. }
  138. void lwp_user_object_lock(struct rt_lwp *lwp)
  139. {
  140. if (lwp)
  141. {
  142. rt_mutex_take(&lwp->object_mutex, RT_WAITING_FOREVER);
  143. }
  144. else
  145. {
  146. RT_ASSERT(0);
  147. }
  148. }
  149. void lwp_user_object_unlock(struct rt_lwp *lwp)
  150. {
  151. if (lwp)
  152. {
  153. rt_mutex_release(&lwp->object_mutex);
  154. }
  155. else
  156. {
  157. RT_ASSERT(0);
  158. }
  159. }
  160. int lwp_user_object_add(struct rt_lwp *lwp, rt_object_t object)
  161. {
  162. int ret = -1;
  163. if (lwp && object)
  164. {
  165. lwp_user_object_lock(lwp);
  166. if (!lwp_avl_find((avl_key_t)object, lwp->object_root))
  167. {
  168. struct lwp_avl_struct *node;
  169. node = (struct lwp_avl_struct *)rt_malloc(sizeof(struct lwp_avl_struct));
  170. if (node)
  171. {
  172. rt_base_t level;
  173. level = rt_hw_interrupt_disable();
  174. object->lwp_ref_count++;
  175. rt_hw_interrupt_enable(level);
  176. node->avl_key = (avl_key_t)object;
  177. lwp_avl_insert(node, &lwp->object_root);
  178. ret = 0;
  179. }
  180. }
  181. lwp_user_object_unlock(lwp);
  182. }
  183. return ret;
  184. }
  185. static rt_err_t _object_node_delete(struct rt_lwp *lwp, struct lwp_avl_struct *node)
  186. {
  187. rt_err_t ret = -1;
  188. rt_object_t object;
  189. if (!lwp || !node)
  190. {
  191. return ret;
  192. }
  193. object = (rt_object_t)node->avl_key;
  194. object->lwp_ref_count--;
  195. if (object->lwp_ref_count == 0)
  196. {
  197. /* remove from kernel object list */
  198. switch (object->type)
  199. {
  200. case RT_Object_Class_Semaphore:
  201. ret = rt_sem_delete((rt_sem_t)object);
  202. break;
  203. case RT_Object_Class_Mutex:
  204. ret = rt_mutex_delete((rt_mutex_t)object);
  205. break;
  206. case RT_Object_Class_Event:
  207. ret = rt_event_delete((rt_event_t)object);
  208. break;
  209. case RT_Object_Class_MailBox:
  210. ret = rt_mb_delete((rt_mailbox_t)object);
  211. break;
  212. case RT_Object_Class_MessageQueue:
  213. ret = rt_mq_delete((rt_mq_t)object);
  214. break;
  215. case RT_Object_Class_Timer:
  216. ret = rt_timer_delete((rt_timer_t)object);
  217. break;
  218. case RT_Object_Class_Custom:
  219. ret = rt_custom_object_destroy(object);
  220. break;
  221. default:
  222. LOG_E("input object type(%d) error", object->type);
  223. break;
  224. }
  225. }
  226. else
  227. {
  228. ret = 0;
  229. }
  230. lwp_avl_remove(node, &lwp->object_root);
  231. rt_free(node);
  232. return ret;
  233. }
  234. rt_err_t lwp_user_object_delete(struct rt_lwp *lwp, rt_object_t object)
  235. {
  236. rt_err_t ret = -1;
  237. if (lwp && object)
  238. {
  239. struct lwp_avl_struct *node;
  240. lwp_user_object_lock(lwp);
  241. node = lwp_avl_find((avl_key_t)object, lwp->object_root);
  242. ret = _object_node_delete(lwp, node);
  243. lwp_user_object_unlock(lwp);
  244. }
  245. return ret;
  246. }
  247. void lwp_user_object_clear(struct rt_lwp *lwp)
  248. {
  249. struct lwp_avl_struct *node;
  250. lwp_user_object_lock(lwp);
  251. while ((node = lwp_map_find_first(lwp->object_root)) != RT_NULL)
  252. {
  253. _object_node_delete(lwp, node);
  254. }
  255. lwp_user_object_unlock(lwp);
  256. }
  257. static int _object_dup(struct lwp_avl_struct *node, void *arg)
  258. {
  259. rt_object_t object;
  260. struct rt_lwp *dst_lwp = (struct rt_lwp *)arg;
  261. object = (rt_object_t)node->avl_key;
  262. lwp_user_object_add(dst_lwp, object);
  263. return 0;
  264. }
  265. void lwp_user_object_dup(struct rt_lwp *dst_lwp, struct rt_lwp *src_lwp)
  266. {
  267. lwp_user_object_lock(src_lwp);
  268. lwp_avl_traversal(src_lwp->object_root, _object_dup, dst_lwp);
  269. lwp_user_object_unlock(src_lwp);
  270. }
  271. struct rt_lwp* lwp_new(void)
  272. {
  273. pid_t pid;
  274. rt_base_t level;
  275. struct rt_lwp* lwp = RT_NULL;
  276. level = rt_hw_interrupt_disable();
  277. pid = lwp_pid_get();
  278. if (pid == 0)
  279. {
  280. LOG_E("pid slot fulled!\n");
  281. goto out;
  282. }
  283. lwp = (struct rt_lwp *)rt_malloc(sizeof(struct rt_lwp));
  284. if (lwp == RT_NULL)
  285. {
  286. LOG_E("no memory for lwp struct!\n");
  287. goto out;
  288. }
  289. rt_memset(lwp, 0, sizeof(*lwp));
  290. rt_list_init(&lwp->wait_list);
  291. lwp->pid = pid;
  292. lwp_pid_set_lwp(pid, lwp);
  293. rt_list_init(&lwp->t_grp);
  294. lwp_user_object_lock_init(lwp);
  295. lwp->address_search_head = RT_NULL;
  296. rt_wqueue_init(&lwp->wait_queue);
  297. lwp->ref = 1;
  298. out:
  299. rt_hw_interrupt_enable(level);
  300. return lwp;
  301. }
  302. void lwp_free(struct rt_lwp* lwp)
  303. {
  304. rt_base_t level;
  305. if (lwp == RT_NULL)
  306. {
  307. return;
  308. }
  309. LOG_D("lwp free: %p\n", lwp);
  310. level = rt_hw_interrupt_disable();
  311. lwp->finish = 1;
  312. if (lwp->args != RT_NULL)
  313. {
  314. #ifndef ARCH_MM_MMU
  315. lwp->args_length = RT_NULL;
  316. #ifndef ARCH_MM_MPU
  317. rt_free(lwp->args);
  318. #endif /* not defined ARCH_MM_MPU */
  319. #endif /* ARCH_MM_MMU */
  320. lwp->args = RT_NULL;
  321. }
  322. if (lwp->fdt.fds != RT_NULL)
  323. {
  324. /* auto clean fds */
  325. __exit_files(lwp);
  326. rt_free(lwp->fdt.fds);
  327. lwp->fdt.fds = RT_NULL;
  328. }
  329. lwp_user_object_clear(lwp);
  330. lwp_user_object_lock_destroy(lwp);
  331. /* free data section */
  332. if (lwp->data_entry != RT_NULL)
  333. {
  334. #ifdef ARCH_MM_MMU
  335. rt_free_align(lwp->data_entry);
  336. #else
  337. #ifdef ARCH_MM_MPU
  338. rt_lwp_umap_user(lwp, lwp->text_entry, 0);
  339. rt_lwp_free_user(lwp, lwp->data_entry, lwp->data_size);
  340. #else
  341. rt_free_align(lwp->data_entry);
  342. #endif /* ARCH_MM_MPU */
  343. #endif /* ARCH_MM_MMU */
  344. lwp->data_entry = RT_NULL;
  345. }
  346. /* free text section */
  347. if (lwp->lwp_type == LWP_TYPE_DYN_ADDR)
  348. {
  349. if (lwp->text_entry)
  350. {
  351. LOG_D("lwp text free: %p", lwp->text_entry);
  352. #ifndef ARCH_MM_MMU
  353. rt_free((void*)lwp->text_entry);
  354. #endif /* not defined ARCH_MM_MMU */
  355. lwp->text_entry = RT_NULL;
  356. }
  357. }
  358. #ifdef RT_USING_USERSPACE
  359. lwp_unmap_user_space(lwp);
  360. #endif
  361. /* for children */
  362. while (lwp->first_child)
  363. {
  364. struct rt_lwp *child;
  365. child = lwp->first_child;
  366. lwp->first_child = child->sibling;
  367. if (child->finish)
  368. {
  369. lwp_pid_put(lwp_to_pid(child));
  370. rt_free(child);
  371. }
  372. else
  373. {
  374. child->sibling = RT_NULL;
  375. child->parent = RT_NULL;
  376. }
  377. }
  378. /* for parent */
  379. {
  380. struct rt_lwp *console_lwp;
  381. console_lwp = rt_console_get_foreground();
  382. if (lwp == console_lwp)
  383. {
  384. rt_console_set_foreground(lwp->parent);
  385. }
  386. if (lwp->parent)
  387. {
  388. struct rt_thread *thread;
  389. if (!rt_list_isempty(&lwp->wait_list))
  390. {
  391. thread = rt_list_entry(lwp->wait_list.next, struct rt_thread, tlist);
  392. thread->error = RT_EOK;
  393. thread->msg_ret = (void*)(rt_size_t)lwp->lwp_ret;
  394. rt_thread_resume(thread);
  395. rt_hw_interrupt_enable(level);
  396. return;
  397. }
  398. else
  399. {
  400. struct rt_lwp **it = &lwp->parent->first_child;
  401. while (*it != lwp)
  402. {
  403. it = &(*it)->sibling;
  404. }
  405. *it = lwp->sibling;
  406. }
  407. }
  408. lwp_pid_put(lwp_to_pid(lwp));
  409. rt_free(lwp);
  410. }
  411. rt_hw_interrupt_enable(level);
  412. }
  413. void lwp_ref_inc(struct rt_lwp *lwp)
  414. {
  415. rt_base_t level;
  416. level = rt_hw_interrupt_disable();
  417. lwp->ref++;
  418. rt_hw_interrupt_enable(level);
  419. }
  420. void lwp_ref_dec(struct rt_lwp *lwp)
  421. {
  422. rt_base_t level;
  423. int ref;
  424. level = rt_hw_interrupt_disable();
  425. if (lwp->ref)
  426. {
  427. lwp->ref--;
  428. ref = lwp->ref;
  429. if (!ref)
  430. {
  431. #ifdef RT_USING_GDBSERVER
  432. struct rt_channel_msg msg;
  433. if (lwp->debug)
  434. {
  435. memset(&msg, 0, sizeof msg);
  436. rt_raw_channel_send(gdb_get_server_channel(), &msg);
  437. }
  438. #endif /* RT_USING_GDBSERVER */
  439. #ifndef ARCH_MM_MMU
  440. #ifdef RT_LWP_USING_SHM
  441. lwp_shm_lwp_free(lwp);
  442. #endif /* RT_LWP_USING_SHM */
  443. #endif /* not defined ARCH_MM_MMU */
  444. lwp_free(lwp);
  445. }
  446. }
  447. rt_hw_interrupt_enable(level);
  448. }
  449. struct rt_lwp* lwp_from_pid(pid_t pid)
  450. {
  451. rt_base_t level;
  452. struct lwp_avl_struct *p;
  453. struct rt_lwp *lwp = RT_NULL;
  454. level = rt_hw_interrupt_disable();
  455. p = lwp_avl_find(pid, lwp_pid_root);
  456. if (p)
  457. {
  458. lwp = (struct rt_lwp *)p->data;
  459. }
  460. rt_hw_interrupt_enable(level);
  461. return lwp;
  462. }
  463. pid_t lwp_to_pid(struct rt_lwp* lwp)
  464. {
  465. if (!lwp)
  466. {
  467. return 0;
  468. }
  469. return lwp->pid;
  470. }
  471. char* lwp_pid2name(int32_t pid)
  472. {
  473. struct rt_lwp *lwp;
  474. char* process_name = RT_NULL;
  475. lwp = lwp_from_pid(pid);
  476. if (lwp)
  477. {
  478. process_name = strrchr(lwp->cmd, '/');
  479. process_name = process_name? process_name + 1: lwp->cmd;
  480. }
  481. return process_name;
  482. }
  483. pid_t lwp_name2pid(const char *name)
  484. {
  485. int idx;
  486. pid_t pid = 0;
  487. rt_thread_t main_thread;
  488. char* process_name = RT_NULL;
  489. rt_base_t level;
  490. level = rt_hw_interrupt_disable();
  491. for (idx = 0; idx < RT_LWP_MAX_NR; idx++)
  492. {
  493. /* 0 is reserved */
  494. struct rt_lwp *lwp = (struct rt_lwp *)lwp_pid_ary[idx].data;
  495. if (lwp)
  496. {
  497. process_name = strrchr(lwp->cmd, '/');
  498. process_name = process_name? process_name + 1: lwp->cmd;
  499. if (!rt_strncmp(name, process_name, RT_NAME_MAX))
  500. {
  501. main_thread = rt_list_entry(lwp->t_grp.prev, struct rt_thread, sibling);
  502. if (!(main_thread->stat & RT_THREAD_CLOSE))
  503. {
  504. pid = lwp->pid;
  505. }
  506. }
  507. }
  508. }
  509. rt_hw_interrupt_enable(level);
  510. return pid;
  511. }
  512. int lwp_getpid(void)
  513. {
  514. return ((struct rt_lwp *)rt_thread_self()->lwp)->pid;
  515. }
  516. pid_t waitpid(pid_t pid, int *status, int options)
  517. {
  518. pid_t ret = -1;
  519. rt_base_t level;
  520. struct rt_thread *thread;
  521. struct rt_lwp *lwp;
  522. struct rt_lwp *lwp_self;
  523. level = rt_hw_interrupt_disable();
  524. lwp = lwp_from_pid(pid);
  525. if (!lwp)
  526. {
  527. goto quit;
  528. }
  529. lwp_self = (struct rt_lwp *)rt_thread_self()->lwp;
  530. if (!lwp_self)
  531. {
  532. goto quit;
  533. }
  534. if (lwp->parent != lwp_self)
  535. {
  536. goto quit;
  537. }
  538. if (lwp->finish)
  539. {
  540. ret = pid;
  541. }
  542. else
  543. {
  544. if (!rt_list_isempty(&lwp->wait_list))
  545. {
  546. goto quit;
  547. }
  548. thread = rt_thread_self();
  549. rt_thread_suspend_with_flag(thread, RT_UNINTERRUPTIBLE);
  550. rt_list_insert_before(&lwp->wait_list, &(thread->tlist));
  551. rt_schedule();
  552. if (thread->error == RT_EOK)
  553. {
  554. ret = pid;
  555. }
  556. }
  557. if (ret != -1)
  558. {
  559. struct rt_lwp **lwp_node;
  560. *status = lwp->lwp_ret;
  561. lwp_node = &lwp_self->first_child;
  562. while (*lwp_node != lwp)
  563. {
  564. RT_ASSERT(*lwp_node != RT_NULL);
  565. lwp_node = &(*lwp_node)->sibling;
  566. }
  567. (*lwp_node) = lwp->sibling;
  568. lwp_pid_put(pid);
  569. rt_free(lwp);
  570. }
  571. quit:
  572. rt_hw_interrupt_enable(level);
  573. return ret;
  574. }
  575. #ifdef RT_USING_FINSH
  576. /* copy from components/finsh/cmd.c */
  577. static void object_split(int len)
  578. {
  579. while (len--)
  580. {
  581. rt_kprintf("-");
  582. }
  583. }
  584. static void print_thread_info(struct rt_thread* thread, int maxlen)
  585. {
  586. rt_uint8_t *ptr;
  587. rt_uint8_t stat;
  588. #ifdef RT_USING_SMP
  589. if (thread->oncpu != RT_CPU_DETACHED)
  590. rt_kprintf("%-*.*s %3d %3d ", maxlen, RT_NAME_MAX, thread->name, thread->oncpu, thread->current_priority);
  591. else
  592. rt_kprintf("%-*.*s N/A %3d ", maxlen, RT_NAME_MAX, thread->name, thread->current_priority);
  593. #else
  594. rt_kprintf("%-*.*s %3d ", maxlen, RT_NAME_MAX, thread->name, thread->current_priority);
  595. #endif /*RT_USING_SMP*/
  596. stat = (thread->stat & RT_THREAD_STAT_MASK);
  597. if (stat == RT_THREAD_READY) rt_kprintf(" ready ");
  598. else if ((stat & RT_THREAD_SUSPEND_MASK) == RT_THREAD_SUSPEND_MASK) rt_kprintf(" suspend");
  599. else if (stat == RT_THREAD_INIT) rt_kprintf(" init ");
  600. else if (stat == RT_THREAD_CLOSE) rt_kprintf(" close ");
  601. else if (stat == RT_THREAD_RUNNING) rt_kprintf(" running");
  602. #if defined(ARCH_CPU_STACK_GROWS_UPWARD)
  603. ptr = (rt_uint8_t *)thread->stack_addr + thread->stack_size;
  604. while (*ptr == '#')ptr--;
  605. rt_kprintf(" 0x%08x 0x%08x %02d%% 0x%08x %03d\n",
  606. ((rt_uint32_t)thread->sp - (rt_uint32_t)thread->stack_addr),
  607. thread->stack_size,
  608. ((rt_uint32_t)ptr - (rt_uint32_t)thread->stack_addr) * 100 / thread->stack_size,
  609. thread->remaining_tick,
  610. thread->error);
  611. #else
  612. ptr = (rt_uint8_t *)thread->stack_addr;
  613. while (*ptr == '#')ptr++;
  614. rt_kprintf(" 0x%08x 0x%08x %02d%% 0x%08x %03d\n",
  615. (thread->stack_size + (rt_uint32_t)(rt_size_t)thread->stack_addr - (rt_uint32_t)(rt_size_t)thread->sp),
  616. thread->stack_size,
  617. (thread->stack_size + (rt_uint32_t)(rt_size_t)thread->stack_addr - (rt_uint32_t)(rt_size_t)ptr) * 100
  618. / thread->stack_size,
  619. thread->remaining_tick,
  620. thread->error);
  621. #endif
  622. }
  623. long list_process(void)
  624. {
  625. int index;
  626. int maxlen;
  627. rt_ubase_t level;
  628. struct rt_thread *thread;
  629. struct rt_list_node *node, *list;
  630. const char *item_title = "thread";
  631. int count = 0;
  632. struct rt_thread **threads;
  633. maxlen = RT_NAME_MAX;
  634. #ifdef RT_USING_SMP
  635. rt_kprintf("%-*.s %-*.s %-*.s cpu pri status sp stack size max used left tick error\n", 4, "PID", maxlen, "CMD", maxlen, item_title);
  636. object_split(4);rt_kprintf(" ");object_split(maxlen);rt_kprintf(" ");object_split(maxlen);rt_kprintf(" ");
  637. rt_kprintf( "--- --- ------- ---------- ---------- ------ ---------- ---\n");
  638. #else
  639. rt_kprintf("%-*.s %-*.s %-*.s pri status sp stack size max used left tick error\n", 4, "PID", maxlen, "CMD", maxlen, item_title);
  640. object_split(4);rt_kprintf(" ");object_split(maxlen);rt_kprintf(" ");object_split(maxlen);rt_kprintf(" ");
  641. rt_kprintf( "--- ------- ---------- ---------- ------ ---------- ---\n");
  642. #endif /*RT_USING_SMP*/
  643. count = rt_object_get_length(RT_Object_Class_Thread);
  644. if (count > 0)
  645. {
  646. /* get thread pointers */
  647. threads = (struct rt_thread **)rt_calloc(count, sizeof(struct rt_thread *));
  648. if (threads)
  649. {
  650. index = rt_object_get_pointers(RT_Object_Class_Thread, (rt_object_t *)threads, count);
  651. if (index > 0)
  652. {
  653. for (index = 0; index <count; index++)
  654. {
  655. struct rt_thread th;
  656. thread = threads[index];
  657. level = rt_hw_interrupt_disable();
  658. if ((thread->type & ~RT_Object_Class_Static) != RT_Object_Class_Thread)
  659. {
  660. rt_hw_interrupt_enable(level);
  661. continue;
  662. }
  663. rt_memcpy(&th, thread, sizeof(struct rt_thread));
  664. rt_hw_interrupt_enable(level);
  665. if (th.lwp == RT_NULL)
  666. {
  667. rt_kprintf(" %-*.*s ", maxlen, RT_NAME_MAX, "kernel");
  668. print_thread_info(&th, maxlen);
  669. }
  670. }
  671. }
  672. rt_free(threads);
  673. }
  674. }
  675. for (index = 0; index < RT_LWP_MAX_NR; index++)
  676. {
  677. struct rt_lwp *lwp = (struct rt_lwp *)lwp_pid_ary[index].data;
  678. if (lwp)
  679. {
  680. list = &lwp->t_grp;
  681. for (node = list->next; node != list; node = node->next)
  682. {
  683. thread = rt_list_entry(node, struct rt_thread, sibling);
  684. rt_kprintf("%4d %-*.*s ", lwp_to_pid(lwp), maxlen, RT_NAME_MAX, lwp->cmd);
  685. print_thread_info(thread, maxlen);
  686. }
  687. }
  688. }
  689. return 0;
  690. }
  691. MSH_CMD_EXPORT(list_process, list process);
  692. static void cmd_kill(int argc, char** argv)
  693. {
  694. int pid;
  695. int sig = 0;
  696. if (argc < 2)
  697. {
  698. rt_kprintf("kill pid or kill pid -s signal\n");
  699. return;
  700. }
  701. pid = atoi(argv[1]);
  702. if (argc >= 4)
  703. {
  704. if (argv[2][0] == '-' && argv[2][1] == 's')
  705. {
  706. sig = atoi(argv[3]);
  707. }
  708. }
  709. lwp_kill(pid, sig);
  710. }
  711. MSH_CMD_EXPORT_ALIAS(cmd_kill, kill, send a signal to a process);
  712. static void cmd_killall(int argc, char** argv)
  713. {
  714. int pid;
  715. if (argc < 2)
  716. {
  717. rt_kprintf("killall processes_name\n");
  718. return;
  719. }
  720. while((pid = lwp_name2pid(argv[1])) >= 0)
  721. {
  722. lwp_kill(pid, 0);
  723. rt_thread_mdelay(100);
  724. }
  725. }
  726. MSH_CMD_EXPORT_ALIAS(cmd_killall, killall, kill processes by name);
  727. #endif
  728. int lwp_check_exit_request(void)
  729. {
  730. rt_thread_t thread = rt_thread_self();
  731. if (!thread->lwp)
  732. {
  733. return 0;
  734. }
  735. if (thread->exit_request == LWP_EXIT_REQUEST_TRIGGERED)
  736. {
  737. thread->exit_request = LWP_EXIT_REQUEST_IN_PROCESS;
  738. return 1;
  739. }
  740. return 0;
  741. }
  742. static int found_thread(struct rt_lwp* lwp, rt_thread_t thread)
  743. {
  744. int found = 0;
  745. rt_base_t level;
  746. rt_list_t *list;
  747. level = rt_hw_interrupt_disable();
  748. list = lwp->t_grp.next;
  749. while (list != &lwp->t_grp)
  750. {
  751. rt_thread_t iter_thread;
  752. iter_thread = rt_list_entry(list, struct rt_thread, sibling);
  753. if (thread == iter_thread)
  754. {
  755. found = 1;
  756. break;
  757. }
  758. list = list->next;
  759. }
  760. rt_hw_interrupt_enable(level);
  761. return found;
  762. }
  763. void lwp_request_thread_exit(rt_thread_t thread_to_exit)
  764. {
  765. rt_thread_t main_thread;
  766. rt_base_t level;
  767. rt_list_t *list;
  768. struct rt_lwp *lwp;
  769. lwp = lwp_self();
  770. if ((!thread_to_exit) || (!lwp))
  771. {
  772. return;
  773. }
  774. level = rt_hw_interrupt_disable();
  775. main_thread = rt_list_entry(lwp->t_grp.prev, struct rt_thread, sibling);
  776. if (thread_to_exit == main_thread)
  777. {
  778. goto finish;
  779. }
  780. if ((struct rt_lwp *)thread_to_exit->lwp != lwp)
  781. {
  782. goto finish;
  783. }
  784. for (list = lwp->t_grp.next; list != &lwp->t_grp; list = list->next)
  785. {
  786. rt_thread_t thread;
  787. thread = rt_list_entry(list, struct rt_thread, sibling);
  788. if (thread != thread_to_exit)
  789. {
  790. continue;
  791. }
  792. if (thread->exit_request == LWP_EXIT_REQUEST_NONE)
  793. {
  794. thread->exit_request = LWP_EXIT_REQUEST_TRIGGERED;
  795. }
  796. if ((thread->stat & RT_THREAD_SUSPEND_MASK) == RT_THREAD_SUSPEND_MASK)
  797. {
  798. thread->error = RT_EINTR;
  799. rt_hw_dsb();
  800. rt_thread_wakeup(thread);
  801. }
  802. break;
  803. }
  804. while (found_thread(lwp, thread_to_exit))
  805. {
  806. rt_thread_mdelay(10);
  807. }
  808. finish:
  809. rt_hw_interrupt_enable(level);
  810. return;
  811. }
  812. void lwp_terminate(struct rt_lwp *lwp)
  813. {
  814. rt_base_t level;
  815. rt_list_t *list;
  816. if (!lwp)
  817. {
  818. /* kernel thread not support */
  819. return;
  820. }
  821. level = rt_hw_interrupt_disable();
  822. for (list = lwp->t_grp.next; list != &lwp->t_grp; list = list->next)
  823. {
  824. rt_thread_t thread;
  825. thread = rt_list_entry(list, struct rt_thread, sibling);
  826. if (thread->exit_request == LWP_EXIT_REQUEST_NONE)
  827. {
  828. thread->exit_request = LWP_EXIT_REQUEST_TRIGGERED;
  829. }
  830. if ((thread->stat & RT_THREAD_SUSPEND_MASK) == RT_THREAD_SUSPEND_MASK)
  831. {
  832. thread->error = RT_EINTR;
  833. rt_hw_dsb();
  834. rt_thread_wakeup(thread);
  835. }
  836. }
  837. rt_hw_interrupt_enable(level);
  838. }
  839. void lwp_wait_subthread_exit(void)
  840. {
  841. rt_base_t level;
  842. struct rt_lwp *lwp;
  843. rt_thread_t thread;
  844. rt_thread_t main_thread;
  845. lwp = lwp_self();
  846. if (!lwp)
  847. {
  848. return;
  849. }
  850. thread = rt_thread_self();
  851. main_thread = rt_list_entry(lwp->t_grp.prev, struct rt_thread, sibling);
  852. if (thread != main_thread)
  853. {
  854. return;
  855. }
  856. while (1)
  857. {
  858. int subthread_is_terminated;
  859. level = rt_hw_interrupt_disable();
  860. subthread_is_terminated = (int)(thread->sibling.prev == &lwp->t_grp);
  861. if (!subthread_is_terminated)
  862. {
  863. rt_thread_t sub_thread;
  864. rt_list_t *list;
  865. int all_subthread_in_init = 1;
  866. /* check all subthread is in init state */
  867. for (list = thread->sibling.prev; list != &lwp->t_grp; list = list->prev)
  868. {
  869. sub_thread = rt_list_entry(list, struct rt_thread, sibling);
  870. if ((sub_thread->stat & RT_THREAD_STAT_MASK) != RT_THREAD_INIT)
  871. {
  872. all_subthread_in_init = 0;
  873. break;
  874. }
  875. }
  876. if (all_subthread_in_init)
  877. {
  878. /* delete all subthread */
  879. while ((list = thread->sibling.prev) != &lwp->t_grp)
  880. {
  881. sub_thread = rt_list_entry(list, struct rt_thread, sibling);
  882. rt_list_remove(&sub_thread->sibling);
  883. rt_thread_delete(sub_thread);
  884. }
  885. subthread_is_terminated = 1;
  886. }
  887. }
  888. rt_hw_interrupt_enable(level);
  889. if (subthread_is_terminated)
  890. {
  891. break;
  892. }
  893. rt_thread_mdelay(10);
  894. }
  895. }