lwp_pid.c 24 KB

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