lwp_pid.c 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014
  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 RT_USING_USERSPACE
  315. rt_free(lwp->args);
  316. #endif
  317. lwp->args = RT_NULL;
  318. }
  319. if (lwp->fdt.fds != RT_NULL)
  320. {
  321. /* auto clean fds */
  322. __exit_files(lwp);
  323. rt_free(lwp->fdt.fds);
  324. lwp->fdt.fds = RT_NULL;
  325. }
  326. lwp_user_object_clear(lwp);
  327. lwp_user_object_lock_destroy(lwp);
  328. /* free data section */
  329. if (lwp->data_entry != RT_NULL)
  330. {
  331. rt_free_align(lwp->data_entry);
  332. lwp->data_entry = RT_NULL;
  333. }
  334. /* free text section */
  335. if (lwp->lwp_type == LWP_TYPE_DYN_ADDR)
  336. {
  337. if (lwp->text_entry)
  338. {
  339. LOG_D("lwp text free: %p", lwp->text_entry);
  340. #ifndef RT_USING_USERSPACE
  341. #ifdef RT_USING_CACHE
  342. rt_free_align((void*)lwp->load_off);
  343. #else
  344. rt_free((void*)lwp->load_off);
  345. #endif
  346. #endif
  347. lwp->text_entry = RT_NULL;
  348. }
  349. }
  350. #ifdef RT_USING_USERSPACE
  351. lwp_unmap_user_space(lwp);
  352. #endif
  353. /* for children */
  354. while (lwp->first_child)
  355. {
  356. struct rt_lwp *child;
  357. child = lwp->first_child;
  358. lwp->first_child = child->sibling;
  359. if (child->finish)
  360. {
  361. lwp_pid_put(lwp_to_pid(child));
  362. rt_free(child);
  363. }
  364. else
  365. {
  366. child->sibling = RT_NULL;
  367. child->parent = RT_NULL;
  368. }
  369. }
  370. /* for parent */
  371. {
  372. struct rt_lwp *console_lwp;
  373. console_lwp = rt_console_get_foreground();
  374. if (lwp == console_lwp)
  375. {
  376. rt_console_set_foreground(lwp->parent);
  377. }
  378. if (lwp->parent)
  379. {
  380. struct rt_thread *thread;
  381. if (!rt_list_isempty(&lwp->wait_list))
  382. {
  383. thread = rt_list_entry(lwp->wait_list.next, struct rt_thread, tlist);
  384. thread->error = RT_EOK;
  385. thread->msg_ret = (void*)(rt_size_t)lwp->lwp_ret;
  386. rt_thread_resume(thread);
  387. rt_hw_interrupt_enable(level);
  388. return;
  389. }
  390. else
  391. {
  392. struct rt_lwp **it = &lwp->parent->first_child;
  393. while (*it != lwp)
  394. {
  395. it = &(*it)->sibling;
  396. }
  397. *it = lwp->sibling;
  398. }
  399. }
  400. lwp_pid_put(lwp_to_pid(lwp));
  401. rt_free(lwp);
  402. }
  403. rt_hw_interrupt_enable(level);
  404. }
  405. void lwp_ref_inc(struct rt_lwp *lwp)
  406. {
  407. rt_base_t level;
  408. level = rt_hw_interrupt_disable();
  409. lwp->ref++;
  410. rt_hw_interrupt_enable(level);
  411. }
  412. void lwp_ref_dec(struct rt_lwp *lwp)
  413. {
  414. rt_base_t level;
  415. int ref;
  416. level = rt_hw_interrupt_disable();
  417. if (lwp->ref)
  418. {
  419. lwp->ref--;
  420. ref = lwp->ref;
  421. if (!ref)
  422. {
  423. #ifdef RT_USING_GDBSERVER
  424. struct rt_channel_msg msg;
  425. if (lwp->debug)
  426. {
  427. memset(&msg, 0, sizeof msg);
  428. rt_raw_channel_send(gdb_get_server_channel(), &msg);
  429. }
  430. #endif
  431. lwp_free(lwp);
  432. }
  433. }
  434. rt_hw_interrupt_enable(level);
  435. }
  436. struct rt_lwp* lwp_from_pid(pid_t pid)
  437. {
  438. rt_base_t level;
  439. struct lwp_avl_struct *p;
  440. struct rt_lwp *lwp = RT_NULL;
  441. level = rt_hw_interrupt_disable();
  442. p = lwp_avl_find(pid, lwp_pid_root);
  443. if (p)
  444. {
  445. lwp = (struct rt_lwp *)p->data;
  446. }
  447. rt_hw_interrupt_enable(level);
  448. return lwp;
  449. }
  450. pid_t lwp_to_pid(struct rt_lwp* lwp)
  451. {
  452. if (!lwp)
  453. {
  454. return 0;
  455. }
  456. return lwp->pid;
  457. }
  458. char* lwp_pid2name(int32_t pid)
  459. {
  460. struct rt_lwp *lwp;
  461. char* process_name = RT_NULL;
  462. lwp = lwp_from_pid(pid);
  463. if (lwp)
  464. {
  465. process_name = strrchr(lwp->cmd, '/');
  466. process_name = process_name? process_name + 1: lwp->cmd;
  467. }
  468. return process_name;
  469. }
  470. pid_t lwp_name2pid(const char *name)
  471. {
  472. int idx;
  473. pid_t pid = 0;
  474. rt_thread_t main_thread;
  475. char* process_name = RT_NULL;
  476. rt_base_t level;
  477. level = rt_hw_interrupt_disable();
  478. for (idx = 0; idx < RT_LWP_MAX_NR; idx++)
  479. {
  480. /* 0 is reserved */
  481. struct rt_lwp *lwp = (struct rt_lwp *)lwp_pid_ary[idx].data;
  482. if (lwp)
  483. {
  484. process_name = strrchr(lwp->cmd, '/');
  485. process_name = process_name? process_name + 1: lwp->cmd;
  486. if (!rt_strncmp(name, process_name, RT_NAME_MAX))
  487. {
  488. main_thread = rt_list_entry(lwp->t_grp.prev, struct rt_thread, sibling);
  489. if (!(main_thread->stat & RT_THREAD_CLOSE))
  490. {
  491. pid = lwp->pid;
  492. }
  493. }
  494. }
  495. }
  496. rt_hw_interrupt_enable(level);
  497. return pid;
  498. }
  499. int lwp_getpid(void)
  500. {
  501. return ((struct rt_lwp *)rt_thread_self()->lwp)->pid;
  502. }
  503. pid_t waitpid(pid_t pid, int *status, int options)
  504. {
  505. pid_t ret = -1;
  506. rt_base_t level;
  507. struct rt_thread *thread;
  508. struct rt_lwp *lwp;
  509. struct rt_lwp *lwp_self;
  510. level = rt_hw_interrupt_disable();
  511. lwp = lwp_from_pid(pid);
  512. if (!lwp)
  513. {
  514. goto quit;
  515. }
  516. lwp_self = (struct rt_lwp *)rt_thread_self()->lwp;
  517. if (!lwp_self)
  518. {
  519. goto quit;
  520. }
  521. if (lwp->parent != lwp_self)
  522. {
  523. goto quit;
  524. }
  525. if (lwp->finish)
  526. {
  527. ret = pid;
  528. }
  529. else
  530. {
  531. if (!rt_list_isempty(&lwp->wait_list))
  532. {
  533. goto quit;
  534. }
  535. thread = rt_thread_self();
  536. rt_thread_suspend_with_flag(thread, RT_UNINTERRUPTIBLE);
  537. rt_list_insert_before(&lwp->wait_list, &(thread->tlist));
  538. rt_schedule();
  539. if (thread->error == RT_EOK)
  540. {
  541. ret = pid;
  542. }
  543. }
  544. if (ret != -1)
  545. {
  546. struct rt_lwp **lwp_node;
  547. *status = lwp->lwp_ret;
  548. lwp_node = &lwp_self->first_child;
  549. while (*lwp_node != lwp)
  550. {
  551. RT_ASSERT(*lwp_node != RT_NULL);
  552. lwp_node = &(*lwp_node)->sibling;
  553. }
  554. (*lwp_node) = lwp->sibling;
  555. lwp_pid_put(pid);
  556. rt_free(lwp);
  557. }
  558. quit:
  559. rt_hw_interrupt_enable(level);
  560. return ret;
  561. }
  562. #ifdef RT_USING_FINSH
  563. /* copy from components/finsh/cmd.c */
  564. static void object_split(int len)
  565. {
  566. while (len--)
  567. {
  568. rt_kprintf("-");
  569. }
  570. }
  571. static void print_thread_info(struct rt_thread* thread, int maxlen)
  572. {
  573. rt_uint8_t *ptr;
  574. rt_uint8_t stat;
  575. #ifdef RT_USING_SMP
  576. if (thread->oncpu != RT_CPU_DETACHED)
  577. rt_kprintf("%-*.*s %3d %3d ", maxlen, RT_NAME_MAX, thread->name, thread->oncpu, thread->current_priority);
  578. else
  579. rt_kprintf("%-*.*s N/A %3d ", maxlen, RT_NAME_MAX, thread->name, thread->current_priority);
  580. #else
  581. rt_kprintf("%-*.*s %3d ", maxlen, RT_NAME_MAX, thread->name, thread->current_priority);
  582. #endif /*RT_USING_SMP*/
  583. stat = (thread->stat & RT_THREAD_STAT_MASK);
  584. if (stat == RT_THREAD_READY) rt_kprintf(" ready ");
  585. else if ((stat & RT_THREAD_SUSPEND_MASK) == RT_THREAD_SUSPEND_MASK) rt_kprintf(" suspend");
  586. else if (stat == RT_THREAD_INIT) rt_kprintf(" init ");
  587. else if (stat == RT_THREAD_CLOSE) rt_kprintf(" close ");
  588. else if (stat == RT_THREAD_RUNNING) rt_kprintf(" running");
  589. #if defined(ARCH_CPU_STACK_GROWS_UPWARD)
  590. ptr = (rt_uint8_t *)thread->stack_addr + thread->stack_size;
  591. while (*ptr == '#')ptr--;
  592. rt_kprintf(" 0x%08x 0x%08x %02d%% 0x%08x %03d\n",
  593. ((rt_uint32_t)thread->sp - (rt_uint32_t)thread->stack_addr),
  594. thread->stack_size,
  595. ((rt_uint32_t)ptr - (rt_uint32_t)thread->stack_addr) * 100 / thread->stack_size,
  596. thread->remaining_tick,
  597. thread->error);
  598. #else
  599. ptr = (rt_uint8_t *)thread->stack_addr;
  600. while (*ptr == '#')ptr++;
  601. rt_kprintf(" 0x%08x 0x%08x %02d%% 0x%08x %03d\n",
  602. (thread->stack_size + (rt_uint32_t)(rt_size_t)thread->stack_addr - (rt_uint32_t)(rt_size_t)thread->sp),
  603. thread->stack_size,
  604. (thread->stack_size + (rt_uint32_t)(rt_size_t)thread->stack_addr - (rt_uint32_t)(rt_size_t)ptr) * 100
  605. / thread->stack_size,
  606. thread->remaining_tick,
  607. thread->error);
  608. #endif
  609. }
  610. long list_process(void)
  611. {
  612. int index;
  613. int maxlen;
  614. rt_ubase_t level;
  615. struct rt_thread *thread;
  616. struct rt_list_node *node, *list;
  617. const char *item_title = "thread";
  618. int count = 0;
  619. struct rt_thread **threads;
  620. maxlen = RT_NAME_MAX;
  621. #ifdef RT_USING_SMP
  622. rt_kprintf("%-*.s %-*.s %-*.s cpu 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. #else
  626. rt_kprintf("%-*.s %-*.s %-*.s pri status sp stack size max used left tick error\n", 4, "PID", maxlen, "CMD", maxlen, item_title);
  627. object_split(4);rt_kprintf(" ");object_split(maxlen);rt_kprintf(" ");object_split(maxlen);rt_kprintf(" ");
  628. rt_kprintf( "--- ------- ---------- ---------- ------ ---------- ---\n");
  629. #endif /*RT_USING_SMP*/
  630. count = rt_object_get_length(RT_Object_Class_Thread);
  631. if (count > 0)
  632. {
  633. /* get thread pointers */
  634. threads = (struct rt_thread **)rt_calloc(count, sizeof(struct rt_thread *));
  635. if (threads)
  636. {
  637. index = rt_object_get_pointers(RT_Object_Class_Thread, (rt_object_t *)threads, count);
  638. if (index > 0)
  639. {
  640. for (index = 0; index <count; index++)
  641. {
  642. struct rt_thread th;
  643. thread = threads[index];
  644. level = rt_hw_interrupt_disable();
  645. if ((thread->type & ~RT_Object_Class_Static) != RT_Object_Class_Thread)
  646. {
  647. rt_hw_interrupt_enable(level);
  648. continue;
  649. }
  650. rt_memcpy(&th, thread, sizeof(struct rt_thread));
  651. rt_hw_interrupt_enable(level);
  652. if (th.lwp == RT_NULL)
  653. {
  654. rt_kprintf(" %-*.*s ", maxlen, RT_NAME_MAX, "kernel");
  655. print_thread_info(&th, maxlen);
  656. }
  657. }
  658. }
  659. rt_free(threads);
  660. }
  661. }
  662. for (index = 0; index < RT_LWP_MAX_NR; index++)
  663. {
  664. struct rt_lwp *lwp = (struct rt_lwp *)lwp_pid_ary[index].data;
  665. if (lwp)
  666. {
  667. list = &lwp->t_grp;
  668. for (node = list->next; node != list; node = node->next)
  669. {
  670. thread = rt_list_entry(node, struct rt_thread, sibling);
  671. rt_kprintf("%4d %-*.*s ", lwp_to_pid(lwp), maxlen, RT_NAME_MAX, lwp->cmd);
  672. print_thread_info(thread, maxlen);
  673. }
  674. }
  675. }
  676. return 0;
  677. }
  678. MSH_CMD_EXPORT(list_process, list process);
  679. static void cmd_kill(int argc, char** argv)
  680. {
  681. int pid;
  682. int sig = 0;
  683. if (argc < 2)
  684. {
  685. rt_kprintf("kill pid or kill pid -s signal\n");
  686. return;
  687. }
  688. pid = atoi(argv[1]);
  689. if (argc >= 4)
  690. {
  691. if (argv[2][0] == '-' && argv[2][1] == 's')
  692. {
  693. sig = atoi(argv[3]);
  694. }
  695. }
  696. lwp_kill(pid, sig);
  697. }
  698. MSH_CMD_EXPORT_ALIAS(cmd_kill, kill, send a signal to a process);
  699. static void cmd_killall(int argc, char** argv)
  700. {
  701. int pid;
  702. if (argc < 2)
  703. {
  704. rt_kprintf("killall processes_name\n");
  705. return;
  706. }
  707. while((pid = lwp_name2pid(argv[1])) >= 0)
  708. {
  709. lwp_kill(pid, 0);
  710. rt_thread_mdelay(100);
  711. }
  712. }
  713. MSH_CMD_EXPORT_ALIAS(cmd_killall, killall, kill processes by name);
  714. #endif
  715. int lwp_check_exit_request(void)
  716. {
  717. rt_thread_t thread = rt_thread_self();
  718. if (!thread->lwp)
  719. {
  720. return 0;
  721. }
  722. if (thread->exit_request == LWP_EXIT_REQUEST_TRIGGERED)
  723. {
  724. thread->exit_request = LWP_EXIT_REQUEST_IN_PROCESS;
  725. return 1;
  726. }
  727. return 0;
  728. }
  729. static int found_thread(struct rt_lwp* lwp, rt_thread_t thread)
  730. {
  731. int found = 0;
  732. rt_base_t level;
  733. rt_list_t *list;
  734. level = rt_hw_interrupt_disable();
  735. list = lwp->t_grp.next;
  736. while (list != &lwp->t_grp)
  737. {
  738. rt_thread_t iter_thread;
  739. iter_thread = rt_list_entry(list, struct rt_thread, sibling);
  740. if (thread == iter_thread)
  741. {
  742. found = 1;
  743. break;
  744. }
  745. list = list->next;
  746. }
  747. rt_hw_interrupt_enable(level);
  748. return found;
  749. }
  750. void lwp_request_thread_exit(rt_thread_t thread_to_exit)
  751. {
  752. rt_thread_t main_thread;
  753. rt_base_t level;
  754. rt_list_t *list;
  755. struct rt_lwp *lwp;
  756. lwp = lwp_self();
  757. if ((!thread_to_exit) || (!lwp))
  758. {
  759. return;
  760. }
  761. level = rt_hw_interrupt_disable();
  762. main_thread = rt_list_entry(lwp->t_grp.prev, struct rt_thread, sibling);
  763. if (thread_to_exit == main_thread)
  764. {
  765. goto finish;
  766. }
  767. if ((struct rt_lwp *)thread_to_exit->lwp != lwp)
  768. {
  769. goto finish;
  770. }
  771. for (list = lwp->t_grp.next; list != &lwp->t_grp; list = list->next)
  772. {
  773. rt_thread_t thread;
  774. thread = rt_list_entry(list, struct rt_thread, sibling);
  775. if (thread != thread_to_exit)
  776. {
  777. continue;
  778. }
  779. if (thread->exit_request == LWP_EXIT_REQUEST_NONE)
  780. {
  781. thread->exit_request = LWP_EXIT_REQUEST_TRIGGERED;
  782. }
  783. if ((thread->stat & RT_THREAD_SUSPEND_MASK) == RT_THREAD_SUSPEND_MASK)
  784. {
  785. thread->error = RT_EINTR;
  786. rt_hw_dsb();
  787. rt_thread_wakeup(thread);
  788. }
  789. break;
  790. }
  791. while (found_thread(lwp, thread_to_exit))
  792. {
  793. rt_thread_mdelay(10);
  794. }
  795. finish:
  796. rt_hw_interrupt_enable(level);
  797. return;
  798. }
  799. void lwp_terminate(struct rt_lwp *lwp)
  800. {
  801. rt_base_t level;
  802. rt_list_t *list;
  803. if (!lwp)
  804. {
  805. /* kernel thread not support */
  806. return;
  807. }
  808. level = rt_hw_interrupt_disable();
  809. for (list = lwp->t_grp.next; list != &lwp->t_grp; list = list->next)
  810. {
  811. rt_thread_t thread;
  812. thread = rt_list_entry(list, struct rt_thread, sibling);
  813. if (thread->exit_request == LWP_EXIT_REQUEST_NONE)
  814. {
  815. thread->exit_request = LWP_EXIT_REQUEST_TRIGGERED;
  816. }
  817. if ((thread->stat & RT_THREAD_SUSPEND_MASK) == RT_THREAD_SUSPEND_MASK)
  818. {
  819. thread->error = RT_EINTR;
  820. rt_hw_dsb();
  821. rt_thread_wakeup(thread);
  822. }
  823. }
  824. rt_hw_interrupt_enable(level);
  825. }
  826. void lwp_wait_subthread_exit(void)
  827. {
  828. rt_base_t level;
  829. struct rt_lwp *lwp;
  830. rt_thread_t thread;
  831. rt_thread_t main_thread;
  832. lwp = lwp_self();
  833. if (!lwp)
  834. {
  835. return;
  836. }
  837. thread = rt_thread_self();
  838. main_thread = rt_list_entry(lwp->t_grp.prev, struct rt_thread, sibling);
  839. if (thread != main_thread)
  840. {
  841. return;
  842. }
  843. while (1)
  844. {
  845. int subthread_is_terminated;
  846. level = rt_hw_interrupt_disable();
  847. subthread_is_terminated = (int)(thread->sibling.prev == &lwp->t_grp);
  848. if (!subthread_is_terminated)
  849. {
  850. rt_thread_t sub_thread;
  851. rt_list_t *list;
  852. int all_subthread_in_init = 1;
  853. /* check all subthread is in init state */
  854. for (list = thread->sibling.prev; list != &lwp->t_grp; list = list->prev)
  855. {
  856. sub_thread = rt_list_entry(list, struct rt_thread, sibling);
  857. if ((sub_thread->stat & RT_THREAD_STAT_MASK) != RT_THREAD_INIT)
  858. {
  859. all_subthread_in_init = 0;
  860. break;
  861. }
  862. }
  863. if (all_subthread_in_init)
  864. {
  865. /* delete all subthread */
  866. while ((list = thread->sibling.prev) != &lwp->t_grp)
  867. {
  868. sub_thread = rt_list_entry(list, struct rt_thread, sibling);
  869. rt_list_remove(&sub_thread->sibling);
  870. rt_thread_delete(sub_thread);
  871. }
  872. subthread_is_terminated = 1;
  873. }
  874. }
  875. rt_hw_interrupt_enable(level);
  876. if (subthread_is_terminated)
  877. {
  878. break;
  879. }
  880. rt_thread_mdelay(10);
  881. }
  882. }