lwp_pid.c 25 KB

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