thread.c 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899
  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. * 2006-03-28 Bernard first version
  9. * 2006-04-29 Bernard implement thread timer
  10. * 2006-04-30 Bernard added THREAD_DEBUG
  11. * 2006-05-27 Bernard fixed the rt_thread_yield bug
  12. * 2006-06-03 Bernard fixed the thread timer init bug
  13. * 2006-08-10 Bernard fixed the timer bug in thread_sleep
  14. * 2006-09-03 Bernard changed rt_timer_delete to rt_timer_detach
  15. * 2006-09-03 Bernard implement rt_thread_detach
  16. * 2008-02-16 Bernard fixed the rt_thread_timeout bug
  17. * 2010-03-21 Bernard change the errno of rt_thread_delay/sleep to
  18. * RT_EOK.
  19. * 2010-11-10 Bernard add cleanup callback function in thread exit.
  20. * 2011-09-01 Bernard fixed rt_thread_exit issue when the current
  21. * thread preempted, which reported by Jiaxing Lee.
  22. * 2011-09-08 Bernard fixed the scheduling issue in rt_thread_startup.
  23. * 2012-12-29 Bernard fixed compiling warning.
  24. * 2016-08-09 ArdaFu add thread suspend and resume hook.
  25. * 2017-04-10 armink fixed the rt_thread_delete and rt_thread_detach
  26. * bug when thread has not startup.
  27. * 2018-11-22 Jesven yield is same to rt_schedule
  28. * add support for tasks bound to cpu
  29. * 2021-02-24 Meco Man rearrange rt_thread_control() - schedule the thread when close it
  30. */
  31. #include <rthw.h>
  32. #include <rtthread.h>
  33. extern rt_list_t rt_thread_defunct;
  34. #ifdef RT_USING_HOOK
  35. static void (*rt_thread_suspend_hook)(rt_thread_t thread);
  36. static void (*rt_thread_resume_hook) (rt_thread_t thread);
  37. static void (*rt_thread_inited_hook) (rt_thread_t thread);
  38. /**
  39. * @ingroup Hook
  40. * This function sets a hook function when the system suspend a thread.
  41. *
  42. * @param hook the specified hook function
  43. *
  44. * @note the hook function must be simple and never be blocked or suspend.
  45. */
  46. void rt_thread_suspend_sethook(void (*hook)(rt_thread_t thread))
  47. {
  48. rt_thread_suspend_hook = hook;
  49. }
  50. /**
  51. * @ingroup Hook
  52. * This function sets a hook function when the system resume a thread.
  53. *
  54. * @param hook the specified hook function
  55. *
  56. * @note the hook function must be simple and never be blocked or suspend.
  57. */
  58. void rt_thread_resume_sethook(void (*hook)(rt_thread_t thread))
  59. {
  60. rt_thread_resume_hook = hook;
  61. }
  62. /**
  63. * @ingroup Hook
  64. * This function sets a hook function when a thread is initialized.
  65. *
  66. * @param hook the specified hook function
  67. */
  68. void rt_thread_inited_sethook(void (*hook)(rt_thread_t thread))
  69. {
  70. rt_thread_inited_hook = hook;
  71. }
  72. #endif
  73. /* must be invoke witch rt_hw_interrupt_disable */
  74. static void _thread_cleanup_execute(rt_thread_t thread)
  75. {
  76. register rt_base_t level;
  77. #ifdef RT_USING_MODULE
  78. struct rt_dlmodule *module = RT_NULL;
  79. #endif
  80. level = rt_hw_interrupt_disable();
  81. #ifdef RT_USING_MODULE
  82. module = (struct rt_dlmodule*)thread->module_id;
  83. if (module)
  84. {
  85. dlmodule_destroy(module);
  86. }
  87. #endif
  88. /* invoke thread cleanup */
  89. if (thread->cleanup != RT_NULL)
  90. thread->cleanup(thread);
  91. #ifdef RT_USING_SIGNALS
  92. rt_thread_free_sig(thread);
  93. #endif
  94. rt_hw_interrupt_enable(level);
  95. }
  96. void rt_thread_exit(void)
  97. {
  98. struct rt_thread *thread;
  99. register rt_base_t level;
  100. /* get current thread */
  101. thread = rt_thread_self();
  102. /* disable interrupt */
  103. level = rt_hw_interrupt_disable();
  104. _thread_cleanup_execute(thread);
  105. /* remove from schedule */
  106. rt_schedule_remove_thread(thread);
  107. /* change stat */
  108. thread->stat = RT_THREAD_CLOSE;
  109. /* remove it from timer list */
  110. rt_timer_detach(&thread->thread_timer);
  111. if (rt_object_is_systemobject((rt_object_t)thread) == RT_TRUE)
  112. {
  113. rt_object_detach((rt_object_t)thread);
  114. }
  115. else
  116. {
  117. /* insert to defunct thread list */
  118. rt_list_insert_after(&rt_thread_defunct, &(thread->tlist));
  119. }
  120. /* switch to next task */
  121. rt_schedule();
  122. /* enable interrupt */
  123. rt_hw_interrupt_enable(level);
  124. }
  125. static rt_err_t _rt_thread_init(struct rt_thread *thread,
  126. const char *name,
  127. void (*entry)(void *parameter),
  128. void *parameter,
  129. void *stack_start,
  130. rt_uint32_t stack_size,
  131. rt_uint8_t priority,
  132. rt_uint32_t tick)
  133. {
  134. /* init thread list */
  135. rt_list_init(&(thread->tlist));
  136. thread->entry = (void *)entry;
  137. thread->parameter = parameter;
  138. /* stack init */
  139. thread->stack_addr = stack_start;
  140. thread->stack_size = stack_size;
  141. /* init thread stack */
  142. rt_memset(thread->stack_addr, '#', thread->stack_size);
  143. #ifdef ARCH_CPU_STACK_GROWS_UPWARD
  144. thread->sp = (void *)rt_hw_stack_init(thread->entry, thread->parameter,
  145. (void *)((char *)thread->stack_addr),
  146. (void *)rt_thread_exit);
  147. #else
  148. thread->sp = (void *)rt_hw_stack_init(thread->entry, thread->parameter,
  149. (rt_uint8_t *)((char *)thread->stack_addr + thread->stack_size - sizeof(rt_ubase_t)),
  150. (void *)rt_thread_exit);
  151. #endif
  152. /* priority init */
  153. RT_ASSERT(priority < RT_THREAD_PRIORITY_MAX);
  154. thread->init_priority = priority;
  155. thread->current_priority = priority;
  156. thread->number_mask = 0;
  157. #if RT_THREAD_PRIORITY_MAX > 32
  158. thread->number = 0;
  159. thread->high_mask = 0;
  160. #endif
  161. /* tick init */
  162. thread->init_tick = tick;
  163. thread->remaining_tick = tick;
  164. /* error and flags */
  165. thread->error = RT_EOK;
  166. thread->stat = RT_THREAD_INIT;
  167. #ifdef RT_USING_SMP
  168. /* not bind on any cpu */
  169. thread->bind_cpu = RT_CPUS_NR;
  170. thread->oncpu = RT_CPU_DETACHED;
  171. /* lock init */
  172. thread->scheduler_lock_nest = 0;
  173. thread->cpus_lock_nest = 0;
  174. thread->critical_lock_nest = 0;
  175. #endif /*RT_USING_SMP*/
  176. /* initialize cleanup function and user data */
  177. thread->cleanup = 0;
  178. thread->user_data = 0;
  179. /* initialize thread timer */
  180. rt_timer_init(&(thread->thread_timer),
  181. thread->name,
  182. rt_thread_timeout,
  183. thread,
  184. 0,
  185. RT_TIMER_FLAG_ONE_SHOT);
  186. /* initialize signal */
  187. #ifdef RT_USING_SIGNALS
  188. thread->sig_mask = 0x00;
  189. thread->sig_pending = 0x00;
  190. #ifndef RT_USING_SMP
  191. thread->sig_ret = RT_NULL;
  192. #endif
  193. thread->sig_vectors = RT_NULL;
  194. thread->si_list = RT_NULL;
  195. #endif
  196. #ifdef RT_USING_LWP
  197. thread->lwp = RT_NULL;
  198. #endif
  199. RT_OBJECT_HOOK_CALL(rt_thread_inited_hook, (thread));
  200. return RT_EOK;
  201. }
  202. /**
  203. * @addtogroup Thread
  204. */
  205. /**@{*/
  206. /**
  207. * This function will initialize a thread, normally it's used to initialize a
  208. * static thread object.
  209. *
  210. * @param thread the static thread object
  211. * @param name the name of thread, which shall be unique
  212. * @param entry the entry function of thread
  213. * @param parameter the parameter of thread enter function
  214. * @param stack_start the start address of thread stack
  215. * @param stack_size the size of thread stack
  216. * @param priority the priority of thread
  217. * @param tick the time slice if there are same priority thread
  218. *
  219. * @return the operation status, RT_EOK on OK, -RT_ERROR on error
  220. */
  221. rt_err_t rt_thread_init(struct rt_thread *thread,
  222. const char *name,
  223. void (*entry)(void *parameter),
  224. void *parameter,
  225. void *stack_start,
  226. rt_uint32_t stack_size,
  227. rt_uint8_t priority,
  228. rt_uint32_t tick)
  229. {
  230. /* thread check */
  231. RT_ASSERT(thread != RT_NULL);
  232. RT_ASSERT(stack_start != RT_NULL);
  233. /* initialize thread object */
  234. rt_object_init((rt_object_t)thread, RT_Object_Class_Thread, name);
  235. return _rt_thread_init(thread,
  236. name,
  237. entry,
  238. parameter,
  239. stack_start,
  240. stack_size,
  241. priority,
  242. tick);
  243. }
  244. RTM_EXPORT(rt_thread_init);
  245. /**
  246. * This function will return self thread object
  247. *
  248. * @return the self thread object
  249. */
  250. rt_thread_t rt_thread_self(void)
  251. {
  252. #ifdef RT_USING_SMP
  253. rt_base_t lock;
  254. rt_thread_t self;
  255. lock = rt_hw_local_irq_disable();
  256. self = rt_cpu_self()->current_thread;
  257. rt_hw_local_irq_enable(lock);
  258. return self;
  259. #else
  260. extern rt_thread_t rt_current_thread;
  261. return rt_current_thread;
  262. #endif
  263. }
  264. RTM_EXPORT(rt_thread_self);
  265. /**
  266. * This function will start a thread and put it to system ready queue
  267. *
  268. * @param thread the thread to be started
  269. *
  270. * @return the operation status, RT_EOK on OK, -RT_ERROR on error
  271. */
  272. rt_err_t rt_thread_startup(rt_thread_t thread)
  273. {
  274. /* thread check */
  275. RT_ASSERT(thread != RT_NULL);
  276. RT_ASSERT((thread->stat & RT_THREAD_STAT_MASK) == RT_THREAD_INIT);
  277. RT_ASSERT(rt_object_get_type((rt_object_t)thread) == RT_Object_Class_Thread);
  278. /* set current priority to initialize priority */
  279. thread->current_priority = thread->init_priority;
  280. /* calculate priority attribute */
  281. #if RT_THREAD_PRIORITY_MAX > 32
  282. thread->number = thread->current_priority >> 3; /* 5bit */
  283. thread->number_mask = 1L << thread->number;
  284. thread->high_mask = 1L << (thread->current_priority & 0x07); /* 3bit */
  285. #else
  286. thread->number_mask = 1L << thread->current_priority;
  287. #endif
  288. RT_DEBUG_LOG(RT_DEBUG_THREAD, ("startup a thread:%s with priority:%d\n",
  289. thread->name, thread->init_priority));
  290. /* change thread stat */
  291. thread->stat = RT_THREAD_SUSPEND;
  292. /* then resume it */
  293. rt_thread_resume(thread);
  294. if (rt_thread_self() != RT_NULL)
  295. {
  296. /* do a scheduling */
  297. rt_schedule();
  298. }
  299. return RT_EOK;
  300. }
  301. RTM_EXPORT(rt_thread_startup);
  302. /**
  303. * This function will detach a thread. The thread object will be removed from
  304. * thread queue and detached/deleted from system object management.
  305. *
  306. * @param thread the thread to be deleted
  307. *
  308. * @return the operation status, RT_EOK on OK, -RT_ERROR on error
  309. */
  310. rt_err_t rt_thread_detach(rt_thread_t thread)
  311. {
  312. /* thread check */
  313. RT_ASSERT(thread != RT_NULL);
  314. RT_ASSERT(rt_object_get_type((rt_object_t)thread) == RT_Object_Class_Thread);
  315. RT_ASSERT(rt_object_is_systemobject((rt_object_t)thread));
  316. if ((thread->stat & RT_THREAD_STAT_MASK) == RT_THREAD_CLOSE)
  317. return RT_EOK;
  318. if ((thread->stat & RT_THREAD_STAT_MASK) != RT_THREAD_INIT)
  319. {
  320. /* remove from schedule */
  321. rt_schedule_remove_thread(thread);
  322. }
  323. _thread_cleanup_execute(thread);
  324. /* release thread timer */
  325. rt_timer_detach(&(thread->thread_timer));
  326. /* change stat */
  327. thread->stat = RT_THREAD_CLOSE;
  328. /* detach thread object */
  329. rt_object_detach((rt_object_t)thread);
  330. return RT_EOK;
  331. }
  332. RTM_EXPORT(rt_thread_detach);
  333. #ifdef RT_USING_HEAP
  334. /**
  335. * This function will create a thread object and allocate thread object memory
  336. * and stack.
  337. *
  338. * @param name the name of thread, which shall be unique
  339. * @param entry the entry function of thread
  340. * @param parameter the parameter of thread enter function
  341. * @param stack_size the size of thread stack
  342. * @param priority the priority of thread
  343. * @param tick the time slice if there are same priority thread
  344. *
  345. * @return the created thread object
  346. */
  347. rt_thread_t rt_thread_create(const char *name,
  348. void (*entry)(void *parameter),
  349. void *parameter,
  350. rt_uint32_t stack_size,
  351. rt_uint8_t priority,
  352. rt_uint32_t tick)
  353. {
  354. struct rt_thread *thread;
  355. void *stack_start;
  356. thread = (struct rt_thread *)rt_object_allocate(RT_Object_Class_Thread,
  357. name);
  358. if (thread == RT_NULL)
  359. return RT_NULL;
  360. stack_start = (void *)RT_KERNEL_MALLOC(stack_size);
  361. if (stack_start == RT_NULL)
  362. {
  363. /* allocate stack failure */
  364. rt_object_delete((rt_object_t)thread);
  365. return RT_NULL;
  366. }
  367. _rt_thread_init(thread,
  368. name,
  369. entry,
  370. parameter,
  371. stack_start,
  372. stack_size,
  373. priority,
  374. tick);
  375. return thread;
  376. }
  377. RTM_EXPORT(rt_thread_create);
  378. /**
  379. * This function will delete a thread. The thread object will be removed from
  380. * thread queue and deleted from system object management in the idle thread.
  381. *
  382. * @param thread the thread to be deleted
  383. *
  384. * @return the operation status, RT_EOK on OK, -RT_ERROR on error
  385. */
  386. rt_err_t rt_thread_delete(rt_thread_t thread)
  387. {
  388. rt_base_t lock;
  389. /* thread check */
  390. RT_ASSERT(thread != RT_NULL);
  391. RT_ASSERT(rt_object_get_type((rt_object_t)thread) == RT_Object_Class_Thread);
  392. RT_ASSERT(rt_object_is_systemobject((rt_object_t)thread) == RT_FALSE);
  393. if ((thread->stat & RT_THREAD_STAT_MASK) == RT_THREAD_CLOSE)
  394. return RT_EOK;
  395. if ((thread->stat & RT_THREAD_STAT_MASK) != RT_THREAD_INIT)
  396. {
  397. /* remove from schedule */
  398. rt_schedule_remove_thread(thread);
  399. }
  400. _thread_cleanup_execute(thread);
  401. /* release thread timer */
  402. rt_timer_detach(&(thread->thread_timer));
  403. /* disable interrupt */
  404. lock = rt_hw_interrupt_disable();
  405. /* change stat */
  406. thread->stat = RT_THREAD_CLOSE;
  407. /* insert to defunct thread list */
  408. rt_list_insert_after(&rt_thread_defunct, &(thread->tlist));
  409. /* enable interrupt */
  410. rt_hw_interrupt_enable(lock);
  411. return RT_EOK;
  412. }
  413. RTM_EXPORT(rt_thread_delete);
  414. #endif
  415. /**
  416. * This function will let current thread yield processor, and scheduler will
  417. * choose a highest thread to run. After yield processor, the current thread
  418. * is still in READY state.
  419. *
  420. * @return RT_EOK
  421. */
  422. rt_err_t rt_thread_yield(void)
  423. {
  424. struct rt_thread *thread;
  425. rt_base_t lock;
  426. thread = rt_thread_self();
  427. lock = rt_hw_interrupt_disable();
  428. thread->remaining_tick = thread->init_tick;
  429. thread->stat |= RT_THREAD_STAT_YIELD;
  430. rt_schedule();
  431. rt_hw_interrupt_enable(lock);
  432. return RT_EOK;
  433. }
  434. RTM_EXPORT(rt_thread_yield);
  435. /**
  436. * This function will let current thread sleep for some ticks.
  437. *
  438. * @param tick the sleep ticks
  439. *
  440. * @return RT_EOK
  441. */
  442. rt_err_t rt_thread_sleep(rt_tick_t tick)
  443. {
  444. register rt_base_t temp;
  445. struct rt_thread *thread;
  446. /* set to current thread */
  447. thread = rt_thread_self();
  448. RT_ASSERT(thread != RT_NULL);
  449. RT_ASSERT(rt_object_get_type((rt_object_t)thread) == RT_Object_Class_Thread);
  450. /* disable interrupt */
  451. temp = rt_hw_interrupt_disable();
  452. /* suspend thread */
  453. rt_thread_suspend(thread);
  454. /* reset the timeout of thread timer and start it */
  455. rt_timer_control(&(thread->thread_timer), RT_TIMER_CTRL_SET_TIME, &tick);
  456. rt_timer_start(&(thread->thread_timer));
  457. /* enable interrupt */
  458. rt_hw_interrupt_enable(temp);
  459. rt_schedule();
  460. /* clear error number of this thread to RT_EOK */
  461. if (thread->error == -RT_ETIMEOUT)
  462. thread->error = RT_EOK;
  463. return RT_EOK;
  464. }
  465. /**
  466. * This function will let current thread delay for some ticks.
  467. *
  468. * @param tick the delay ticks
  469. *
  470. * @return RT_EOK
  471. */
  472. rt_err_t rt_thread_delay(rt_tick_t tick)
  473. {
  474. return rt_thread_sleep(tick);
  475. }
  476. RTM_EXPORT(rt_thread_delay);
  477. /**
  478. * This function will let current thread delay until (*tick + inc_tick).
  479. *
  480. * @param tick the tick of last wakeup.
  481. * @param inc_tick the increment tick
  482. *
  483. * @return RT_EOK
  484. */
  485. rt_err_t rt_thread_delay_until(rt_tick_t *tick, rt_tick_t inc_tick)
  486. {
  487. register rt_base_t level;
  488. struct rt_thread *thread;
  489. rt_tick_t cur_tick;
  490. RT_ASSERT(tick != RT_NULL);
  491. /* set to current thread */
  492. thread = rt_thread_self();
  493. RT_ASSERT(thread != RT_NULL);
  494. RT_ASSERT(rt_object_get_type((rt_object_t)thread) == RT_Object_Class_Thread);
  495. /* disable interrupt */
  496. level = rt_hw_interrupt_disable();
  497. cur_tick = rt_tick_get();
  498. if (cur_tick - *tick < inc_tick)
  499. {
  500. rt_tick_t left_tick;
  501. *tick += inc_tick;
  502. left_tick = *tick - cur_tick;
  503. /* suspend thread */
  504. rt_thread_suspend(thread);
  505. /* reset the timeout of thread timer and start it */
  506. rt_timer_control(&(thread->thread_timer), RT_TIMER_CTRL_SET_TIME, &left_tick);
  507. rt_timer_start(&(thread->thread_timer));
  508. /* enable interrupt */
  509. rt_hw_interrupt_enable(level);
  510. rt_schedule();
  511. /* clear error number of this thread to RT_EOK */
  512. if (thread->error == -RT_ETIMEOUT)
  513. {
  514. thread->error = RT_EOK;
  515. }
  516. }
  517. else
  518. {
  519. *tick = cur_tick;
  520. rt_hw_interrupt_enable(level);
  521. }
  522. return RT_EOK;
  523. }
  524. RTM_EXPORT(rt_thread_delay_until);
  525. /**
  526. * This function will let current thread delay for some milliseconds.
  527. *
  528. * @param ms the delay ms time
  529. *
  530. * @return RT_EOK
  531. */
  532. rt_err_t rt_thread_mdelay(rt_int32_t ms)
  533. {
  534. rt_tick_t tick;
  535. tick = rt_tick_from_millisecond(ms);
  536. return rt_thread_sleep(tick);
  537. }
  538. RTM_EXPORT(rt_thread_mdelay);
  539. /**
  540. * This function will control thread behaviors according to control command.
  541. *
  542. * @param thread the specified thread to be controlled
  543. * @param cmd the control command, which includes
  544. * RT_THREAD_CTRL_CHANGE_PRIORITY for changing priority level of thread;
  545. * RT_THREAD_CTRL_STARTUP for starting a thread;
  546. * RT_THREAD_CTRL_CLOSE for delete a thread;
  547. * RT_THREAD_CTRL_BIND_CPU for bind the thread to a CPU.
  548. * @param arg the argument of control command
  549. *
  550. * @return RT_EOK
  551. */
  552. rt_err_t rt_thread_control(rt_thread_t thread, int cmd, void *arg)
  553. {
  554. register rt_base_t temp;
  555. /* thread check */
  556. RT_ASSERT(thread != RT_NULL);
  557. RT_ASSERT(rt_object_get_type((rt_object_t)thread) == RT_Object_Class_Thread);
  558. switch (cmd)
  559. {
  560. case RT_THREAD_CTRL_CHANGE_PRIORITY:
  561. {
  562. /* disable interrupt */
  563. temp = rt_hw_interrupt_disable();
  564. /* for ready thread, change queue */
  565. if ((thread->stat & RT_THREAD_STAT_MASK) == RT_THREAD_READY)
  566. {
  567. /* remove thread from schedule queue first */
  568. rt_schedule_remove_thread(thread);
  569. /* change thread priority */
  570. thread->current_priority = *(rt_uint8_t *)arg;
  571. /* recalculate priority attribute */
  572. #if RT_THREAD_PRIORITY_MAX > 32
  573. thread->number = thread->current_priority >> 3; /* 5bit */
  574. thread->number_mask = 1 << thread->number;
  575. thread->high_mask = 1 << (thread->current_priority & 0x07); /* 3bit */
  576. #else
  577. thread->number_mask = 1 << thread->current_priority;
  578. #endif
  579. /* insert thread to schedule queue again */
  580. rt_schedule_insert_thread(thread);
  581. }
  582. else
  583. {
  584. thread->current_priority = *(rt_uint8_t *)arg;
  585. /* recalculate priority attribute */
  586. #if RT_THREAD_PRIORITY_MAX > 32
  587. thread->number = thread->current_priority >> 3; /* 5bit */
  588. thread->number_mask = 1 << thread->number;
  589. thread->high_mask = 1 << (thread->current_priority & 0x07); /* 3bit */
  590. #else
  591. thread->number_mask = 1 << thread->current_priority;
  592. #endif
  593. }
  594. /* enable interrupt */
  595. rt_hw_interrupt_enable(temp);
  596. break;
  597. }
  598. case RT_THREAD_CTRL_STARTUP:
  599. {
  600. return rt_thread_startup(thread);
  601. }
  602. case RT_THREAD_CTRL_CLOSE:
  603. {
  604. rt_err_t rt_err;
  605. if (rt_object_is_systemobject((rt_object_t)thread) == RT_TRUE)
  606. {
  607. rt_err = rt_thread_detach(thread);
  608. }
  609. #ifdef RT_USING_HEAP
  610. else
  611. {
  612. rt_err = rt_thread_delete(thread);
  613. }
  614. #endif
  615. rt_schedule();
  616. return rt_err;
  617. }
  618. #ifdef RT_USING_SMP
  619. case RT_THREAD_CTRL_BIND_CPU:
  620. {
  621. rt_uint8_t cpu;
  622. if ((thread->stat & RT_THREAD_STAT_MASK) != RT_THREAD_INIT)
  623. {
  624. /* we only support bind cpu before started phase. */
  625. return RT_ERROR;
  626. }
  627. cpu = (rt_uint8_t)(size_t)arg;
  628. thread->bind_cpu = cpu > RT_CPUS_NR? RT_CPUS_NR : cpu;
  629. break;
  630. }
  631. #endif /*RT_USING_SMP*/
  632. default:
  633. break;
  634. }
  635. return RT_EOK;
  636. }
  637. RTM_EXPORT(rt_thread_control);
  638. /**
  639. * This function will suspend the specified thread.
  640. *
  641. * @param thread the thread to be suspended
  642. *
  643. * @return the operation status, RT_EOK on OK, -RT_ERROR on error
  644. *
  645. * @note if suspend self thread, after this function call, the
  646. * rt_schedule() must be invoked.
  647. */
  648. rt_err_t rt_thread_suspend(rt_thread_t thread)
  649. {
  650. register rt_base_t stat;
  651. register rt_base_t temp;
  652. /* thread check */
  653. RT_ASSERT(thread != RT_NULL);
  654. RT_ASSERT(rt_object_get_type((rt_object_t)thread) == RT_Object_Class_Thread);
  655. RT_DEBUG_LOG(RT_DEBUG_THREAD, ("thread suspend: %s\n", thread->name));
  656. stat = thread->stat & RT_THREAD_STAT_MASK;
  657. if ((stat != RT_THREAD_READY) && (stat != RT_THREAD_RUNNING))
  658. {
  659. RT_DEBUG_LOG(RT_DEBUG_THREAD, ("thread suspend: thread disorder, 0x%2x\n",
  660. thread->stat));
  661. return -RT_ERROR;
  662. }
  663. /* disable interrupt */
  664. temp = rt_hw_interrupt_disable();
  665. if (stat == RT_THREAD_RUNNING)
  666. {
  667. /* not suspend running status thread on other core */
  668. RT_ASSERT(thread == rt_thread_self());
  669. }
  670. /* change thread stat */
  671. rt_schedule_remove_thread(thread);
  672. thread->stat = RT_THREAD_SUSPEND | (thread->stat & ~RT_THREAD_STAT_MASK);
  673. /* stop thread timer anyway */
  674. rt_timer_stop(&(thread->thread_timer));
  675. /* enable interrupt */
  676. rt_hw_interrupt_enable(temp);
  677. RT_OBJECT_HOOK_CALL(rt_thread_suspend_hook, (thread));
  678. return RT_EOK;
  679. }
  680. RTM_EXPORT(rt_thread_suspend);
  681. /**
  682. * This function will resume a thread and put it to system ready queue.
  683. *
  684. * @param thread the thread to be resumed
  685. *
  686. * @return the operation status, RT_EOK on OK, -RT_ERROR on error
  687. */
  688. rt_err_t rt_thread_resume(rt_thread_t thread)
  689. {
  690. register rt_base_t temp;
  691. /* thread check */
  692. RT_ASSERT(thread != RT_NULL);
  693. RT_ASSERT(rt_object_get_type((rt_object_t)thread) == RT_Object_Class_Thread);
  694. RT_DEBUG_LOG(RT_DEBUG_THREAD, ("thread resume: %s\n", thread->name));
  695. if ((thread->stat & RT_THREAD_STAT_MASK) != RT_THREAD_SUSPEND)
  696. {
  697. RT_DEBUG_LOG(RT_DEBUG_THREAD, ("thread resume: thread disorder, %d\n",
  698. thread->stat));
  699. return -RT_ERROR;
  700. }
  701. /* disable interrupt */
  702. temp = rt_hw_interrupt_disable();
  703. /* remove from suspend list */
  704. rt_list_remove(&(thread->tlist));
  705. rt_timer_stop(&thread->thread_timer);
  706. /* enable interrupt */
  707. rt_hw_interrupt_enable(temp);
  708. /* insert to schedule ready list */
  709. rt_schedule_insert_thread(thread);
  710. RT_OBJECT_HOOK_CALL(rt_thread_resume_hook, (thread));
  711. return RT_EOK;
  712. }
  713. RTM_EXPORT(rt_thread_resume);
  714. /**
  715. * This function is the timeout function for thread, normally which is invoked
  716. * when thread is timeout to wait some resource.
  717. *
  718. * @param parameter the parameter of thread timeout function
  719. */
  720. void rt_thread_timeout(void *parameter)
  721. {
  722. struct rt_thread *thread;
  723. thread = (struct rt_thread *)parameter;
  724. /* thread check */
  725. RT_ASSERT(thread != RT_NULL);
  726. RT_ASSERT((thread->stat & RT_THREAD_STAT_MASK) == RT_THREAD_SUSPEND);
  727. RT_ASSERT(rt_object_get_type((rt_object_t)thread) == RT_Object_Class_Thread);
  728. /* set error number */
  729. thread->error = -RT_ETIMEOUT;
  730. /* remove from suspend list */
  731. rt_list_remove(&(thread->tlist));
  732. /* insert to schedule ready list */
  733. rt_schedule_insert_thread(thread);
  734. /* do schedule */
  735. rt_schedule();
  736. }
  737. RTM_EXPORT(rt_thread_timeout);
  738. /**
  739. * This function will find the specified thread.
  740. *
  741. * @param name the name of thread finding
  742. *
  743. * @return the found thread
  744. *
  745. * @note please don't invoke this function in interrupt status.
  746. */
  747. rt_thread_t rt_thread_find(char *name)
  748. {
  749. return (rt_thread_t)rt_object_find(name, RT_Object_Class_Thread);
  750. }
  751. RTM_EXPORT(rt_thread_find);
  752. /**@}*/