ipc.c 74 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773
  1. /*
  2. * Copyright (c) 2006-2018, RT-Thread Development Team
  3. *
  4. * SPDX-License-Identifier: Apache-2.0
  5. *
  6. * Change Logs:
  7. * Date Author Notes
  8. * 2006-03-14 Bernard the first version
  9. * 2006-04-25 Bernard implement semaphore
  10. * 2006-05-03 Bernard add RT_IPC_DEBUG
  11. * modify the type of IPC waiting time to rt_int32_t
  12. * 2006-05-10 Bernard fix the semaphore take bug and add IPC object
  13. * 2006-05-12 Bernard implement mailbox and message queue
  14. * 2006-05-20 Bernard implement mutex
  15. * 2006-05-23 Bernard implement fast event
  16. * 2006-05-24 Bernard implement event
  17. * 2006-06-03 Bernard fix the thread timer init bug
  18. * 2006-06-05 Bernard fix the mutex release bug
  19. * 2006-06-07 Bernard fix the message queue send bug
  20. * 2006-08-04 Bernard add hook support
  21. * 2009-05-21 Yi.qiu fix the sem release bug
  22. * 2009-07-18 Bernard fix the event clear bug
  23. * 2009-09-09 Bernard remove fast event and fix ipc release bug
  24. * 2009-10-10 Bernard change semaphore and mutex value to unsigned value
  25. * 2009-10-25 Bernard change the mb/mq receive timeout to 0 if the
  26. * re-calculated delta tick is a negative number.
  27. * 2009-12-16 Bernard fix the rt_ipc_object_suspend issue when IPC flag
  28. * is RT_IPC_FLAG_PRIO
  29. * 2010-01-20 mbbill remove rt_ipc_object_decrease function.
  30. * 2010-04-20 Bernard move memcpy outside interrupt disable in mq
  31. * 2010-10-26 yi.qiu add module support in rt_mp_delete and rt_mq_delete
  32. * 2010-11-10 Bernard add IPC reset command implementation.
  33. * 2011-12-18 Bernard add more parameter checking in message queue
  34. * 2013-09-14 Grissiom add an option check in rt_event_recv
  35. * 2018-10-02 Bernard add 64bit support for mailbox
  36. * 2019-09-16 tyx add send wait support for message queue
  37. * 2020-07-29 Meco Man fix thread->event_set/event_info when received an
  38. * event without pending
  39. * 2020-10-11 Meco Man add value overflow-check code
  40. */
  41. #include <rtthread.h>
  42. #include <rthw.h>
  43. #ifdef RT_USING_HOOK
  44. extern void (*rt_object_trytake_hook)(struct rt_object *object);
  45. extern void (*rt_object_take_hook)(struct rt_object *object);
  46. extern void (*rt_object_put_hook)(struct rt_object *object);
  47. #endif
  48. /**
  49. * @addtogroup IPC
  50. */
  51. /**@{*/
  52. /**
  53. * This function will initialize an IPC object
  54. *
  55. * @param ipc the IPC object
  56. *
  57. * @return the operation status, RT_EOK on successful
  58. */
  59. rt_inline rt_err_t rt_ipc_object_init(struct rt_ipc_object *ipc)
  60. {
  61. /* initialize ipc object */
  62. rt_list_init(&(ipc->suspend_thread));
  63. return RT_EOK;
  64. }
  65. /**
  66. * This function will suspend a thread to a specified list. IPC object or some
  67. * double-queue object (mailbox etc.) contains this kind of list.
  68. *
  69. * @param list the IPC suspended thread list
  70. * @param thread the thread object to be suspended
  71. * @param flag the IPC object flag,
  72. * which shall be RT_IPC_FLAG_FIFO/RT_IPC_FLAG_PRIO.
  73. *
  74. * @return the operation status, RT_EOK on successful
  75. */
  76. rt_inline rt_err_t rt_ipc_list_suspend(rt_list_t *list,
  77. struct rt_thread *thread,
  78. rt_uint8_t flag,
  79. int suspend_flag)
  80. {
  81. rt_err_t ret = rt_thread_suspend_with_flag(thread, suspend_flag);
  82. /* suspend thread */
  83. if (ret != RT_EOK)
  84. {
  85. return ret;
  86. }
  87. switch (flag)
  88. {
  89. case RT_IPC_FLAG_FIFO:
  90. rt_list_insert_before(list, &(thread->tlist));
  91. break;
  92. case RT_IPC_FLAG_PRIO:
  93. {
  94. struct rt_list_node *n;
  95. struct rt_thread *sthread;
  96. /* find a suitable position */
  97. for (n = list->next; n != list; n = n->next)
  98. {
  99. sthread = rt_list_entry(n, struct rt_thread, tlist);
  100. /* find out */
  101. if (thread->current_priority < sthread->current_priority)
  102. {
  103. /* insert this thread before the sthread */
  104. rt_list_insert_before(&(sthread->tlist), &(thread->tlist));
  105. break;
  106. }
  107. }
  108. /*
  109. * not found a suitable position,
  110. * append to the end of suspend_thread list
  111. */
  112. if (n == list)
  113. rt_list_insert_before(list, &(thread->tlist));
  114. }
  115. break;
  116. default:
  117. break;
  118. }
  119. return RT_EOK;
  120. }
  121. /**
  122. * This function will resume the first thread in the list of a IPC object:
  123. * - remove the thread from suspend queue of IPC object
  124. * - put the thread into system ready queue
  125. *
  126. * @param list the thread list
  127. *
  128. * @return the operation status, RT_EOK on successful
  129. */
  130. rt_inline rt_err_t rt_ipc_list_resume(rt_list_t *list)
  131. {
  132. struct rt_thread *thread;
  133. /* get thread entry */
  134. thread = rt_list_entry(list->next, struct rt_thread, tlist);
  135. thread->error = RT_EOK;
  136. RT_DEBUG_LOG(RT_DEBUG_IPC, ("resume thread:%s\n", thread->name));
  137. /* resume it */
  138. rt_thread_resume(thread);
  139. return RT_EOK;
  140. }
  141. /**
  142. * This function will resume all suspended threads in a list, including
  143. * suspend list of IPC object and private list of mailbox etc.
  144. *
  145. * @param list of the threads to resume
  146. *
  147. * @return the operation status, RT_EOK on successful
  148. */
  149. rt_inline rt_err_t rt_ipc_list_resume_all(rt_list_t *list)
  150. {
  151. struct rt_thread *thread;
  152. register rt_ubase_t temp;
  153. /* wakeup all suspended threads */
  154. while (!rt_list_isempty(list))
  155. {
  156. /* disable interrupt */
  157. temp = rt_hw_interrupt_disable();
  158. /* get next suspended thread */
  159. thread = rt_list_entry(list->next, struct rt_thread, tlist);
  160. /* set error code to RT_ERROR */
  161. thread->error = -RT_ERROR;
  162. /*
  163. * resume thread
  164. * In rt_thread_resume function, it will remove current thread from
  165. * suspended list
  166. */
  167. rt_thread_resume(thread);
  168. /* enable interrupt */
  169. rt_hw_interrupt_enable(temp);
  170. }
  171. return RT_EOK;
  172. }
  173. #ifdef RT_USING_SEMAPHORE
  174. /**
  175. * This function will initialize a semaphore and put it under control of
  176. * resource management.
  177. *
  178. * @param sem the semaphore object
  179. * @param name the name of semaphore
  180. * @param value the initial value of semaphore
  181. * @param flag the flag of semaphore
  182. *
  183. * @return the operation status, RT_EOK on successful
  184. */
  185. rt_err_t rt_sem_init(rt_sem_t sem,
  186. const char *name,
  187. rt_uint32_t value,
  188. rt_uint8_t flag)
  189. {
  190. RT_ASSERT(sem != RT_NULL);
  191. RT_ASSERT(value < 0x10000U);
  192. /* initialize object */
  193. rt_object_init(&(sem->parent.parent), RT_Object_Class_Semaphore, name);
  194. /* initialize ipc object */
  195. rt_ipc_object_init(&(sem->parent));
  196. /* set initial value */
  197. sem->value = (rt_uint16_t)value;
  198. /* set parent */
  199. sem->parent.parent.flag = flag;
  200. return RT_EOK;
  201. }
  202. RTM_EXPORT(rt_sem_init);
  203. /**
  204. * This function will detach a semaphore from resource management
  205. *
  206. * @param sem the semaphore object
  207. *
  208. * @return the operation status, RT_EOK on successful
  209. *
  210. * @see rt_sem_delete
  211. */
  212. rt_err_t rt_sem_detach(rt_sem_t sem)
  213. {
  214. /* parameter check */
  215. RT_ASSERT(sem != RT_NULL);
  216. RT_ASSERT(rt_object_get_type(&sem->parent.parent) == RT_Object_Class_Semaphore);
  217. RT_ASSERT(rt_object_is_systemobject(&sem->parent.parent));
  218. /* wakeup all suspended threads */
  219. rt_ipc_list_resume_all(&(sem->parent.suspend_thread));
  220. /* detach semaphore object */
  221. rt_object_detach(&(sem->parent.parent));
  222. return RT_EOK;
  223. }
  224. RTM_EXPORT(rt_sem_detach);
  225. #ifdef RT_USING_HEAP
  226. /**
  227. * This function will create a semaphore from system resource
  228. *
  229. * @param name the name of semaphore
  230. * @param value the initial value of semaphore
  231. * @param flag the flag of semaphore
  232. *
  233. * @return the created semaphore, RT_NULL on error happen
  234. *
  235. * @see rt_sem_init
  236. */
  237. rt_sem_t rt_sem_create(const char *name, rt_uint32_t value, rt_uint8_t flag)
  238. {
  239. rt_sem_t sem;
  240. RT_DEBUG_NOT_IN_INTERRUPT;
  241. RT_ASSERT(value < 0x10000U);
  242. /* allocate object */
  243. sem = (rt_sem_t)rt_object_allocate(RT_Object_Class_Semaphore, name);
  244. if (sem == RT_NULL)
  245. return sem;
  246. /* initialize ipc object */
  247. rt_ipc_object_init(&(sem->parent));
  248. /* set initial value */
  249. sem->value = value;
  250. /* set parent */
  251. sem->parent.parent.flag = flag;
  252. return sem;
  253. }
  254. RTM_EXPORT(rt_sem_create);
  255. /**
  256. * This function will delete a semaphore object and release the memory
  257. *
  258. * @param sem the semaphore object
  259. *
  260. * @return the error code
  261. *
  262. * @see rt_sem_detach
  263. */
  264. rt_err_t rt_sem_delete(rt_sem_t sem)
  265. {
  266. RT_DEBUG_NOT_IN_INTERRUPT;
  267. /* parameter check */
  268. RT_ASSERT(sem != RT_NULL);
  269. RT_ASSERT(rt_object_get_type(&sem->parent.parent) == RT_Object_Class_Semaphore);
  270. RT_ASSERT(rt_object_is_systemobject(&sem->parent.parent) == RT_FALSE);
  271. /* wakeup all suspended threads */
  272. rt_ipc_list_resume_all(&(sem->parent.suspend_thread));
  273. /* delete semaphore object */
  274. rt_object_delete(&(sem->parent.parent));
  275. return RT_EOK;
  276. }
  277. RTM_EXPORT(rt_sem_delete);
  278. #endif
  279. /**
  280. * This function will take a semaphore, if the semaphore is unavailable, the
  281. * thread shall wait for a specified time.
  282. *
  283. * @param sem the semaphore object
  284. * @param time the waiting time
  285. *
  286. * @return the error code
  287. */
  288. static rt_err_t _rt_sem_take(rt_sem_t sem, rt_int32_t time, int suspend_flag)
  289. {
  290. register rt_base_t temp;
  291. struct rt_thread *thread;
  292. rt_err_t ret;
  293. /* parameter check */
  294. RT_ASSERT(sem != RT_NULL);
  295. RT_ASSERT(rt_object_get_type(&sem->parent.parent) == RT_Object_Class_Semaphore);
  296. RT_OBJECT_HOOK_CALL(rt_object_trytake_hook, (&(sem->parent.parent)));
  297. /* disable interrupt */
  298. temp = rt_hw_interrupt_disable();
  299. RT_DEBUG_LOG(RT_DEBUG_IPC, ("thread %s take sem:%s, which value is: %d\n",
  300. rt_thread_self()->name,
  301. ((struct rt_object *)sem)->name,
  302. sem->value));
  303. if (sem->value > 0)
  304. {
  305. /* semaphore is available */
  306. sem->value --;
  307. /* enable interrupt */
  308. rt_hw_interrupt_enable(temp);
  309. }
  310. else
  311. {
  312. /* no waiting, return with timeout */
  313. if (time == 0)
  314. {
  315. rt_hw_interrupt_enable(temp);
  316. return -RT_ETIMEOUT;
  317. }
  318. else
  319. {
  320. /* current context checking */
  321. RT_DEBUG_IN_THREAD_CONTEXT;
  322. /* semaphore is unavailable, push to suspend list */
  323. /* get current thread */
  324. thread = rt_thread_self();
  325. /* reset thread error number */
  326. thread->error = -RT_EINTR;
  327. RT_DEBUG_LOG(RT_DEBUG_IPC, ("sem take: suspend thread - %s\n",
  328. thread->name));
  329. /* suspend thread */
  330. ret = rt_ipc_list_suspend(&(sem->parent.suspend_thread),
  331. thread,
  332. sem->parent.parent.flag,
  333. suspend_flag);
  334. if (ret != RT_EOK)
  335. {
  336. rt_hw_interrupt_enable(temp);
  337. return ret;
  338. }
  339. /* has waiting time, start thread timer */
  340. if (time > 0)
  341. {
  342. RT_DEBUG_LOG(RT_DEBUG_IPC, ("set thread:%s to timer list\n",
  343. thread->name));
  344. /* reset the timeout of thread timer and start it */
  345. rt_timer_control(&(thread->thread_timer),
  346. RT_TIMER_CTRL_SET_TIME,
  347. &time);
  348. rt_timer_start(&(thread->thread_timer));
  349. }
  350. /* enable interrupt */
  351. rt_hw_interrupt_enable(temp);
  352. /* do schedule */
  353. rt_schedule();
  354. if (thread->error != RT_EOK)
  355. {
  356. return thread->error;
  357. }
  358. }
  359. }
  360. RT_OBJECT_HOOK_CALL(rt_object_take_hook, (&(sem->parent.parent)));
  361. return RT_EOK;
  362. }
  363. rt_err_t rt_sem_take(rt_sem_t sem, rt_int32_t time)
  364. {
  365. return _rt_sem_take(sem, time, RT_UNINTERRUPTIBLE);
  366. }
  367. RTM_EXPORT(rt_sem_take);
  368. rt_err_t rt_sem_take_interruptible(rt_sem_t sem, rt_int32_t time)
  369. {
  370. return _rt_sem_take(sem, time, RT_INTERRUPTIBLE);
  371. }
  372. RTM_EXPORT(rt_sem_take_interruptible);
  373. rt_err_t rt_sem_take_killable(rt_sem_t sem, rt_int32_t time)
  374. {
  375. return _rt_sem_take(sem, time, RT_KILLABLE);
  376. }
  377. RTM_EXPORT(rt_sem_take_killable);
  378. /**
  379. * This function will try to take a semaphore and immediately return
  380. *
  381. * @param sem the semaphore object
  382. *
  383. * @return the error code
  384. */
  385. rt_err_t rt_sem_trytake(rt_sem_t sem)
  386. {
  387. return rt_sem_take(sem, 0);
  388. }
  389. RTM_EXPORT(rt_sem_trytake);
  390. /**
  391. * This function will release a semaphore, if there are threads suspended on
  392. * semaphore, it will be waked up.
  393. *
  394. * @param sem the semaphore object
  395. *
  396. * @return the error code
  397. */
  398. rt_err_t rt_sem_release(rt_sem_t sem)
  399. {
  400. register rt_base_t temp;
  401. register rt_bool_t need_schedule;
  402. /* parameter check */
  403. RT_ASSERT(sem != RT_NULL);
  404. RT_ASSERT(rt_object_get_type(&sem->parent.parent) == RT_Object_Class_Semaphore);
  405. RT_OBJECT_HOOK_CALL(rt_object_put_hook, (&(sem->parent.parent)));
  406. need_schedule = RT_FALSE;
  407. /* disable interrupt */
  408. temp = rt_hw_interrupt_disable();
  409. RT_DEBUG_LOG(RT_DEBUG_IPC, ("thread %s releases sem:%s, which value is: %d\n",
  410. rt_thread_self()->name,
  411. ((struct rt_object *)sem)->name,
  412. sem->value));
  413. if (!rt_list_isempty(&sem->parent.suspend_thread))
  414. {
  415. /* resume the suspended thread */
  416. rt_ipc_list_resume(&(sem->parent.suspend_thread));
  417. need_schedule = RT_TRUE;
  418. }
  419. else
  420. {
  421. if(sem->value < RT_SEM_VALUE_MAX)
  422. {
  423. sem->value ++; /* increase value */
  424. }
  425. else
  426. {
  427. rt_hw_interrupt_enable(temp); /* enable interrupt */
  428. return -RT_EFULL; /* value overflowed */
  429. }
  430. }
  431. /* enable interrupt */
  432. rt_hw_interrupt_enable(temp);
  433. /* resume a thread, re-schedule */
  434. if (need_schedule == RT_TRUE)
  435. rt_schedule();
  436. return RT_EOK;
  437. }
  438. RTM_EXPORT(rt_sem_release);
  439. /**
  440. * This function can get or set some extra attributions of a semaphore object.
  441. *
  442. * @param sem the semaphore object
  443. * @param cmd the execution command
  444. * @param arg the execution argument
  445. *
  446. * @return the error code
  447. */
  448. rt_err_t rt_sem_control(rt_sem_t sem, int cmd, void *arg)
  449. {
  450. rt_ubase_t level;
  451. /* parameter check */
  452. RT_ASSERT(sem != RT_NULL);
  453. RT_ASSERT(rt_object_get_type(&sem->parent.parent) == RT_Object_Class_Semaphore);
  454. if (cmd == RT_IPC_CMD_RESET)
  455. {
  456. rt_ubase_t value;
  457. /* get value */
  458. value = (rt_ubase_t)arg;
  459. /* disable interrupt */
  460. level = rt_hw_interrupt_disable();
  461. /* resume all waiting thread */
  462. rt_ipc_list_resume_all(&sem->parent.suspend_thread);
  463. /* set new value */
  464. sem->value = (rt_uint16_t)value;
  465. /* enable interrupt */
  466. rt_hw_interrupt_enable(level);
  467. rt_schedule();
  468. return RT_EOK;
  469. }
  470. return -RT_ERROR;
  471. }
  472. RTM_EXPORT(rt_sem_control);
  473. #endif /* end of RT_USING_SEMAPHORE */
  474. #ifdef RT_USING_MUTEX
  475. /**
  476. * This function will initialize a mutex and put it under control of resource
  477. * management.
  478. *
  479. * @param mutex the mutex object
  480. * @param name the name of mutex
  481. * @param flag the flag of mutex
  482. *
  483. * @return the operation status, RT_EOK on successful
  484. */
  485. rt_err_t rt_mutex_init(rt_mutex_t mutex, const char *name, rt_uint8_t flag)
  486. {
  487. /* parameter check */
  488. RT_ASSERT(mutex != RT_NULL);
  489. /* initialize object */
  490. rt_object_init(&(mutex->parent.parent), RT_Object_Class_Mutex, name);
  491. /* initialize ipc object */
  492. rt_ipc_object_init(&(mutex->parent));
  493. mutex->value = 1;
  494. mutex->owner = RT_NULL;
  495. mutex->original_priority = 0xFF;
  496. mutex->hold = 0;
  497. /* set flag */
  498. mutex->parent.parent.flag = flag;
  499. return RT_EOK;
  500. }
  501. RTM_EXPORT(rt_mutex_init);
  502. /**
  503. * This function will detach a mutex from resource management
  504. *
  505. * @param mutex the mutex object
  506. *
  507. * @return the operation status, RT_EOK on successful
  508. *
  509. * @see rt_mutex_delete
  510. */
  511. rt_err_t rt_mutex_detach(rt_mutex_t mutex)
  512. {
  513. /* parameter check */
  514. RT_ASSERT(mutex != RT_NULL);
  515. RT_ASSERT(rt_object_get_type(&mutex->parent.parent) == RT_Object_Class_Mutex);
  516. RT_ASSERT(rt_object_is_systemobject(&mutex->parent.parent));
  517. /* wakeup all suspended threads */
  518. rt_ipc_list_resume_all(&(mutex->parent.suspend_thread));
  519. /* detach semaphore object */
  520. rt_object_detach(&(mutex->parent.parent));
  521. return RT_EOK;
  522. }
  523. RTM_EXPORT(rt_mutex_detach);
  524. #ifdef RT_USING_HEAP
  525. /**
  526. * This function will create a mutex from system resource
  527. *
  528. * @param name the name of mutex
  529. * @param flag the flag of mutex
  530. *
  531. * @return the created mutex, RT_NULL on error happen
  532. *
  533. * @see rt_mutex_init
  534. */
  535. rt_mutex_t rt_mutex_create(const char *name, rt_uint8_t flag)
  536. {
  537. struct rt_mutex *mutex;
  538. RT_DEBUG_NOT_IN_INTERRUPT;
  539. /* allocate object */
  540. mutex = (rt_mutex_t)rt_object_allocate(RT_Object_Class_Mutex, name);
  541. if (mutex == RT_NULL)
  542. return mutex;
  543. /* initialize ipc object */
  544. rt_ipc_object_init(&(mutex->parent));
  545. mutex->value = 1;
  546. mutex->owner = RT_NULL;
  547. mutex->original_priority = 0xFF;
  548. mutex->hold = 0;
  549. /* set flag */
  550. mutex->parent.parent.flag = flag;
  551. return mutex;
  552. }
  553. RTM_EXPORT(rt_mutex_create);
  554. /**
  555. * This function will delete a mutex object and release the memory
  556. *
  557. * @param mutex the mutex object
  558. *
  559. * @return the error code
  560. *
  561. * @see rt_mutex_detach
  562. */
  563. rt_err_t rt_mutex_delete(rt_mutex_t mutex)
  564. {
  565. RT_DEBUG_NOT_IN_INTERRUPT;
  566. /* parameter check */
  567. RT_ASSERT(mutex != RT_NULL);
  568. RT_ASSERT(rt_object_get_type(&mutex->parent.parent) == RT_Object_Class_Mutex);
  569. RT_ASSERT(rt_object_is_systemobject(&mutex->parent.parent) == RT_FALSE);
  570. /* wakeup all suspended threads */
  571. rt_ipc_list_resume_all(&(mutex->parent.suspend_thread));
  572. /* delete mutex object */
  573. rt_object_delete(&(mutex->parent.parent));
  574. return RT_EOK;
  575. }
  576. RTM_EXPORT(rt_mutex_delete);
  577. #endif
  578. /**
  579. * This function will take a mutex, if the mutex is unavailable, the
  580. * thread shall wait for a specified time.
  581. *
  582. * @param mutex the mutex object
  583. * @param time the waiting time
  584. *
  585. * @return the error code
  586. */
  587. static rt_err_t _rt_mutex_take(rt_mutex_t mutex, rt_int32_t time, int suspend_flag)
  588. {
  589. register rt_base_t temp;
  590. struct rt_thread *thread;
  591. rt_err_t ret;
  592. /* this function must not be used in interrupt even if time = 0 */
  593. RT_DEBUG_IN_THREAD_CONTEXT;
  594. /* parameter check */
  595. RT_ASSERT(mutex != RT_NULL);
  596. RT_ASSERT(rt_object_get_type(&mutex->parent.parent) == RT_Object_Class_Mutex);
  597. /* get current thread */
  598. thread = rt_thread_self();
  599. /* disable interrupt */
  600. temp = rt_hw_interrupt_disable();
  601. RT_OBJECT_HOOK_CALL(rt_object_trytake_hook, (&(mutex->parent.parent)));
  602. RT_DEBUG_LOG(RT_DEBUG_IPC,
  603. ("mutex_take: current thread %s, mutex value: %d, hold: %d\n",
  604. thread->name, mutex->value, mutex->hold));
  605. /* reset thread error */
  606. thread->error = -RT_EINTR;
  607. if (mutex->owner == thread)
  608. {
  609. if(mutex->hold < RT_MUTEX_HOLD_MAX)
  610. {
  611. /* it's the same thread */
  612. mutex->hold ++;
  613. }
  614. else
  615. {
  616. rt_hw_interrupt_enable(temp); /* enable interrupt */
  617. return -RT_EFULL; /* value overflowed */
  618. }
  619. }
  620. else
  621. {
  622. /* The value of mutex is 1 in initial status. Therefore, if the
  623. * value is great than 0, it indicates the mutex is avaible.
  624. */
  625. if (mutex->value > 0)
  626. {
  627. /* mutex is available */
  628. mutex->value --;
  629. /* set mutex owner and original priority */
  630. mutex->owner = thread;
  631. mutex->original_priority = thread->current_priority;
  632. if(mutex->hold < RT_MUTEX_HOLD_MAX)
  633. {
  634. mutex->hold ++;
  635. }
  636. else
  637. {
  638. rt_hw_interrupt_enable(temp); /* enable interrupt */
  639. return -RT_EFULL; /* value overflowed */
  640. }
  641. }
  642. else
  643. {
  644. /* no waiting, return with timeout */
  645. if (time == 0)
  646. {
  647. /* set error as timeout */
  648. thread->error = -RT_ETIMEOUT;
  649. /* enable interrupt */
  650. rt_hw_interrupt_enable(temp);
  651. return -RT_ETIMEOUT;
  652. }
  653. else
  654. {
  655. /* mutex is unavailable, push to suspend list */
  656. RT_DEBUG_LOG(RT_DEBUG_IPC, ("mutex_take: suspend thread: %s\n",
  657. thread->name));
  658. /* change the owner thread priority of mutex */
  659. if (thread->current_priority < mutex->owner->current_priority)
  660. {
  661. /* change the owner thread priority */
  662. rt_thread_control(mutex->owner,
  663. RT_THREAD_CTRL_CHANGE_PRIORITY,
  664. &thread->current_priority);
  665. }
  666. /* suspend current thread */
  667. ret = rt_ipc_list_suspend(&(mutex->parent.suspend_thread),
  668. thread,
  669. mutex->parent.parent.flag,
  670. suspend_flag);
  671. if (ret != RT_EOK)
  672. {
  673. rt_hw_interrupt_enable(temp);
  674. return ret;
  675. }
  676. /* has waiting time, start thread timer */
  677. if (time > 0)
  678. {
  679. RT_DEBUG_LOG(RT_DEBUG_IPC,
  680. ("mutex_take: start the timer of thread:%s\n",
  681. thread->name));
  682. /* reset the timeout of thread timer and start it */
  683. rt_timer_control(&(thread->thread_timer),
  684. RT_TIMER_CTRL_SET_TIME,
  685. &time);
  686. rt_timer_start(&(thread->thread_timer));
  687. }
  688. /* enable interrupt */
  689. rt_hw_interrupt_enable(temp);
  690. /* do schedule */
  691. rt_schedule();
  692. if (thread->error != RT_EOK)
  693. {
  694. /* return error */
  695. return thread->error;
  696. }
  697. else
  698. {
  699. /* the mutex is taken successfully. */
  700. /* disable interrupt */
  701. temp = rt_hw_interrupt_disable();
  702. }
  703. }
  704. }
  705. }
  706. /* enable interrupt */
  707. rt_hw_interrupt_enable(temp);
  708. RT_OBJECT_HOOK_CALL(rt_object_take_hook, (&(mutex->parent.parent)));
  709. return RT_EOK;
  710. }
  711. rt_err_t rt_mutex_take(rt_mutex_t mutex, rt_int32_t time)
  712. {
  713. return _rt_mutex_take(mutex, time, RT_UNINTERRUPTIBLE);
  714. }
  715. RTM_EXPORT(rt_mutex_take);
  716. rt_err_t rt_mutex_take_interruptible(rt_mutex_t mutex, rt_int32_t time)
  717. {
  718. return _rt_mutex_take(mutex, time, RT_INTERRUPTIBLE);
  719. }
  720. RTM_EXPORT(rt_mutex_take_interruptible);
  721. rt_err_t rt_mutex_take_killable(rt_mutex_t mutex, rt_int32_t time)
  722. {
  723. return _rt_mutex_take(mutex, time, RT_KILLABLE);
  724. }
  725. RTM_EXPORT(rt_mutex_take_killable);
  726. /**
  727. * This function will release a mutex, if there are threads suspended on mutex,
  728. * it will be waked up.
  729. *
  730. * @param mutex the mutex object
  731. *
  732. * @return the error code
  733. */
  734. rt_err_t rt_mutex_release(rt_mutex_t mutex)
  735. {
  736. register rt_base_t temp;
  737. struct rt_thread *thread;
  738. rt_bool_t need_schedule;
  739. /* parameter check */
  740. RT_ASSERT(mutex != RT_NULL);
  741. RT_ASSERT(rt_object_get_type(&mutex->parent.parent) == RT_Object_Class_Mutex);
  742. need_schedule = RT_FALSE;
  743. /* only thread could release mutex because we need test the ownership */
  744. RT_DEBUG_IN_THREAD_CONTEXT;
  745. /* get current thread */
  746. thread = rt_thread_self();
  747. /* disable interrupt */
  748. temp = rt_hw_interrupt_disable();
  749. RT_DEBUG_LOG(RT_DEBUG_IPC,
  750. ("mutex_release:current thread %s, mutex value: %d, hold: %d\n",
  751. thread->name, mutex->value, mutex->hold));
  752. RT_OBJECT_HOOK_CALL(rt_object_put_hook, (&(mutex->parent.parent)));
  753. /* mutex only can be released by owner */
  754. if (thread != mutex->owner)
  755. {
  756. thread->error = -RT_ERROR;
  757. /* enable interrupt */
  758. rt_hw_interrupt_enable(temp);
  759. return -RT_ERROR;
  760. }
  761. /* decrease hold */
  762. mutex->hold --;
  763. /* if no hold */
  764. if (mutex->hold == 0)
  765. {
  766. /* change the owner thread to original priority */
  767. if (mutex->original_priority != mutex->owner->current_priority)
  768. {
  769. rt_thread_control(mutex->owner,
  770. RT_THREAD_CTRL_CHANGE_PRIORITY,
  771. &(mutex->original_priority));
  772. }
  773. /* wakeup suspended thread */
  774. if (!rt_list_isempty(&mutex->parent.suspend_thread))
  775. {
  776. /* get suspended thread */
  777. thread = rt_list_entry(mutex->parent.suspend_thread.next,
  778. struct rt_thread,
  779. tlist);
  780. RT_DEBUG_LOG(RT_DEBUG_IPC, ("mutex_release: resume thread: %s\n",
  781. thread->name));
  782. /* set new owner and priority */
  783. mutex->owner = thread;
  784. mutex->original_priority = thread->current_priority;
  785. if(mutex->hold < RT_MUTEX_HOLD_MAX)
  786. {
  787. mutex->hold ++;
  788. }
  789. else
  790. {
  791. rt_hw_interrupt_enable(temp); /* enable interrupt */
  792. return -RT_EFULL; /* value overflowed */
  793. }
  794. /* resume thread */
  795. rt_ipc_list_resume(&(mutex->parent.suspend_thread));
  796. need_schedule = RT_TRUE;
  797. }
  798. else
  799. {
  800. if(mutex->value < RT_MUTEX_VALUE_MAX)
  801. {
  802. /* increase value */
  803. mutex->value ++;
  804. }
  805. else
  806. {
  807. rt_hw_interrupt_enable(temp); /* enable interrupt */
  808. return -RT_EFULL; /* value overflowed */
  809. }
  810. /* clear owner */
  811. mutex->owner = RT_NULL;
  812. mutex->original_priority = 0xff;
  813. }
  814. }
  815. /* enable interrupt */
  816. rt_hw_interrupt_enable(temp);
  817. /* perform a schedule */
  818. if (need_schedule == RT_TRUE)
  819. rt_schedule();
  820. return RT_EOK;
  821. }
  822. RTM_EXPORT(rt_mutex_release);
  823. /**
  824. * This function can get or set some extra attributions of a mutex object.
  825. *
  826. * @param mutex the mutex object
  827. * @param cmd the execution command
  828. * @param arg the execution argument
  829. *
  830. * @return the error code
  831. */
  832. rt_err_t rt_mutex_control(rt_mutex_t mutex, int cmd, void *arg)
  833. {
  834. /* parameter check */
  835. RT_ASSERT(mutex != RT_NULL);
  836. RT_ASSERT(rt_object_get_type(&mutex->parent.parent) == RT_Object_Class_Mutex);
  837. return -RT_ERROR;
  838. }
  839. RTM_EXPORT(rt_mutex_control);
  840. #endif /* end of RT_USING_MUTEX */
  841. #ifdef RT_USING_EVENT
  842. /**
  843. * This function will initialize an event and put it under control of resource
  844. * management.
  845. *
  846. * @param event the event object
  847. * @param name the name of event
  848. * @param flag the flag of event
  849. *
  850. * @return the operation status, RT_EOK on successful
  851. */
  852. rt_err_t rt_event_init(rt_event_t event, const char *name, rt_uint8_t flag)
  853. {
  854. /* parameter check */
  855. RT_ASSERT(event != RT_NULL);
  856. /* initialize object */
  857. rt_object_init(&(event->parent.parent), RT_Object_Class_Event, name);
  858. /* set parent flag */
  859. event->parent.parent.flag = flag;
  860. /* initialize ipc object */
  861. rt_ipc_object_init(&(event->parent));
  862. /* initialize event */
  863. event->set = 0;
  864. return RT_EOK;
  865. }
  866. RTM_EXPORT(rt_event_init);
  867. /**
  868. * This function will detach an event object from resource management
  869. *
  870. * @param event the event object
  871. *
  872. * @return the operation status, RT_EOK on successful
  873. */
  874. rt_err_t rt_event_detach(rt_event_t event)
  875. {
  876. /* parameter check */
  877. RT_ASSERT(event != RT_NULL);
  878. RT_ASSERT(rt_object_get_type(&event->parent.parent) == RT_Object_Class_Event);
  879. RT_ASSERT(rt_object_is_systemobject(&event->parent.parent));
  880. /* resume all suspended thread */
  881. rt_ipc_list_resume_all(&(event->parent.suspend_thread));
  882. /* detach event object */
  883. rt_object_detach(&(event->parent.parent));
  884. return RT_EOK;
  885. }
  886. RTM_EXPORT(rt_event_detach);
  887. #ifdef RT_USING_HEAP
  888. /**
  889. * This function will create an event object from system resource
  890. *
  891. * @param name the name of event
  892. * @param flag the flag of event
  893. *
  894. * @return the created event, RT_NULL on error happen
  895. */
  896. rt_event_t rt_event_create(const char *name, rt_uint8_t flag)
  897. {
  898. rt_event_t event;
  899. RT_DEBUG_NOT_IN_INTERRUPT;
  900. /* allocate object */
  901. event = (rt_event_t)rt_object_allocate(RT_Object_Class_Event, name);
  902. if (event == RT_NULL)
  903. return event;
  904. /* set parent */
  905. event->parent.parent.flag = flag;
  906. /* initialize ipc object */
  907. rt_ipc_object_init(&(event->parent));
  908. /* initialize event */
  909. event->set = 0;
  910. return event;
  911. }
  912. RTM_EXPORT(rt_event_create);
  913. /**
  914. * This function will delete an event object and release the memory
  915. *
  916. * @param event the event object
  917. *
  918. * @return the error code
  919. */
  920. rt_err_t rt_event_delete(rt_event_t event)
  921. {
  922. /* parameter check */
  923. RT_ASSERT(event != RT_NULL);
  924. RT_ASSERT(rt_object_get_type(&event->parent.parent) == RT_Object_Class_Event);
  925. RT_ASSERT(rt_object_is_systemobject(&event->parent.parent) == RT_FALSE);
  926. RT_DEBUG_NOT_IN_INTERRUPT;
  927. /* resume all suspended thread */
  928. rt_ipc_list_resume_all(&(event->parent.suspend_thread));
  929. /* delete event object */
  930. rt_object_delete(&(event->parent.parent));
  931. return RT_EOK;
  932. }
  933. RTM_EXPORT(rt_event_delete);
  934. #endif
  935. /**
  936. * This function will send an event to the event object, if there are threads
  937. * suspended on event object, it will be waked up.
  938. *
  939. * @param event the event object
  940. * @param set the event set
  941. *
  942. * @return the error code
  943. */
  944. rt_err_t rt_event_send(rt_event_t event, rt_uint32_t set)
  945. {
  946. struct rt_list_node *n;
  947. struct rt_thread *thread;
  948. register rt_ubase_t level;
  949. register rt_base_t status;
  950. rt_bool_t need_schedule;
  951. /* parameter check */
  952. RT_ASSERT(event != RT_NULL);
  953. RT_ASSERT(rt_object_get_type(&event->parent.parent) == RT_Object_Class_Event);
  954. if (set == 0)
  955. return -RT_ERROR;
  956. need_schedule = RT_FALSE;
  957. /* disable interrupt */
  958. level = rt_hw_interrupt_disable();
  959. /* set event */
  960. event->set |= set;
  961. RT_OBJECT_HOOK_CALL(rt_object_put_hook, (&(event->parent.parent)));
  962. if (!rt_list_isempty(&event->parent.suspend_thread))
  963. {
  964. /* search thread list to resume thread */
  965. n = event->parent.suspend_thread.next;
  966. while (n != &(event->parent.suspend_thread))
  967. {
  968. /* get thread */
  969. thread = rt_list_entry(n, struct rt_thread, tlist);
  970. status = -RT_ERROR;
  971. if (thread->event_info & RT_EVENT_FLAG_AND)
  972. {
  973. if ((thread->event_set & event->set) == thread->event_set)
  974. {
  975. /* received an AND event */
  976. status = RT_EOK;
  977. }
  978. }
  979. else if (thread->event_info & RT_EVENT_FLAG_OR)
  980. {
  981. if (thread->event_set & event->set)
  982. {
  983. /* save the received event set */
  984. thread->event_set = thread->event_set & event->set;
  985. /* received an OR event */
  986. status = RT_EOK;
  987. }
  988. }
  989. else
  990. {
  991. /* enable interrupt */
  992. rt_hw_interrupt_enable(level);
  993. return -RT_EINVAL;
  994. }
  995. /* move node to the next */
  996. n = n->next;
  997. /* condition is satisfied, resume thread */
  998. if (status == RT_EOK)
  999. {
  1000. /* clear event */
  1001. if (thread->event_info & RT_EVENT_FLAG_CLEAR)
  1002. event->set &= ~thread->event_set;
  1003. /* resume thread, and thread list breaks out */
  1004. rt_thread_resume(thread);
  1005. thread->error = RT_EOK;
  1006. /* need do a scheduling */
  1007. need_schedule = RT_TRUE;
  1008. }
  1009. }
  1010. }
  1011. /* enable interrupt */
  1012. rt_hw_interrupt_enable(level);
  1013. /* do a schedule */
  1014. if (need_schedule == RT_TRUE)
  1015. rt_schedule();
  1016. return RT_EOK;
  1017. }
  1018. RTM_EXPORT(rt_event_send);
  1019. /**
  1020. * This function will receive an event from event object, if the event is
  1021. * unavailable, the thread shall wait for a specified time.
  1022. *
  1023. * @param event the fast event object
  1024. * @param set the interested event set
  1025. * @param option the receive option, either RT_EVENT_FLAG_AND or
  1026. * RT_EVENT_FLAG_OR should be set.
  1027. * @param timeout the waiting time
  1028. * @param recved the received event, if you don't care, RT_NULL can be set.
  1029. *
  1030. * @return the error code
  1031. */
  1032. static rt_err_t _rt_event_recv(rt_event_t event,
  1033. rt_uint32_t set,
  1034. rt_uint8_t option,
  1035. rt_int32_t timeout,
  1036. rt_uint32_t *recved,
  1037. int suspend_flag)
  1038. {
  1039. struct rt_thread *thread;
  1040. register rt_ubase_t level;
  1041. register rt_base_t status;
  1042. rt_err_t ret;
  1043. RT_DEBUG_IN_THREAD_CONTEXT;
  1044. /* parameter check */
  1045. RT_ASSERT(event != RT_NULL);
  1046. RT_ASSERT(rt_object_get_type(&event->parent.parent) == RT_Object_Class_Event);
  1047. if (set == 0)
  1048. return -RT_ERROR;
  1049. /* initialize status */
  1050. status = -RT_ERROR;
  1051. /* get current thread */
  1052. thread = rt_thread_self();
  1053. /* reset thread error */
  1054. thread->error = -RT_EINTR;
  1055. RT_OBJECT_HOOK_CALL(rt_object_trytake_hook, (&(event->parent.parent)));
  1056. /* disable interrupt */
  1057. level = rt_hw_interrupt_disable();
  1058. /* check event set */
  1059. if (option & RT_EVENT_FLAG_AND)
  1060. {
  1061. if ((event->set & set) == set)
  1062. status = RT_EOK;
  1063. }
  1064. else if (option & RT_EVENT_FLAG_OR)
  1065. {
  1066. if (event->set & set)
  1067. status = RT_EOK;
  1068. }
  1069. else
  1070. {
  1071. /* either RT_EVENT_FLAG_AND or RT_EVENT_FLAG_OR should be set */
  1072. RT_ASSERT(0);
  1073. }
  1074. if (status == RT_EOK)
  1075. {
  1076. thread->error = RT_EOK;
  1077. /* set received event */
  1078. if (recved)
  1079. *recved = (event->set & set);
  1080. /* fill thread event info */
  1081. thread->event_set = (event->set & set);
  1082. thread->event_info = option;
  1083. /* received event */
  1084. if (option & RT_EVENT_FLAG_CLEAR)
  1085. event->set &= ~set;
  1086. }
  1087. else if (timeout == 0)
  1088. {
  1089. /* no waiting */
  1090. thread->error = -RT_ETIMEOUT;
  1091. /* enable interrupt */
  1092. rt_hw_interrupt_enable(level);
  1093. return -RT_ETIMEOUT;
  1094. }
  1095. else
  1096. {
  1097. /* fill thread event info */
  1098. thread->event_set = set;
  1099. thread->event_info = option;
  1100. /* put thread to suspended thread list */
  1101. ret = rt_ipc_list_suspend(&(event->parent.suspend_thread),
  1102. thread,
  1103. event->parent.parent.flag,
  1104. suspend_flag);
  1105. if (ret != RT_EOK)
  1106. {
  1107. rt_hw_interrupt_enable(level);
  1108. return ret;
  1109. }
  1110. /* if there is a waiting timeout, active thread timer */
  1111. if (timeout > 0)
  1112. {
  1113. /* reset the timeout of thread timer and start it */
  1114. rt_timer_control(&(thread->thread_timer),
  1115. RT_TIMER_CTRL_SET_TIME,
  1116. &timeout);
  1117. rt_timer_start(&(thread->thread_timer));
  1118. }
  1119. /* enable interrupt */
  1120. rt_hw_interrupt_enable(level);
  1121. /* do a schedule */
  1122. rt_schedule();
  1123. if (thread->error != RT_EOK)
  1124. {
  1125. /* return error */
  1126. return thread->error;
  1127. }
  1128. /* received an event, disable interrupt to protect */
  1129. level = rt_hw_interrupt_disable();
  1130. /* set received event */
  1131. if (recved)
  1132. *recved = thread->event_set;
  1133. }
  1134. /* enable interrupt */
  1135. rt_hw_interrupt_enable(level);
  1136. RT_OBJECT_HOOK_CALL(rt_object_take_hook, (&(event->parent.parent)));
  1137. return thread->error;
  1138. }
  1139. rt_err_t rt_event_recv(rt_event_t event,
  1140. rt_uint32_t set,
  1141. rt_uint8_t option,
  1142. rt_int32_t timeout,
  1143. rt_uint32_t *recved)
  1144. {
  1145. return _rt_event_recv(event, set, option, timeout, recved, RT_UNINTERRUPTIBLE);
  1146. }
  1147. RTM_EXPORT(rt_event_recv);
  1148. rt_err_t rt_event_recv_interruptible(rt_event_t event,
  1149. rt_uint32_t set,
  1150. rt_uint8_t option,
  1151. rt_int32_t timeout,
  1152. rt_uint32_t *recved)
  1153. {
  1154. return _rt_event_recv(event, set, option, timeout, recved, RT_INTERRUPTIBLE);
  1155. }
  1156. RTM_EXPORT(rt_event_recv_interruptible);
  1157. rt_err_t rt_event_recv_killable(rt_event_t event,
  1158. rt_uint32_t set,
  1159. rt_uint8_t option,
  1160. rt_int32_t timeout,
  1161. rt_uint32_t *recved)
  1162. {
  1163. return _rt_event_recv(event, set, option, timeout, recved, RT_KILLABLE);
  1164. }
  1165. RTM_EXPORT(rt_event_recv_killable);
  1166. /**
  1167. * This function can get or set some extra attributions of an event object.
  1168. *
  1169. * @param event the event object
  1170. * @param cmd the execution command
  1171. * @param arg the execution argument
  1172. *
  1173. * @return the error code
  1174. */
  1175. rt_err_t rt_event_control(rt_event_t event, int cmd, void *arg)
  1176. {
  1177. rt_ubase_t level;
  1178. /* parameter check */
  1179. RT_ASSERT(event != RT_NULL);
  1180. RT_ASSERT(rt_object_get_type(&event->parent.parent) == RT_Object_Class_Event);
  1181. if (cmd == RT_IPC_CMD_RESET)
  1182. {
  1183. /* disable interrupt */
  1184. level = rt_hw_interrupt_disable();
  1185. /* resume all waiting thread */
  1186. rt_ipc_list_resume_all(&event->parent.suspend_thread);
  1187. /* initialize event set */
  1188. event->set = 0;
  1189. /* enable interrupt */
  1190. rt_hw_interrupt_enable(level);
  1191. rt_schedule();
  1192. return RT_EOK;
  1193. }
  1194. return -RT_ERROR;
  1195. }
  1196. RTM_EXPORT(rt_event_control);
  1197. #endif /* end of RT_USING_EVENT */
  1198. #ifdef RT_USING_MAILBOX
  1199. /**
  1200. * This function will initialize a mailbox and put it under control of resource
  1201. * management.
  1202. *
  1203. * @param mb the mailbox object
  1204. * @param name the name of mailbox
  1205. * @param msgpool the begin address of buffer to save received mail
  1206. * @param size the size of mailbox
  1207. * @param flag the flag of mailbox
  1208. *
  1209. * @return the operation status, RT_EOK on successful
  1210. */
  1211. rt_err_t rt_mb_init(rt_mailbox_t mb,
  1212. const char *name,
  1213. void *msgpool,
  1214. rt_size_t size,
  1215. rt_uint8_t flag)
  1216. {
  1217. RT_ASSERT(mb != RT_NULL);
  1218. /* initialize object */
  1219. rt_object_init(&(mb->parent.parent), RT_Object_Class_MailBox, name);
  1220. /* set parent flag */
  1221. mb->parent.parent.flag = flag;
  1222. /* initialize ipc object */
  1223. rt_ipc_object_init(&(mb->parent));
  1224. /* initialize mailbox */
  1225. mb->msg_pool = (rt_ubase_t *)msgpool;
  1226. mb->size = size;
  1227. mb->entry = 0;
  1228. mb->in_offset = 0;
  1229. mb->out_offset = 0;
  1230. /* initialize an additional list of sender suspend thread */
  1231. rt_list_init(&(mb->suspend_sender_thread));
  1232. return RT_EOK;
  1233. }
  1234. RTM_EXPORT(rt_mb_init);
  1235. /**
  1236. * This function will detach a mailbox from resource management
  1237. *
  1238. * @param mb the mailbox object
  1239. *
  1240. * @return the operation status, RT_EOK on successful
  1241. */
  1242. rt_err_t rt_mb_detach(rt_mailbox_t mb)
  1243. {
  1244. /* parameter check */
  1245. RT_ASSERT(mb != RT_NULL);
  1246. RT_ASSERT(rt_object_get_type(&mb->parent.parent) == RT_Object_Class_MailBox);
  1247. RT_ASSERT(rt_object_is_systemobject(&mb->parent.parent));
  1248. /* resume all suspended thread */
  1249. rt_ipc_list_resume_all(&(mb->parent.suspend_thread));
  1250. /* also resume all mailbox private suspended thread */
  1251. rt_ipc_list_resume_all(&(mb->suspend_sender_thread));
  1252. /* detach mailbox object */
  1253. rt_object_detach(&(mb->parent.parent));
  1254. return RT_EOK;
  1255. }
  1256. RTM_EXPORT(rt_mb_detach);
  1257. #ifdef RT_USING_HEAP
  1258. /**
  1259. * This function will create a mailbox object from system resource
  1260. *
  1261. * @param name the name of mailbox
  1262. * @param size the size of mailbox
  1263. * @param flag the flag of mailbox
  1264. *
  1265. * @return the created mailbox, RT_NULL on error happen
  1266. */
  1267. rt_mailbox_t rt_mb_create(const char *name, rt_size_t size, rt_uint8_t flag)
  1268. {
  1269. rt_mailbox_t mb;
  1270. RT_DEBUG_NOT_IN_INTERRUPT;
  1271. /* allocate object */
  1272. mb = (rt_mailbox_t)rt_object_allocate(RT_Object_Class_MailBox, name);
  1273. if (mb == RT_NULL)
  1274. return mb;
  1275. /* set parent */
  1276. mb->parent.parent.flag = flag;
  1277. /* initialize ipc object */
  1278. rt_ipc_object_init(&(mb->parent));
  1279. /* initialize mailbox */
  1280. mb->size = size;
  1281. mb->msg_pool = (rt_ubase_t *)RT_KERNEL_MALLOC(mb->size * sizeof(rt_ubase_t));
  1282. if (mb->msg_pool == RT_NULL)
  1283. {
  1284. /* delete mailbox object */
  1285. rt_object_delete(&(mb->parent.parent));
  1286. return RT_NULL;
  1287. }
  1288. mb->entry = 0;
  1289. mb->in_offset = 0;
  1290. mb->out_offset = 0;
  1291. /* initialize an additional list of sender suspend thread */
  1292. rt_list_init(&(mb->suspend_sender_thread));
  1293. return mb;
  1294. }
  1295. RTM_EXPORT(rt_mb_create);
  1296. /**
  1297. * This function will delete a mailbox object and release the memory
  1298. *
  1299. * @param mb the mailbox object
  1300. *
  1301. * @return the error code
  1302. */
  1303. rt_err_t rt_mb_delete(rt_mailbox_t mb)
  1304. {
  1305. RT_DEBUG_NOT_IN_INTERRUPT;
  1306. /* parameter check */
  1307. RT_ASSERT(mb != RT_NULL);
  1308. RT_ASSERT(rt_object_get_type(&mb->parent.parent) == RT_Object_Class_MailBox);
  1309. RT_ASSERT(rt_object_is_systemobject(&mb->parent.parent) == RT_FALSE);
  1310. /* resume all suspended thread */
  1311. rt_ipc_list_resume_all(&(mb->parent.suspend_thread));
  1312. /* also resume all mailbox private suspended thread */
  1313. rt_ipc_list_resume_all(&(mb->suspend_sender_thread));
  1314. /* free mailbox pool */
  1315. RT_KERNEL_FREE(mb->msg_pool);
  1316. /* delete mailbox object */
  1317. rt_object_delete(&(mb->parent.parent));
  1318. return RT_EOK;
  1319. }
  1320. RTM_EXPORT(rt_mb_delete);
  1321. #endif
  1322. /**
  1323. * This function will send a mail to mailbox object. If the mailbox is full,
  1324. * current thread will be suspended until timeout.
  1325. *
  1326. * @param mb the mailbox object
  1327. * @param value the mail
  1328. * @param timeout the waiting time
  1329. *
  1330. * @return the error code
  1331. */
  1332. static rt_err_t _rt_mb_send_wait(rt_mailbox_t mb,
  1333. rt_ubase_t value,
  1334. rt_int32_t timeout,
  1335. int suspend_flag)
  1336. {
  1337. struct rt_thread *thread;
  1338. register rt_ubase_t temp;
  1339. rt_uint32_t tick_delta;
  1340. rt_err_t ret;
  1341. /* parameter check */
  1342. RT_ASSERT(mb != RT_NULL);
  1343. RT_ASSERT(rt_object_get_type(&mb->parent.parent) == RT_Object_Class_MailBox);
  1344. /* initialize delta tick */
  1345. tick_delta = 0;
  1346. /* get current thread */
  1347. thread = rt_thread_self();
  1348. RT_OBJECT_HOOK_CALL(rt_object_put_hook, (&(mb->parent.parent)));
  1349. /* disable interrupt */
  1350. temp = rt_hw_interrupt_disable();
  1351. /* for non-blocking call */
  1352. if (mb->entry == mb->size && timeout == 0)
  1353. {
  1354. rt_hw_interrupt_enable(temp);
  1355. return -RT_EFULL;
  1356. }
  1357. /* mailbox is full */
  1358. while (mb->entry == mb->size)
  1359. {
  1360. /* reset error number in thread */
  1361. thread->error = -RT_EINTR;
  1362. /* no waiting, return timeout */
  1363. if (timeout == 0)
  1364. {
  1365. /* enable interrupt */
  1366. rt_hw_interrupt_enable(temp);
  1367. return -RT_EFULL;
  1368. }
  1369. RT_DEBUG_IN_THREAD_CONTEXT;
  1370. /* suspend current thread */
  1371. ret = rt_ipc_list_suspend(&(mb->suspend_sender_thread),
  1372. thread,
  1373. mb->parent.parent.flag,
  1374. suspend_flag);
  1375. if (ret != RT_EOK)
  1376. {
  1377. rt_hw_interrupt_enable(temp);
  1378. return ret;
  1379. }
  1380. /* has waiting time, start thread timer */
  1381. if (timeout > 0)
  1382. {
  1383. /* get the start tick of timer */
  1384. tick_delta = rt_tick_get();
  1385. RT_DEBUG_LOG(RT_DEBUG_IPC, ("mb_send_wait: start timer of thread:%s\n",
  1386. thread->name));
  1387. /* reset the timeout of thread timer and start it */
  1388. rt_timer_control(&(thread->thread_timer),
  1389. RT_TIMER_CTRL_SET_TIME,
  1390. &timeout);
  1391. rt_timer_start(&(thread->thread_timer));
  1392. }
  1393. /* enable interrupt */
  1394. rt_hw_interrupt_enable(temp);
  1395. /* re-schedule */
  1396. rt_schedule();
  1397. /* resume from suspend state */
  1398. if (thread->error != RT_EOK)
  1399. {
  1400. /* return error */
  1401. return thread->error;
  1402. }
  1403. /* disable interrupt */
  1404. temp = rt_hw_interrupt_disable();
  1405. /* if it's not waiting forever and then re-calculate timeout tick */
  1406. if (timeout > 0)
  1407. {
  1408. tick_delta = rt_tick_get() - tick_delta;
  1409. timeout -= tick_delta;
  1410. if (timeout < 0)
  1411. timeout = 0;
  1412. }
  1413. }
  1414. /* set ptr */
  1415. mb->msg_pool[mb->in_offset] = value;
  1416. /* increase input offset */
  1417. ++ mb->in_offset;
  1418. if (mb->in_offset >= mb->size)
  1419. mb->in_offset = 0;
  1420. if(mb->entry < RT_MB_ENTRY_MAX)
  1421. {
  1422. /* increase message entry */
  1423. mb->entry ++;
  1424. }
  1425. else
  1426. {
  1427. rt_hw_interrupt_enable(temp); /* enable interrupt */
  1428. return -RT_EFULL; /* value overflowed */
  1429. }
  1430. /* resume suspended thread */
  1431. if (!rt_list_isempty(&mb->parent.suspend_thread))
  1432. {
  1433. rt_ipc_list_resume(&(mb->parent.suspend_thread));
  1434. /* enable interrupt */
  1435. rt_hw_interrupt_enable(temp);
  1436. rt_schedule();
  1437. return RT_EOK;
  1438. }
  1439. /* enable interrupt */
  1440. rt_hw_interrupt_enable(temp);
  1441. return RT_EOK;
  1442. }
  1443. rt_err_t rt_mb_send_wait(rt_mailbox_t mb,
  1444. rt_ubase_t value,
  1445. rt_int32_t timeout)
  1446. {
  1447. return _rt_mb_send_wait(mb, value, timeout, RT_UNINTERRUPTIBLE);
  1448. }
  1449. RTM_EXPORT(rt_mb_send_wait);
  1450. rt_err_t rt_mb_send_wait_interruptible(rt_mailbox_t mb,
  1451. rt_ubase_t value,
  1452. rt_int32_t timeout)
  1453. {
  1454. return _rt_mb_send_wait(mb, value, timeout, RT_INTERRUPTIBLE);
  1455. }
  1456. RTM_EXPORT(rt_mb_send_wait_interruptible);
  1457. rt_err_t rt_mb_send_wait_killable(rt_mailbox_t mb,
  1458. rt_ubase_t value,
  1459. rt_int32_t timeout)
  1460. {
  1461. return _rt_mb_send_wait(mb, value, timeout, RT_KILLABLE);
  1462. }
  1463. RTM_EXPORT(rt_mb_send_wait_killable);
  1464. /**
  1465. * This function will send a mail to mailbox object, if there are threads
  1466. * suspended on mailbox object, it will be waked up. This function will return
  1467. * immediately, if you want blocking send, use rt_mb_send_wait instead.
  1468. *
  1469. * @param mb the mailbox object
  1470. * @param value the mail
  1471. *
  1472. * @return the error code
  1473. */
  1474. rt_err_t rt_mb_send(rt_mailbox_t mb, rt_ubase_t value)
  1475. {
  1476. return rt_mb_send_wait(mb, value, 0);
  1477. }
  1478. RTM_EXPORT(rt_mb_send);
  1479. rt_err_t rt_mb_send_interruptible(rt_mailbox_t mb, rt_ubase_t value)
  1480. {
  1481. return rt_mb_send_wait_interruptible(mb, value, 0);
  1482. }
  1483. RTM_EXPORT(rt_mb_send_interruptible);
  1484. rt_err_t rt_mb_send_killable(rt_mailbox_t mb, rt_ubase_t value)
  1485. {
  1486. return rt_mb_send_wait_killable(mb, value, 0);
  1487. }
  1488. RTM_EXPORT(rt_mb_send_killable);
  1489. /**
  1490. * This function will receive a mail from mailbox object, if there is no mail
  1491. * in mailbox object, the thread shall wait for a specified time.
  1492. *
  1493. * @param mb the mailbox object
  1494. * @param value the received mail will be saved in
  1495. * @param timeout the waiting time
  1496. *
  1497. * @return the error code
  1498. */
  1499. static rt_err_t _rt_mb_recv(rt_mailbox_t mb, rt_ubase_t *value, rt_int32_t timeout, int suspend_flag)
  1500. {
  1501. struct rt_thread *thread;
  1502. register rt_ubase_t temp;
  1503. rt_uint32_t tick_delta;
  1504. rt_err_t ret;
  1505. /* parameter check */
  1506. RT_ASSERT(mb != RT_NULL);
  1507. RT_ASSERT(rt_object_get_type(&mb->parent.parent) == RT_Object_Class_MailBox);
  1508. /* initialize delta tick */
  1509. tick_delta = 0;
  1510. /* get current thread */
  1511. thread = rt_thread_self();
  1512. RT_OBJECT_HOOK_CALL(rt_object_trytake_hook, (&(mb->parent.parent)));
  1513. /* disable interrupt */
  1514. temp = rt_hw_interrupt_disable();
  1515. /* for non-blocking call */
  1516. if (mb->entry == 0 && timeout == 0)
  1517. {
  1518. rt_hw_interrupt_enable(temp);
  1519. return -RT_ETIMEOUT;
  1520. }
  1521. /* mailbox is empty */
  1522. while (mb->entry == 0)
  1523. {
  1524. /* reset error number in thread */
  1525. thread->error = -RT_EINTR;
  1526. /* no waiting, return timeout */
  1527. if (timeout == 0)
  1528. {
  1529. /* enable interrupt */
  1530. rt_hw_interrupt_enable(temp);
  1531. thread->error = -RT_ETIMEOUT;
  1532. return -RT_ETIMEOUT;
  1533. }
  1534. RT_DEBUG_IN_THREAD_CONTEXT;
  1535. /* suspend current thread */
  1536. ret = rt_ipc_list_suspend(&(mb->parent.suspend_thread),
  1537. thread,
  1538. mb->parent.parent.flag,
  1539. suspend_flag);
  1540. if (ret != RT_EOK)
  1541. {
  1542. rt_hw_interrupt_enable(temp);
  1543. return ret;
  1544. }
  1545. /* has waiting time, start thread timer */
  1546. if (timeout > 0)
  1547. {
  1548. /* get the start tick of timer */
  1549. tick_delta = rt_tick_get();
  1550. RT_DEBUG_LOG(RT_DEBUG_IPC, ("mb_recv: start timer of thread:%s\n",
  1551. thread->name));
  1552. /* reset the timeout of thread timer and start it */
  1553. rt_timer_control(&(thread->thread_timer),
  1554. RT_TIMER_CTRL_SET_TIME,
  1555. &timeout);
  1556. rt_timer_start(&(thread->thread_timer));
  1557. }
  1558. /* enable interrupt */
  1559. rt_hw_interrupt_enable(temp);
  1560. /* re-schedule */
  1561. rt_schedule();
  1562. /* resume from suspend state */
  1563. if (thread->error != RT_EOK)
  1564. {
  1565. /* return error */
  1566. return thread->error;
  1567. }
  1568. /* disable interrupt */
  1569. temp = rt_hw_interrupt_disable();
  1570. /* if it's not waiting forever and then re-calculate timeout tick */
  1571. if (timeout > 0)
  1572. {
  1573. tick_delta = rt_tick_get() - tick_delta;
  1574. timeout -= tick_delta;
  1575. if (timeout < 0)
  1576. timeout = 0;
  1577. }
  1578. }
  1579. /* fill ptr */
  1580. *value = mb->msg_pool[mb->out_offset];
  1581. /* increase output offset */
  1582. ++ mb->out_offset;
  1583. if (mb->out_offset >= mb->size)
  1584. mb->out_offset = 0;
  1585. /* decrease message entry */
  1586. mb->entry --;
  1587. /* resume suspended thread */
  1588. if (!rt_list_isempty(&(mb->suspend_sender_thread)))
  1589. {
  1590. rt_ipc_list_resume(&(mb->suspend_sender_thread));
  1591. /* enable interrupt */
  1592. rt_hw_interrupt_enable(temp);
  1593. RT_OBJECT_HOOK_CALL(rt_object_take_hook, (&(mb->parent.parent)));
  1594. rt_schedule();
  1595. return RT_EOK;
  1596. }
  1597. /* enable interrupt */
  1598. rt_hw_interrupt_enable(temp);
  1599. RT_OBJECT_HOOK_CALL(rt_object_take_hook, (&(mb->parent.parent)));
  1600. return RT_EOK;
  1601. }
  1602. rt_err_t rt_mb_recv(rt_mailbox_t mb, rt_ubase_t *value, rt_int32_t timeout)
  1603. {
  1604. return _rt_mb_recv(mb, value, timeout, RT_UNINTERRUPTIBLE);
  1605. }
  1606. RTM_EXPORT(rt_mb_recv);
  1607. rt_err_t rt_mb_recv_interruptibale(rt_mailbox_t mb, rt_ubase_t *value, rt_int32_t timeout)
  1608. {
  1609. return _rt_mb_recv(mb, value, timeout, RT_INTERRUPTIBLE);
  1610. }
  1611. RTM_EXPORT(rt_mb_recv_interruptibale);
  1612. rt_err_t rt_mb_recv_killable(rt_mailbox_t mb, rt_ubase_t *value, rt_int32_t timeout)
  1613. {
  1614. return _rt_mb_recv(mb, value, timeout, RT_KILLABLE);
  1615. }
  1616. RTM_EXPORT(rt_mb_recv_killable);
  1617. /**
  1618. * This function can get or set some extra attributions of a mailbox object.
  1619. *
  1620. * @param mb the mailbox object
  1621. * @param cmd the execution command
  1622. * @param arg the execution argument
  1623. *
  1624. * @return the error code
  1625. */
  1626. rt_err_t rt_mb_control(rt_mailbox_t mb, int cmd, void *arg)
  1627. {
  1628. rt_ubase_t level;
  1629. /* parameter check */
  1630. RT_ASSERT(mb != RT_NULL);
  1631. RT_ASSERT(rt_object_get_type(&mb->parent.parent) == RT_Object_Class_MailBox);
  1632. if (cmd == RT_IPC_CMD_RESET)
  1633. {
  1634. /* disable interrupt */
  1635. level = rt_hw_interrupt_disable();
  1636. /* resume all waiting thread */
  1637. rt_ipc_list_resume_all(&(mb->parent.suspend_thread));
  1638. /* also resume all mailbox private suspended thread */
  1639. rt_ipc_list_resume_all(&(mb->suspend_sender_thread));
  1640. /* re-init mailbox */
  1641. mb->entry = 0;
  1642. mb->in_offset = 0;
  1643. mb->out_offset = 0;
  1644. /* enable interrupt */
  1645. rt_hw_interrupt_enable(level);
  1646. rt_schedule();
  1647. return RT_EOK;
  1648. }
  1649. return -RT_ERROR;
  1650. }
  1651. RTM_EXPORT(rt_mb_control);
  1652. #endif /* end of RT_USING_MAILBOX */
  1653. #ifdef RT_USING_MESSAGEQUEUE
  1654. struct rt_mq_message
  1655. {
  1656. struct rt_mq_message *next;
  1657. };
  1658. /**
  1659. * This function will initialize a message queue and put it under control of
  1660. * resource management.
  1661. *
  1662. * @param mq the message object
  1663. * @param name the name of message queue
  1664. * @param msgpool the beginning address of buffer to save messages
  1665. * @param msg_size the maximum size of message
  1666. * @param pool_size the size of buffer to save messages
  1667. * @param flag the flag of message queue
  1668. *
  1669. * @return the operation status, RT_EOK on successful
  1670. */
  1671. rt_err_t rt_mq_init(rt_mq_t mq,
  1672. const char *name,
  1673. void *msgpool,
  1674. rt_size_t msg_size,
  1675. rt_size_t pool_size,
  1676. rt_uint8_t flag)
  1677. {
  1678. struct rt_mq_message *head;
  1679. register rt_base_t temp;
  1680. /* parameter check */
  1681. RT_ASSERT(mq != RT_NULL);
  1682. /* initialize object */
  1683. rt_object_init(&(mq->parent.parent), RT_Object_Class_MessageQueue, name);
  1684. /* set parent flag */
  1685. mq->parent.parent.flag = flag;
  1686. /* initialize ipc object */
  1687. rt_ipc_object_init(&(mq->parent));
  1688. /* set message pool */
  1689. mq->msg_pool = msgpool;
  1690. /* get correct message size */
  1691. mq->msg_size = RT_ALIGN(msg_size, RT_ALIGN_SIZE);
  1692. mq->max_msgs = pool_size / (mq->msg_size + sizeof(struct rt_mq_message));
  1693. /* initialize message list */
  1694. mq->msg_queue_head = RT_NULL;
  1695. mq->msg_queue_tail = RT_NULL;
  1696. /* initialize message empty list */
  1697. mq->msg_queue_free = RT_NULL;
  1698. for (temp = 0; temp < mq->max_msgs; temp ++)
  1699. {
  1700. head = (struct rt_mq_message *)((rt_uint8_t *)mq->msg_pool +
  1701. temp * (mq->msg_size + sizeof(struct rt_mq_message)));
  1702. head->next = (struct rt_mq_message *)mq->msg_queue_free;
  1703. mq->msg_queue_free = head;
  1704. }
  1705. /* the initial entry is zero */
  1706. mq->entry = 0;
  1707. /* initialize an additional list of sender suspend thread */
  1708. rt_list_init(&(mq->suspend_sender_thread));
  1709. return RT_EOK;
  1710. }
  1711. RTM_EXPORT(rt_mq_init);
  1712. /**
  1713. * This function will detach a message queue object from resource management
  1714. *
  1715. * @param mq the message queue object
  1716. *
  1717. * @return the operation status, RT_EOK on successful
  1718. */
  1719. rt_err_t rt_mq_detach(rt_mq_t mq)
  1720. {
  1721. /* parameter check */
  1722. RT_ASSERT(mq != RT_NULL);
  1723. RT_ASSERT(rt_object_get_type(&mq->parent.parent) == RT_Object_Class_MessageQueue);
  1724. RT_ASSERT(rt_object_is_systemobject(&mq->parent.parent));
  1725. /* resume all suspended thread */
  1726. rt_ipc_list_resume_all(&mq->parent.suspend_thread);
  1727. /* also resume all message queue private suspended thread */
  1728. rt_ipc_list_resume_all(&(mq->suspend_sender_thread));
  1729. /* detach message queue object */
  1730. rt_object_detach(&(mq->parent.parent));
  1731. return RT_EOK;
  1732. }
  1733. RTM_EXPORT(rt_mq_detach);
  1734. #ifdef RT_USING_HEAP
  1735. /**
  1736. * This function will create a message queue object from system resource
  1737. *
  1738. * @param name the name of message queue
  1739. * @param msg_size the size of message
  1740. * @param max_msgs the maximum number of message in queue
  1741. * @param flag the flag of message queue
  1742. *
  1743. * @return the created message queue, RT_NULL on error happen
  1744. */
  1745. rt_mq_t rt_mq_create(const char *name,
  1746. rt_size_t msg_size,
  1747. rt_size_t max_msgs,
  1748. rt_uint8_t flag)
  1749. {
  1750. struct rt_messagequeue *mq;
  1751. struct rt_mq_message *head;
  1752. register rt_base_t temp;
  1753. RT_DEBUG_NOT_IN_INTERRUPT;
  1754. /* allocate object */
  1755. mq = (rt_mq_t)rt_object_allocate(RT_Object_Class_MessageQueue, name);
  1756. if (mq == RT_NULL)
  1757. return mq;
  1758. /* set parent */
  1759. mq->parent.parent.flag = flag;
  1760. /* initialize ipc object */
  1761. rt_ipc_object_init(&(mq->parent));
  1762. /* initialize message queue */
  1763. /* get correct message size */
  1764. mq->msg_size = RT_ALIGN(msg_size, RT_ALIGN_SIZE);
  1765. mq->max_msgs = max_msgs;
  1766. /* allocate message pool */
  1767. mq->msg_pool = RT_KERNEL_MALLOC((mq->msg_size + sizeof(struct rt_mq_message)) * mq->max_msgs);
  1768. if (mq->msg_pool == RT_NULL)
  1769. {
  1770. rt_object_delete(&(mq->parent.parent));
  1771. return RT_NULL;
  1772. }
  1773. /* initialize message list */
  1774. mq->msg_queue_head = RT_NULL;
  1775. mq->msg_queue_tail = RT_NULL;
  1776. /* initialize message empty list */
  1777. mq->msg_queue_free = RT_NULL;
  1778. for (temp = 0; temp < mq->max_msgs; temp ++)
  1779. {
  1780. head = (struct rt_mq_message *)((rt_uint8_t *)mq->msg_pool +
  1781. temp * (mq->msg_size + sizeof(struct rt_mq_message)));
  1782. head->next = (struct rt_mq_message *)mq->msg_queue_free;
  1783. mq->msg_queue_free = head;
  1784. }
  1785. /* the initial entry is zero */
  1786. mq->entry = 0;
  1787. /* initialize an additional list of sender suspend thread */
  1788. rt_list_init(&(mq->suspend_sender_thread));
  1789. return mq;
  1790. }
  1791. RTM_EXPORT(rt_mq_create);
  1792. /**
  1793. * This function will delete a message queue object and release the memory
  1794. *
  1795. * @param mq the message queue object
  1796. *
  1797. * @return the error code
  1798. */
  1799. rt_err_t rt_mq_delete(rt_mq_t mq)
  1800. {
  1801. RT_DEBUG_NOT_IN_INTERRUPT;
  1802. /* parameter check */
  1803. RT_ASSERT(mq != RT_NULL);
  1804. RT_ASSERT(rt_object_get_type(&mq->parent.parent) == RT_Object_Class_MessageQueue);
  1805. RT_ASSERT(rt_object_is_systemobject(&mq->parent.parent) == RT_FALSE);
  1806. /* resume all suspended thread */
  1807. rt_ipc_list_resume_all(&(mq->parent.suspend_thread));
  1808. /* also resume all message queue private suspended thread */
  1809. rt_ipc_list_resume_all(&(mq->suspend_sender_thread));
  1810. /* free message queue pool */
  1811. RT_KERNEL_FREE(mq->msg_pool);
  1812. /* delete message queue object */
  1813. rt_object_delete(&(mq->parent.parent));
  1814. return RT_EOK;
  1815. }
  1816. RTM_EXPORT(rt_mq_delete);
  1817. #endif
  1818. /**
  1819. * This function will send a message to message queue object. If the message queue is full,
  1820. * current thread will be suspended until timeout.
  1821. *
  1822. * @param mq the message queue object
  1823. * @param buffer the message
  1824. * @param size the size of buffer
  1825. * @param timeout the waiting time
  1826. *
  1827. * @return the error code
  1828. */
  1829. static rt_err_t _rt_mq_send_wait(rt_mq_t mq,
  1830. const void *buffer,
  1831. rt_size_t size,
  1832. rt_int32_t timeout,
  1833. int suspend_flag)
  1834. {
  1835. register rt_ubase_t temp;
  1836. struct rt_mq_message *msg;
  1837. rt_uint32_t tick_delta;
  1838. struct rt_thread *thread;
  1839. rt_err_t ret;
  1840. /* parameter check */
  1841. RT_ASSERT(mq != RT_NULL);
  1842. RT_ASSERT(rt_object_get_type(&mq->parent.parent) == RT_Object_Class_MessageQueue);
  1843. RT_ASSERT(buffer != RT_NULL);
  1844. RT_ASSERT(size != 0);
  1845. /* greater than one message size */
  1846. if (size > mq->msg_size)
  1847. return -RT_ERROR;
  1848. /* initialize delta tick */
  1849. tick_delta = 0;
  1850. /* get current thread */
  1851. thread = rt_thread_self();
  1852. RT_OBJECT_HOOK_CALL(rt_object_put_hook, (&(mq->parent.parent)));
  1853. /* disable interrupt */
  1854. temp = rt_hw_interrupt_disable();
  1855. /* get a free list, there must be an empty item */
  1856. msg = (struct rt_mq_message *)mq->msg_queue_free;
  1857. /* for non-blocking call */
  1858. if (msg == RT_NULL && timeout == 0)
  1859. {
  1860. /* enable interrupt */
  1861. rt_hw_interrupt_enable(temp);
  1862. return -RT_EFULL;
  1863. }
  1864. /* message queue is full */
  1865. while ((msg = mq->msg_queue_free) == RT_NULL)
  1866. {
  1867. /* reset error number in thread */
  1868. thread->error = -RT_EINTR;
  1869. /* no waiting, return timeout */
  1870. if (timeout == 0)
  1871. {
  1872. /* enable interrupt */
  1873. rt_hw_interrupt_enable(temp);
  1874. return -RT_EFULL;
  1875. }
  1876. RT_DEBUG_IN_THREAD_CONTEXT;
  1877. /* suspend current thread */
  1878. ret = rt_ipc_list_suspend(&(mq->suspend_sender_thread),
  1879. thread,
  1880. mq->parent.parent.flag,
  1881. suspend_flag);
  1882. if (ret != RT_EOK)
  1883. {
  1884. rt_hw_interrupt_enable(temp);
  1885. return ret;
  1886. }
  1887. /* has waiting time, start thread timer */
  1888. if (timeout > 0)
  1889. {
  1890. /* get the start tick of timer */
  1891. tick_delta = rt_tick_get();
  1892. RT_DEBUG_LOG(RT_DEBUG_IPC, ("mq_send_wait: start timer of thread:%s\n",
  1893. thread->name));
  1894. /* reset the timeout of thread timer and start it */
  1895. rt_timer_control(&(thread->thread_timer),
  1896. RT_TIMER_CTRL_SET_TIME,
  1897. &timeout);
  1898. rt_timer_start(&(thread->thread_timer));
  1899. }
  1900. /* enable interrupt */
  1901. rt_hw_interrupt_enable(temp);
  1902. /* re-schedule */
  1903. rt_schedule();
  1904. /* resume from suspend state */
  1905. if (thread->error != RT_EOK)
  1906. {
  1907. /* return error */
  1908. return thread->error;
  1909. }
  1910. /* disable interrupt */
  1911. temp = rt_hw_interrupt_disable();
  1912. /* if it's not waiting forever and then re-calculate timeout tick */
  1913. if (timeout > 0)
  1914. {
  1915. tick_delta = rt_tick_get() - tick_delta;
  1916. timeout -= tick_delta;
  1917. if (timeout < 0)
  1918. timeout = 0;
  1919. }
  1920. }
  1921. /* move free list pointer */
  1922. mq->msg_queue_free = msg->next;
  1923. /* enable interrupt */
  1924. rt_hw_interrupt_enable(temp);
  1925. /* the msg is the new tailer of list, the next shall be NULL */
  1926. msg->next = RT_NULL;
  1927. /* copy buffer */
  1928. rt_memcpy(msg + 1, buffer, size);
  1929. /* disable interrupt */
  1930. temp = rt_hw_interrupt_disable();
  1931. /* link msg to message queue */
  1932. if (mq->msg_queue_tail != RT_NULL)
  1933. {
  1934. /* if the tail exists, */
  1935. ((struct rt_mq_message *)mq->msg_queue_tail)->next = msg;
  1936. }
  1937. /* set new tail */
  1938. mq->msg_queue_tail = msg;
  1939. /* if the head is empty, set head */
  1940. if (mq->msg_queue_head == RT_NULL)
  1941. mq->msg_queue_head = msg;
  1942. if(mq->entry < RT_MQ_ENTRY_MAX)
  1943. {
  1944. /* increase message entry */
  1945. mq->entry ++;
  1946. }
  1947. else
  1948. {
  1949. rt_hw_interrupt_enable(temp); /* enable interrupt */
  1950. return -RT_EFULL; /* value overflowed */
  1951. }
  1952. /* resume suspended thread */
  1953. if (!rt_list_isempty(&mq->parent.suspend_thread))
  1954. {
  1955. rt_ipc_list_resume(&(mq->parent.suspend_thread));
  1956. /* enable interrupt */
  1957. rt_hw_interrupt_enable(temp);
  1958. rt_schedule();
  1959. return RT_EOK;
  1960. }
  1961. /* enable interrupt */
  1962. rt_hw_interrupt_enable(temp);
  1963. return RT_EOK;
  1964. }
  1965. rt_err_t rt_mq_send_wait(rt_mq_t mq,
  1966. const void *buffer,
  1967. rt_size_t size,
  1968. rt_int32_t timeout)
  1969. {
  1970. return _rt_mq_send_wait(mq, buffer, size, timeout, RT_UNINTERRUPTIBLE);
  1971. }
  1972. RTM_EXPORT(rt_mq_send_wait)
  1973. rt_err_t rt_mq_send_wait_interruptible(rt_mq_t mq,
  1974. const void *buffer,
  1975. rt_size_t size,
  1976. rt_int32_t timeout)
  1977. {
  1978. return _rt_mq_send_wait(mq, buffer, size, timeout, RT_INTERRUPTIBLE);
  1979. }
  1980. RTM_EXPORT(rt_mq_send_wait_interruptible)
  1981. rt_err_t rt_mq_send_wait_killable(rt_mq_t mq,
  1982. const void *buffer,
  1983. rt_size_t size,
  1984. rt_int32_t timeout)
  1985. {
  1986. return _rt_mq_send_wait(mq, buffer, size, timeout, RT_KILLABLE);
  1987. }
  1988. RTM_EXPORT(rt_mq_send_wait_killable)
  1989. /**
  1990. * This function will send a message to message queue object, if there are
  1991. * threads suspended on message queue object, it will be waked up.
  1992. *
  1993. * @param mq the message queue object
  1994. * @param buffer the message
  1995. * @param size the size of buffer
  1996. *
  1997. * @return the error code
  1998. */
  1999. rt_err_t rt_mq_send(rt_mq_t mq, const void *buffer, rt_size_t size)
  2000. {
  2001. return rt_mq_send_wait(mq, buffer, size, 0);
  2002. }
  2003. RTM_EXPORT(rt_mq_send);
  2004. rt_err_t rt_mq_send_interrupt(rt_mq_t mq, const void *buffer, rt_size_t size)
  2005. {
  2006. return rt_mq_send_wait_interruptible(mq, buffer, size, 0);
  2007. }
  2008. RTM_EXPORT(rt_mq_send_interrupt);
  2009. rt_err_t rt_mq_send_killable(rt_mq_t mq, const void *buffer, rt_size_t size)
  2010. {
  2011. return rt_mq_send_wait_killable(mq, buffer, size, 0);
  2012. }
  2013. RTM_EXPORT(rt_mq_send_killable);
  2014. /**
  2015. * This function will send an urgent message to message queue object, which
  2016. * means the message will be inserted to the head of message queue. If there
  2017. * are threads suspended on message queue object, it will be waked up.
  2018. *
  2019. * @param mq the message queue object
  2020. * @param buffer the message
  2021. * @param size the size of buffer
  2022. *
  2023. * @return the error code
  2024. */
  2025. rt_err_t rt_mq_urgent(rt_mq_t mq, const void *buffer, rt_size_t size)
  2026. {
  2027. register rt_ubase_t temp;
  2028. struct rt_mq_message *msg;
  2029. /* parameter check */
  2030. RT_ASSERT(mq != RT_NULL);
  2031. RT_ASSERT(rt_object_get_type(&mq->parent.parent) == RT_Object_Class_MessageQueue);
  2032. RT_ASSERT(buffer != RT_NULL);
  2033. RT_ASSERT(size != 0);
  2034. /* greater than one message size */
  2035. if (size > mq->msg_size)
  2036. return -RT_ERROR;
  2037. RT_OBJECT_HOOK_CALL(rt_object_put_hook, (&(mq->parent.parent)));
  2038. /* disable interrupt */
  2039. temp = rt_hw_interrupt_disable();
  2040. /* get a free list, there must be an empty item */
  2041. msg = (struct rt_mq_message *)mq->msg_queue_free;
  2042. /* message queue is full */
  2043. if (msg == RT_NULL)
  2044. {
  2045. /* enable interrupt */
  2046. rt_hw_interrupt_enable(temp);
  2047. return -RT_EFULL;
  2048. }
  2049. /* move free list pointer */
  2050. mq->msg_queue_free = msg->next;
  2051. /* enable interrupt */
  2052. rt_hw_interrupt_enable(temp);
  2053. /* copy buffer */
  2054. rt_memcpy(msg + 1, buffer, size);
  2055. /* disable interrupt */
  2056. temp = rt_hw_interrupt_disable();
  2057. /* link msg to the beginning of message queue */
  2058. msg->next = (struct rt_mq_message *)mq->msg_queue_head;
  2059. mq->msg_queue_head = msg;
  2060. /* if there is no tail */
  2061. if (mq->msg_queue_tail == RT_NULL)
  2062. mq->msg_queue_tail = msg;
  2063. if(mq->entry < RT_MQ_ENTRY_MAX)
  2064. {
  2065. /* increase message entry */
  2066. mq->entry ++;
  2067. }
  2068. else
  2069. {
  2070. rt_hw_interrupt_enable(temp); /* enable interrupt */
  2071. return -RT_EFULL; /* value overflowed */
  2072. }
  2073. /* resume suspended thread */
  2074. if (!rt_list_isempty(&mq->parent.suspend_thread))
  2075. {
  2076. rt_ipc_list_resume(&(mq->parent.suspend_thread));
  2077. /* enable interrupt */
  2078. rt_hw_interrupt_enable(temp);
  2079. rt_schedule();
  2080. return RT_EOK;
  2081. }
  2082. /* enable interrupt */
  2083. rt_hw_interrupt_enable(temp);
  2084. return RT_EOK;
  2085. }
  2086. RTM_EXPORT(rt_mq_urgent);
  2087. /**
  2088. * This function will receive a message from message queue object, if there is
  2089. * no message in message queue object, the thread shall wait for a specified
  2090. * time.
  2091. *
  2092. * @param mq the message queue object
  2093. * @param buffer the received message will be saved in
  2094. * @param size the size of buffer
  2095. * @param timeout the waiting time
  2096. *
  2097. * @return the error code
  2098. */
  2099. static rt_err_t _rt_mq_recv(rt_mq_t mq,
  2100. void *buffer,
  2101. rt_size_t size,
  2102. rt_int32_t timeout,
  2103. int suspend_flag)
  2104. {
  2105. struct rt_thread *thread;
  2106. register rt_ubase_t temp;
  2107. struct rt_mq_message *msg;
  2108. rt_uint32_t tick_delta;
  2109. rt_err_t ret;
  2110. /* parameter check */
  2111. RT_ASSERT(mq != RT_NULL);
  2112. RT_ASSERT(rt_object_get_type(&mq->parent.parent) == RT_Object_Class_MessageQueue);
  2113. RT_ASSERT(buffer != RT_NULL);
  2114. RT_ASSERT(size != 0);
  2115. /* initialize delta tick */
  2116. tick_delta = 0;
  2117. /* get current thread */
  2118. thread = rt_thread_self();
  2119. RT_OBJECT_HOOK_CALL(rt_object_trytake_hook, (&(mq->parent.parent)));
  2120. /* disable interrupt */
  2121. temp = rt_hw_interrupt_disable();
  2122. /* for non-blocking call */
  2123. if (mq->entry == 0 && timeout == 0)
  2124. {
  2125. rt_hw_interrupt_enable(temp);
  2126. return -RT_ETIMEOUT;
  2127. }
  2128. /* message queue is empty */
  2129. while (mq->entry == 0)
  2130. {
  2131. RT_DEBUG_IN_THREAD_CONTEXT;
  2132. /* reset error number in thread */
  2133. thread->error = -RT_EINTR;
  2134. /* no waiting, return timeout */
  2135. if (timeout == 0)
  2136. {
  2137. /* enable interrupt */
  2138. rt_hw_interrupt_enable(temp);
  2139. thread->error = -RT_ETIMEOUT;
  2140. return -RT_ETIMEOUT;
  2141. }
  2142. /* suspend current thread */
  2143. ret = rt_ipc_list_suspend(&(mq->parent.suspend_thread),
  2144. thread,
  2145. mq->parent.parent.flag,
  2146. suspend_flag);
  2147. if (ret != RT_EOK)
  2148. {
  2149. rt_hw_interrupt_enable(temp);
  2150. return ret;
  2151. }
  2152. /* has waiting time, start thread timer */
  2153. if (timeout > 0)
  2154. {
  2155. /* get the start tick of timer */
  2156. tick_delta = rt_tick_get();
  2157. RT_DEBUG_LOG(RT_DEBUG_IPC, ("set thread:%s to timer list\n",
  2158. thread->name));
  2159. /* reset the timeout of thread timer and start it */
  2160. rt_timer_control(&(thread->thread_timer),
  2161. RT_TIMER_CTRL_SET_TIME,
  2162. &timeout);
  2163. rt_timer_start(&(thread->thread_timer));
  2164. }
  2165. /* enable interrupt */
  2166. rt_hw_interrupt_enable(temp);
  2167. /* re-schedule */
  2168. rt_schedule();
  2169. /* recv message */
  2170. if (thread->error != RT_EOK)
  2171. {
  2172. /* return error */
  2173. return thread->error;
  2174. }
  2175. /* disable interrupt */
  2176. temp = rt_hw_interrupt_disable();
  2177. /* if it's not waiting forever and then re-calculate timeout tick */
  2178. if (timeout > 0)
  2179. {
  2180. tick_delta = rt_tick_get() - tick_delta;
  2181. timeout -= tick_delta;
  2182. if (timeout < 0)
  2183. timeout = 0;
  2184. }
  2185. }
  2186. /* get message from queue */
  2187. msg = (struct rt_mq_message *)mq->msg_queue_head;
  2188. /* move message queue head */
  2189. mq->msg_queue_head = msg->next;
  2190. /* reach queue tail, set to NULL */
  2191. if (mq->msg_queue_tail == msg)
  2192. mq->msg_queue_tail = RT_NULL;
  2193. /* decrease message entry */
  2194. mq->entry --;
  2195. /* enable interrupt */
  2196. rt_hw_interrupt_enable(temp);
  2197. /* copy message */
  2198. rt_memcpy(buffer, msg + 1, size > mq->msg_size ? mq->msg_size : size);
  2199. /* disable interrupt */
  2200. temp = rt_hw_interrupt_disable();
  2201. /* put message to free list */
  2202. msg->next = (struct rt_mq_message *)mq->msg_queue_free;
  2203. mq->msg_queue_free = msg;
  2204. /* resume suspended thread */
  2205. if (!rt_list_isempty(&(mq->suspend_sender_thread)))
  2206. {
  2207. rt_ipc_list_resume(&(mq->suspend_sender_thread));
  2208. /* enable interrupt */
  2209. rt_hw_interrupt_enable(temp);
  2210. RT_OBJECT_HOOK_CALL(rt_object_take_hook, (&(mq->parent.parent)));
  2211. rt_schedule();
  2212. return RT_EOK;
  2213. }
  2214. /* enable interrupt */
  2215. rt_hw_interrupt_enable(temp);
  2216. RT_OBJECT_HOOK_CALL(rt_object_take_hook, (&(mq->parent.parent)));
  2217. return RT_EOK;
  2218. }
  2219. rt_err_t rt_mq_recv(rt_mq_t mq,
  2220. void *buffer,
  2221. rt_size_t size,
  2222. rt_int32_t timeout)
  2223. {
  2224. return _rt_mq_recv(mq, buffer, size, timeout, RT_UNINTERRUPTIBLE);
  2225. }
  2226. RTM_EXPORT(rt_mq_recv);
  2227. rt_err_t rt_mq_recv_interruptible(rt_mq_t mq,
  2228. void *buffer,
  2229. rt_size_t size,
  2230. rt_int32_t timeout)
  2231. {
  2232. return _rt_mq_recv(mq, buffer, size, timeout, RT_INTERRUPTIBLE);
  2233. }
  2234. RTM_EXPORT(rt_mq_recv_interruptible);
  2235. rt_err_t rt_mq_recv_killable(rt_mq_t mq,
  2236. void *buffer,
  2237. rt_size_t size,
  2238. rt_int32_t timeout)
  2239. {
  2240. return _rt_mq_recv(mq, buffer, size, timeout, RT_KILLABLE);
  2241. }
  2242. RTM_EXPORT(rt_mq_recv_killable);
  2243. /**
  2244. * This function can get or set some extra attributions of a message queue
  2245. * object.
  2246. *
  2247. * @param mq the message queue object
  2248. * @param cmd the execution command
  2249. * @param arg the execution argument
  2250. *
  2251. * @return the error code
  2252. */
  2253. rt_err_t rt_mq_control(rt_mq_t mq, int cmd, void *arg)
  2254. {
  2255. rt_ubase_t level;
  2256. struct rt_mq_message *msg;
  2257. /* parameter check */
  2258. RT_ASSERT(mq != RT_NULL);
  2259. RT_ASSERT(rt_object_get_type(&mq->parent.parent) == RT_Object_Class_MessageQueue);
  2260. if (cmd == RT_IPC_CMD_RESET)
  2261. {
  2262. /* disable interrupt */
  2263. level = rt_hw_interrupt_disable();
  2264. /* resume all waiting thread */
  2265. rt_ipc_list_resume_all(&mq->parent.suspend_thread);
  2266. /* also resume all message queue private suspended thread */
  2267. rt_ipc_list_resume_all(&(mq->suspend_sender_thread));
  2268. /* release all message in the queue */
  2269. while (mq->msg_queue_head != RT_NULL)
  2270. {
  2271. /* get message from queue */
  2272. msg = (struct rt_mq_message *)mq->msg_queue_head;
  2273. /* move message queue head */
  2274. mq->msg_queue_head = msg->next;
  2275. /* reach queue tail, set to NULL */
  2276. if (mq->msg_queue_tail == msg)
  2277. mq->msg_queue_tail = RT_NULL;
  2278. /* put message to free list */
  2279. msg->next = (struct rt_mq_message *)mq->msg_queue_free;
  2280. mq->msg_queue_free = msg;
  2281. }
  2282. /* clean entry */
  2283. mq->entry = 0;
  2284. /* enable interrupt */
  2285. rt_hw_interrupt_enable(level);
  2286. rt_schedule();
  2287. return RT_EOK;
  2288. }
  2289. return -RT_ERROR;
  2290. }
  2291. RTM_EXPORT(rt_mq_control);
  2292. #endif /* end of RT_USING_MESSAGEQUEUE */
  2293. /**@}*/