serial.c 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134
  1. /*
  2. * File : serial.c
  3. * This file is part of RT-Thread RTOS
  4. * COPYRIGHT (C) 2006 - 2012, RT-Thread Development Team
  5. *
  6. * This program is free software; you can redistribute it and/or modify
  7. * it under the terms of the GNU General Public License as published by
  8. * the Free Software Foundation; either version 2 of the License, or
  9. * (at your option) any later version.
  10. *
  11. * This program is distributed in the hope that it will be useful,
  12. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14. * GNU General Public License for more details.
  15. *
  16. * You should have received a copy of the GNU General Public License along
  17. * with this program; if not, write to the Free Software Foundation, Inc.,
  18. * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  19. *
  20. * Change Logs:
  21. * Date Author Notes
  22. * 2006-03-13 bernard first version
  23. * 2012-05-15 lgnq modified according bernard's implementation.
  24. * 2012-05-28 bernard code cleanup
  25. * 2012-11-23 bernard fix compiler warning.
  26. * 2013-02-20 bernard use RT_SERIAL_RB_BUFSZ to define
  27. * the size of ring buffer.
  28. * 2014-07-10 bernard rewrite serial framework
  29. * 2014-12-31 bernard use open_flag for poll_tx stream mode.
  30. * 2015-05-19 Quintin fix DMA tx mod tx_dma->activated flag !=RT_FALSE BUG
  31. * in open function.
  32. * 2015-11-10 bernard fix the poll rx issue when there is no data.
  33. * 2016-05-10 armink add fifo mode to DMA rx when serial->config.bufsz != 0.
  34. * 2017-01-19 aubr.cool prevent change serial rx bufsz when serial is opened.
  35. */
  36. #include <rthw.h>
  37. #include <rtthread.h>
  38. #include <rtdevice.h>
  39. // #define DEBUG_ENABLE
  40. #define DEBUG_LEVEL DBG_LOG
  41. #define DBG_SECTION_NAME "[UART]"
  42. #define DEBUG_COLOR
  43. #include <rtdbg.h>
  44. #ifdef RT_USING_POSIX
  45. #include <dfs_posix.h>
  46. #include <dfs_poll.h>
  47. #ifdef RT_USING_POSIX_TERMIOS
  48. #include <posix_termios.h>
  49. #endif
  50. /* it's possible the 'getc/putc' is defined by stdio.h in gcc/newlib. */
  51. #ifdef getc
  52. #undef getc
  53. #endif
  54. #ifdef putc
  55. #undef putc
  56. #endif
  57. static rt_err_t serial_fops_rx_ind(rt_device_t dev, rt_size_t size)
  58. {
  59. rt_wqueue_wakeup(&(dev->wait_queue), (void*)POLLIN);
  60. return RT_EOK;
  61. }
  62. /* fops for serial */
  63. static int serial_fops_open(struct dfs_fd *fd)
  64. {
  65. rt_err_t ret = 0;
  66. rt_uint16_t flags = 0;
  67. rt_device_t device;
  68. device = (rt_device_t)fd->data;
  69. RT_ASSERT(device != RT_NULL);
  70. switch (fd->flags & O_ACCMODE)
  71. {
  72. case O_RDONLY:
  73. dbg_log(DBG_LOG, "fops open: O_RDONLY!\n");
  74. flags = RT_DEVICE_FLAG_INT_RX | RT_DEVICE_FLAG_RDONLY;
  75. break;
  76. case O_WRONLY:
  77. dbg_log(DBG_LOG, "fops open: O_WRONLY!\n");
  78. flags = RT_DEVICE_FLAG_WRONLY;
  79. break;
  80. case O_RDWR:
  81. dbg_log(DBG_LOG, "fops open: O_RDWR!\n");
  82. flags = RT_DEVICE_FLAG_INT_RX | RT_DEVICE_FLAG_RDWR;
  83. break;
  84. default:
  85. dbg_log(DBG_ERROR, "fops open: unknown mode - %d!\n", fd->flags & O_ACCMODE);
  86. break;
  87. }
  88. if ((fd->flags & O_ACCMODE) != O_WRONLY)
  89. rt_device_set_rx_indicate(device, serial_fops_rx_ind);
  90. ret = rt_device_open(device, flags);
  91. if (ret == RT_EOK) return 0;
  92. return ret;
  93. }
  94. static int serial_fops_close(struct dfs_fd *fd)
  95. {
  96. rt_device_t device;
  97. device = (rt_device_t)fd->data;
  98. rt_device_set_rx_indicate(device, RT_NULL);
  99. rt_device_close(device);
  100. return 0;
  101. }
  102. static int serial_fops_ioctl(struct dfs_fd *fd, int cmd, void *args)
  103. {
  104. rt_device_t device;
  105. device = (rt_device_t)fd->data;
  106. switch (cmd)
  107. {
  108. case FIONREAD:
  109. break;
  110. case FIONWRITE:
  111. break;
  112. }
  113. return rt_device_control(device, cmd, args);
  114. }
  115. static int serial_fops_read(struct dfs_fd *fd, void *buf, size_t count)
  116. {
  117. int size = 0;
  118. rt_device_t device;
  119. device = (rt_device_t)fd->data;
  120. do
  121. {
  122. size = rt_device_read(device, -1, buf, count);
  123. if (size <= 0)
  124. {
  125. if (fd->flags & O_NONBLOCK)
  126. {
  127. size = -EAGAIN;
  128. break;
  129. }
  130. rt_wqueue_wait(&(device->wait_queue), 0, RT_WAITING_FOREVER);
  131. }
  132. }while (size <= 0);
  133. return size;
  134. }
  135. static int serial_fops_write(struct dfs_fd *fd, const void *buf, size_t count)
  136. {
  137. rt_device_t device;
  138. device = (rt_device_t)fd->data;
  139. return rt_device_write(device, -1, buf, count);
  140. }
  141. static int serial_fops_poll(struct dfs_fd *fd, struct rt_pollreq *req)
  142. {
  143. int mask = 0;
  144. int flags = 0;
  145. rt_device_t device;
  146. struct rt_serial_device *serial;
  147. device = (rt_device_t)fd->data;
  148. RT_ASSERT(device != RT_NULL);
  149. serial = (struct rt_serial_device *)device;
  150. /* only support POLLIN */
  151. flags = fd->flags & O_ACCMODE;
  152. if (flags == O_RDONLY || flags == O_RDWR)
  153. {
  154. rt_base_t level;
  155. struct rt_serial_rx_fifo* rx_fifo;
  156. rt_poll_add(&(device->wait_queue), req);
  157. rx_fifo = (struct rt_serial_rx_fifo*) serial->serial_rx;
  158. level = rt_hw_interrupt_disable();
  159. if (rx_fifo->get_index != rx_fifo->put_index)
  160. mask |= POLLIN;
  161. rt_hw_interrupt_enable(level);
  162. }
  163. return mask;
  164. }
  165. const static struct dfs_file_ops _serial_fops =
  166. {
  167. serial_fops_open,
  168. serial_fops_close,
  169. serial_fops_ioctl,
  170. serial_fops_read,
  171. serial_fops_write,
  172. RT_NULL, /* flush */
  173. RT_NULL, /* lseek */
  174. RT_NULL, /* getdents */
  175. serial_fops_poll,
  176. };
  177. #endif
  178. /*
  179. * Serial poll routines
  180. */
  181. rt_inline int _serial_poll_rx(struct rt_serial_device *serial, rt_uint8_t *data, int length)
  182. {
  183. int ch;
  184. int size;
  185. RT_ASSERT(serial != RT_NULL);
  186. size = length;
  187. while (length)
  188. {
  189. ch = serial->ops->getc(serial);
  190. if (ch == -1) break;
  191. *data = ch;
  192. data ++; length --;
  193. if (ch == '\n') break;
  194. }
  195. return size - length;
  196. }
  197. rt_inline int _serial_poll_tx(struct rt_serial_device *serial, const rt_uint8_t *data, int length)
  198. {
  199. int size;
  200. RT_ASSERT(serial != RT_NULL);
  201. size = length;
  202. while (length)
  203. {
  204. /*
  205. * to be polite with serial console add a line feed
  206. * to the carriage return character
  207. */
  208. if (*data == '\n' && (serial->parent.open_flag & RT_DEVICE_FLAG_STREAM))
  209. {
  210. serial->ops->putc(serial, '\r');
  211. }
  212. serial->ops->putc(serial, *data);
  213. ++ data;
  214. -- length;
  215. }
  216. return size - length;
  217. }
  218. /*
  219. * Serial interrupt routines
  220. */
  221. rt_inline int _serial_int_rx(struct rt_serial_device *serial, rt_uint8_t *data, int length)
  222. {
  223. int size;
  224. struct rt_serial_rx_fifo* rx_fifo;
  225. RT_ASSERT(serial != RT_NULL);
  226. size = length;
  227. rx_fifo = (struct rt_serial_rx_fifo*) serial->serial_rx;
  228. RT_ASSERT(rx_fifo != RT_NULL);
  229. /* read from software FIFO */
  230. while (length)
  231. {
  232. int ch;
  233. rt_base_t level;
  234. /* disable interrupt */
  235. level = rt_hw_interrupt_disable();
  236. if (rx_fifo->get_index != rx_fifo->put_index)
  237. {
  238. ch = rx_fifo->buffer[rx_fifo->get_index];
  239. rx_fifo->get_index += 1;
  240. if (rx_fifo->get_index >= serial->config.bufsz) rx_fifo->get_index = 0;
  241. }
  242. else
  243. {
  244. /* no data, enable interrupt and break out */
  245. rt_hw_interrupt_enable(level);
  246. break;
  247. }
  248. /* enable interrupt */
  249. rt_hw_interrupt_enable(level);
  250. *data = ch & 0xff;
  251. data ++; length --;
  252. }
  253. return size - length;
  254. }
  255. rt_inline int _serial_int_tx(struct rt_serial_device *serial, const rt_uint8_t *data, int length)
  256. {
  257. int size;
  258. struct rt_serial_tx_fifo *tx;
  259. RT_ASSERT(serial != RT_NULL);
  260. size = length;
  261. tx = (struct rt_serial_tx_fifo*) serial->serial_tx;
  262. RT_ASSERT(tx != RT_NULL);
  263. while (length)
  264. {
  265. if (serial->ops->putc(serial, *(char*)data) == -1)
  266. {
  267. rt_completion_wait(&(tx->completion), RT_WAITING_FOREVER);
  268. continue;
  269. }
  270. data ++; length --;
  271. }
  272. return size - length;
  273. }
  274. /**
  275. * Calculate DMA received data length.
  276. *
  277. * @param serial serial device
  278. *
  279. * @return length
  280. */
  281. static rt_size_t rt_dma_calc_recved_len(struct rt_serial_device *serial)
  282. {
  283. struct rt_serial_rx_fifo *rx_fifo = (struct rt_serial_rx_fifo *) serial->serial_rx;
  284. RT_ASSERT(rx_fifo != RT_NULL);
  285. if (rx_fifo->put_index > rx_fifo->get_index)
  286. return rx_fifo->put_index - rx_fifo->get_index;
  287. else if (rx_fifo->put_index < rx_fifo->get_index)
  288. return serial->config.bufsz - (rx_fifo->get_index - rx_fifo->put_index);
  289. else
  290. {
  291. if (rx_fifo->is_full)
  292. return serial->config.bufsz;
  293. else
  294. return 0;
  295. }
  296. }
  297. /**
  298. * Read data finish by DMA mode then update the gut index for receive fifo.
  299. *
  300. * @param serial serial device
  301. * @param len get data length for this operate
  302. */
  303. static void rt_dma_recv_update_get_index(struct rt_serial_device *serial, rt_size_t len)
  304. {
  305. struct rt_serial_rx_fifo *rx_fifo = (struct rt_serial_rx_fifo *) serial->serial_rx;
  306. RT_ASSERT(rx_fifo != RT_NULL);
  307. RT_ASSERT(len <= rt_dma_calc_recved_len(serial));
  308. if (rx_fifo->is_full && len != 0) rx_fifo->is_full = RT_FALSE;
  309. rx_fifo->get_index += len;
  310. if (rx_fifo->get_index > serial->config.bufsz)
  311. {
  312. rx_fifo->get_index %= serial->config.bufsz;
  313. }
  314. }
  315. /**
  316. * DMA received finish then update put index for receive fifo.
  317. *
  318. * @param serial serial device
  319. * @param len received length for this transmit
  320. */
  321. static void rt_dma_recv_update_put_index(struct rt_serial_device *serial, rt_size_t len)
  322. {
  323. struct rt_serial_rx_fifo *rx_fifo = (struct rt_serial_rx_fifo *)serial->serial_rx;
  324. RT_ASSERT(rx_fifo != RT_NULL);
  325. if (rx_fifo->get_index <= rx_fifo->put_index)
  326. {
  327. rx_fifo->put_index += len;
  328. /* beyond the fifo end */
  329. if (rx_fifo->put_index >= serial->config.bufsz)
  330. {
  331. rx_fifo->put_index %= serial->config.bufsz;
  332. /* force overwrite get index */
  333. if (rx_fifo->put_index >= rx_fifo->get_index)
  334. {
  335. rx_fifo->get_index = rx_fifo->put_index;
  336. rx_fifo->is_full = RT_TRUE;
  337. }
  338. }
  339. }
  340. else
  341. {
  342. rx_fifo->put_index += len;
  343. if (rx_fifo->put_index >= rx_fifo->get_index)
  344. {
  345. /* beyond the fifo end */
  346. if (rx_fifo->put_index >= serial->config.bufsz)
  347. {
  348. rx_fifo->put_index %= serial->config.bufsz;
  349. }
  350. /* force overwrite get index */
  351. rx_fifo->get_index = rx_fifo->put_index;
  352. rx_fifo->is_full = RT_TRUE;
  353. }
  354. }
  355. if (rx_fifo->get_index >= serial->config.bufsz) rx_fifo->get_index = 0;
  356. }
  357. /*
  358. * Serial DMA routines
  359. */
  360. rt_inline int _serial_dma_rx(struct rt_serial_device *serial, rt_uint8_t *data, int length)
  361. {
  362. rt_base_t level;
  363. RT_ASSERT((serial != RT_NULL) && (data != RT_NULL));
  364. level = rt_hw_interrupt_disable();
  365. if (serial->config.bufsz == 0)
  366. {
  367. int result = RT_EOK;
  368. struct rt_serial_rx_dma *rx_dma;
  369. rx_dma = (struct rt_serial_rx_dma*)serial->serial_rx;
  370. RT_ASSERT(rx_dma != RT_NULL);
  371. if (rx_dma->activated != RT_TRUE)
  372. {
  373. rx_dma->activated = RT_TRUE;
  374. RT_ASSERT(serial->ops->dma_transmit != RT_NULL);
  375. serial->ops->dma_transmit(serial, data, length, RT_SERIAL_DMA_RX);
  376. }
  377. else result = -RT_EBUSY;
  378. rt_hw_interrupt_enable(level);
  379. if (result == RT_EOK) return length;
  380. rt_set_errno(result);
  381. return 0;
  382. }
  383. else
  384. {
  385. struct rt_serial_rx_fifo *rx_fifo = (struct rt_serial_rx_fifo *) serial->serial_rx;
  386. rt_size_t recv_len = 0, fifo_recved_len = rt_dma_calc_recved_len(serial);
  387. RT_ASSERT(rx_fifo != RT_NULL);
  388. if (length < fifo_recved_len)
  389. recv_len = length;
  390. else
  391. recv_len = fifo_recved_len;
  392. if (rx_fifo->get_index + recv_len < serial->config.bufsz)
  393. rt_memcpy(data, rx_fifo->buffer + rx_fifo->get_index, recv_len);
  394. else
  395. {
  396. rt_memcpy(data, rx_fifo->buffer + rx_fifo->get_index,
  397. serial->config.bufsz - rx_fifo->get_index);
  398. rt_memcpy(data + serial->config.bufsz - rx_fifo->get_index, rx_fifo->buffer,
  399. recv_len + rx_fifo->get_index - serial->config.bufsz);
  400. }
  401. rt_dma_recv_update_get_index(serial, recv_len);
  402. rt_hw_interrupt_enable(level);
  403. return recv_len;
  404. }
  405. }
  406. rt_inline int _serial_dma_tx(struct rt_serial_device *serial, const rt_uint8_t *data, int length)
  407. {
  408. rt_base_t level;
  409. rt_err_t result;
  410. struct rt_serial_tx_dma *tx_dma;
  411. tx_dma = (struct rt_serial_tx_dma*)(serial->serial_tx);
  412. result = rt_data_queue_push(&(tx_dma->data_queue), data, length, RT_WAITING_FOREVER);
  413. if (result == RT_EOK)
  414. {
  415. level = rt_hw_interrupt_disable();
  416. if (tx_dma->activated != RT_TRUE)
  417. {
  418. tx_dma->activated = RT_TRUE;
  419. rt_hw_interrupt_enable(level);
  420. /* make a DMA transfer */
  421. serial->ops->dma_transmit(serial, (rt_uint8_t *)data, length, RT_SERIAL_DMA_TX);
  422. }
  423. else
  424. {
  425. rt_hw_interrupt_enable(level);
  426. }
  427. return length;
  428. }
  429. else
  430. {
  431. rt_set_errno(result);
  432. return 0;
  433. }
  434. }
  435. /* RT-Thread Device Interface */
  436. /*
  437. * This function initializes serial device.
  438. */
  439. static rt_err_t rt_serial_init(struct rt_device *dev)
  440. {
  441. rt_err_t result = RT_EOK;
  442. struct rt_serial_device *serial;
  443. RT_ASSERT(dev != RT_NULL);
  444. serial = (struct rt_serial_device *)dev;
  445. /* initialize rx/tx */
  446. serial->serial_rx = RT_NULL;
  447. serial->serial_tx = RT_NULL;
  448. /* apply configuration */
  449. if (serial->ops->configure)
  450. result = serial->ops->configure(serial, &serial->config);
  451. return result;
  452. }
  453. static rt_err_t rt_serial_open(struct rt_device *dev, rt_uint16_t oflag)
  454. {
  455. rt_uint16_t stream_flag = 0;
  456. struct rt_serial_device *serial;
  457. RT_ASSERT(dev != RT_NULL);
  458. serial = (struct rt_serial_device *)dev;
  459. dbg_log(DBG_LOG, "open serial device: 0x%08x with open flag: 0x%04x\n",
  460. dev, oflag);
  461. /* check device flag with the open flag */
  462. if ((oflag & RT_DEVICE_FLAG_DMA_RX) && !(dev->flag & RT_DEVICE_FLAG_DMA_RX))
  463. return -RT_EIO;
  464. if ((oflag & RT_DEVICE_FLAG_DMA_TX) && !(dev->flag & RT_DEVICE_FLAG_DMA_TX))
  465. return -RT_EIO;
  466. if ((oflag & RT_DEVICE_FLAG_INT_RX) && !(dev->flag & RT_DEVICE_FLAG_INT_RX))
  467. return -RT_EIO;
  468. if ((oflag & RT_DEVICE_FLAG_INT_TX) && !(dev->flag & RT_DEVICE_FLAG_INT_TX))
  469. return -RT_EIO;
  470. /* keep steam flag */
  471. if ((oflag & RT_DEVICE_FLAG_STREAM) || (dev->open_flag & RT_DEVICE_FLAG_STREAM))
  472. stream_flag = RT_DEVICE_FLAG_STREAM;
  473. /* get open flags */
  474. dev->open_flag = oflag & 0xff;
  475. /* initialize the Rx/Tx structure according to open flag */
  476. if (serial->serial_rx == RT_NULL)
  477. {
  478. if (oflag & RT_DEVICE_FLAG_DMA_RX)
  479. {
  480. if (serial->config.bufsz == 0) {
  481. struct rt_serial_rx_dma* rx_dma;
  482. rx_dma = (struct rt_serial_rx_dma*) rt_malloc (sizeof(struct rt_serial_rx_dma));
  483. RT_ASSERT(rx_dma != RT_NULL);
  484. rx_dma->activated = RT_FALSE;
  485. serial->serial_rx = rx_dma;
  486. } else {
  487. struct rt_serial_rx_fifo* rx_fifo;
  488. rx_fifo = (struct rt_serial_rx_fifo*) rt_malloc (sizeof(struct rt_serial_rx_fifo) +
  489. serial->config.bufsz);
  490. RT_ASSERT(rx_fifo != RT_NULL);
  491. rx_fifo->buffer = (rt_uint8_t*) (rx_fifo + 1);
  492. rt_memset(rx_fifo->buffer, 0, serial->config.bufsz);
  493. rx_fifo->put_index = 0;
  494. rx_fifo->get_index = 0;
  495. rx_fifo->is_full = RT_FALSE;
  496. serial->serial_rx = rx_fifo;
  497. /* configure fifo address and length to low level device */
  498. serial->ops->control(serial, RT_DEVICE_CTRL_CONFIG, (void *) RT_DEVICE_FLAG_DMA_RX);
  499. }
  500. dev->open_flag |= RT_DEVICE_FLAG_DMA_RX;
  501. }
  502. else if (oflag & RT_DEVICE_FLAG_INT_RX)
  503. {
  504. struct rt_serial_rx_fifo* rx_fifo;
  505. rx_fifo = (struct rt_serial_rx_fifo*) rt_malloc (sizeof(struct rt_serial_rx_fifo) +
  506. serial->config.bufsz);
  507. RT_ASSERT(rx_fifo != RT_NULL);
  508. rx_fifo->buffer = (rt_uint8_t*) (rx_fifo + 1);
  509. rt_memset(rx_fifo->buffer, 0, serial->config.bufsz);
  510. rx_fifo->put_index = 0;
  511. rx_fifo->get_index = 0;
  512. serial->serial_rx = rx_fifo;
  513. dev->open_flag |= RT_DEVICE_FLAG_INT_RX;
  514. /* configure low level device */
  515. serial->ops->control(serial, RT_DEVICE_CTRL_SET_INT, (void *)RT_DEVICE_FLAG_INT_RX);
  516. }
  517. else
  518. {
  519. serial->serial_rx = RT_NULL;
  520. }
  521. }
  522. if (serial->serial_tx == RT_NULL)
  523. {
  524. if (oflag & RT_DEVICE_FLAG_DMA_TX)
  525. {
  526. struct rt_serial_tx_dma* tx_dma;
  527. tx_dma = (struct rt_serial_tx_dma*) rt_malloc (sizeof(struct rt_serial_tx_dma));
  528. RT_ASSERT(tx_dma != RT_NULL);
  529. tx_dma->activated = RT_FALSE;
  530. rt_data_queue_init(&(tx_dma->data_queue), 8, 4, RT_NULL);
  531. serial->serial_tx = tx_dma;
  532. dev->open_flag |= RT_DEVICE_FLAG_DMA_TX;
  533. }
  534. else if (oflag & RT_DEVICE_FLAG_INT_TX)
  535. {
  536. struct rt_serial_tx_fifo *tx_fifo;
  537. tx_fifo = (struct rt_serial_tx_fifo*) rt_malloc(sizeof(struct rt_serial_tx_fifo));
  538. RT_ASSERT(tx_fifo != RT_NULL);
  539. rt_completion_init(&(tx_fifo->completion));
  540. serial->serial_tx = tx_fifo;
  541. dev->open_flag |= RT_DEVICE_FLAG_INT_TX;
  542. /* configure low level device */
  543. serial->ops->control(serial, RT_DEVICE_CTRL_SET_INT, (void *)RT_DEVICE_FLAG_INT_TX);
  544. }
  545. else
  546. {
  547. serial->serial_tx = RT_NULL;
  548. }
  549. }
  550. /* set stream flag */
  551. dev->open_flag |= stream_flag;
  552. return RT_EOK;
  553. }
  554. static rt_err_t rt_serial_close(struct rt_device *dev)
  555. {
  556. struct rt_serial_device *serial;
  557. RT_ASSERT(dev != RT_NULL);
  558. serial = (struct rt_serial_device *)dev;
  559. /* this device has more reference count */
  560. if (dev->ref_count > 1) return RT_EOK;
  561. if (dev->open_flag & RT_DEVICE_FLAG_INT_RX)
  562. {
  563. struct rt_serial_rx_fifo* rx_fifo;
  564. rx_fifo = (struct rt_serial_rx_fifo*)serial->serial_rx;
  565. RT_ASSERT(rx_fifo != RT_NULL);
  566. rt_free(rx_fifo);
  567. serial->serial_rx = RT_NULL;
  568. dev->open_flag &= ~RT_DEVICE_FLAG_INT_RX;
  569. /* configure low level device */
  570. serial->ops->control(serial, RT_DEVICE_CTRL_CLR_INT, (void*)RT_DEVICE_FLAG_INT_RX);
  571. }
  572. else if (dev->open_flag & RT_DEVICE_FLAG_DMA_RX)
  573. {
  574. if (serial->config.bufsz == 0) {
  575. struct rt_serial_rx_dma* rx_dma;
  576. rx_dma = (struct rt_serial_rx_dma*)serial->serial_rx;
  577. RT_ASSERT(rx_dma != RT_NULL);
  578. rt_free(rx_dma);
  579. } else {
  580. struct rt_serial_rx_fifo* rx_fifo;
  581. rx_fifo = (struct rt_serial_rx_fifo*)serial->serial_rx;
  582. RT_ASSERT(rx_fifo != RT_NULL);
  583. rt_free(rx_fifo);
  584. }
  585. /* configure low level device */
  586. serial->ops->control(serial, RT_DEVICE_CTRL_CLR_INT, (void *) RT_DEVICE_FLAG_DMA_RX);
  587. serial->serial_rx = RT_NULL;
  588. dev->open_flag &= ~RT_DEVICE_FLAG_DMA_RX;
  589. }
  590. if (dev->open_flag & RT_DEVICE_FLAG_INT_TX)
  591. {
  592. struct rt_serial_tx_fifo* tx_fifo;
  593. tx_fifo = (struct rt_serial_tx_fifo*)serial->serial_tx;
  594. RT_ASSERT(tx_fifo != RT_NULL);
  595. rt_free(tx_fifo);
  596. serial->serial_tx = RT_NULL;
  597. dev->open_flag &= ~RT_DEVICE_FLAG_INT_TX;
  598. /* configure low level device */
  599. serial->ops->control(serial, RT_DEVICE_CTRL_CLR_INT, (void*)RT_DEVICE_FLAG_INT_TX);
  600. }
  601. else if (dev->open_flag & RT_DEVICE_FLAG_DMA_TX)
  602. {
  603. struct rt_serial_tx_dma* tx_dma;
  604. tx_dma = (struct rt_serial_tx_dma*)serial->serial_tx;
  605. RT_ASSERT(tx_dma != RT_NULL);
  606. rt_free(tx_dma);
  607. serial->serial_tx = RT_NULL;
  608. dev->open_flag &= ~RT_DEVICE_FLAG_DMA_TX;
  609. }
  610. return RT_EOK;
  611. }
  612. static rt_size_t rt_serial_read(struct rt_device *dev,
  613. rt_off_t pos,
  614. void *buffer,
  615. rt_size_t size)
  616. {
  617. struct rt_serial_device *serial;
  618. RT_ASSERT(dev != RT_NULL);
  619. if (size == 0) return 0;
  620. serial = (struct rt_serial_device *)dev;
  621. if (dev->open_flag & RT_DEVICE_FLAG_INT_RX)
  622. {
  623. return _serial_int_rx(serial, buffer, size);
  624. }
  625. else if (dev->open_flag & RT_DEVICE_FLAG_DMA_RX)
  626. {
  627. return _serial_dma_rx(serial, buffer, size);
  628. }
  629. return _serial_poll_rx(serial, buffer, size);
  630. }
  631. static rt_size_t rt_serial_write(struct rt_device *dev,
  632. rt_off_t pos,
  633. const void *buffer,
  634. rt_size_t size)
  635. {
  636. struct rt_serial_device *serial;
  637. RT_ASSERT(dev != RT_NULL);
  638. if (size == 0) return 0;
  639. serial = (struct rt_serial_device *)dev;
  640. if (dev->open_flag & RT_DEVICE_FLAG_INT_TX)
  641. {
  642. return _serial_int_tx(serial, buffer, size);
  643. }
  644. else if (dev->open_flag & RT_DEVICE_FLAG_DMA_TX)
  645. {
  646. return _serial_dma_tx(serial, buffer, size);
  647. }
  648. else
  649. {
  650. return _serial_poll_tx(serial, buffer, size);
  651. }
  652. }
  653. #ifdef RT_USING_POSIX_TERMIOS
  654. struct speed_baudrate_item
  655. {
  656. speed_t speed;
  657. int baudrate;
  658. };
  659. const static struct speed_baudrate_item _tbl[] =
  660. {
  661. {B2400, BAUD_RATE_2400},
  662. {B4800, BAUD_RATE_4800},
  663. {B9600, BAUD_RATE_9600},
  664. {B19200, BAUD_RATE_19200},
  665. {B38400, BAUD_RATE_38400},
  666. {B57600, BAUD_RATE_57600},
  667. {B115200, BAUD_RATE_115200},
  668. {B230400, BAUD_RATE_230400},
  669. {B460800, BAUD_RATE_460800},
  670. {B921600, BAUD_RATE_921600},
  671. {B2000000, BAUD_RATE_2000000},
  672. {B3000000, BAUD_RATE_3000000},
  673. };
  674. static speed_t _get_speed(int baudrate)
  675. {
  676. int index;
  677. for (index = 0; index < sizeof(_tbl)/sizeof(_tbl[0]); index ++)
  678. {
  679. if (_tbl[index].baudrate == baudrate)
  680. return _tbl[index].speed;
  681. }
  682. return B0;
  683. }
  684. static int _get_baudrate(speed_t speed)
  685. {
  686. int index;
  687. for (index = 0; index < sizeof(_tbl)/sizeof(_tbl[0]); index ++)
  688. {
  689. if (_tbl[index].speed == speed)
  690. return _tbl[index].baudrate;
  691. }
  692. return 0;
  693. }
  694. #endif
  695. static rt_err_t rt_serial_control(struct rt_device *dev,
  696. int cmd,
  697. void *args)
  698. {
  699. rt_err_t ret = RT_EOK;
  700. struct rt_serial_device *serial;
  701. RT_ASSERT(dev != RT_NULL);
  702. serial = (struct rt_serial_device *)dev;
  703. switch (cmd)
  704. {
  705. case RT_DEVICE_CTRL_SUSPEND:
  706. /* suspend device */
  707. dev->flag |= RT_DEVICE_FLAG_SUSPENDED;
  708. break;
  709. case RT_DEVICE_CTRL_RESUME:
  710. /* resume device */
  711. dev->flag &= ~RT_DEVICE_FLAG_SUSPENDED;
  712. break;
  713. case RT_DEVICE_CTRL_CONFIG:
  714. if (args)
  715. {
  716. struct serial_configure *pconfig = (struct serial_configure *) args;
  717. if (pconfig->bufsz != serial->config.bufsz && serial->parent.ref_count)
  718. {
  719. /*can not change buffer size*/
  720. return RT_EBUSY;
  721. }
  722. /* set serial configure */
  723. serial->config = *pconfig;
  724. if (serial->parent.ref_count)
  725. {
  726. /* serial device has been opened, to configure it */
  727. serial->ops->configure(serial, (struct serial_configure *) args);
  728. }
  729. }
  730. break;
  731. #ifdef RT_USING_POSIX_TERMIOS
  732. case TCGETA:
  733. {
  734. struct termios *tio = (struct termios*)args;
  735. if (tio == RT_NULL) return -RT_EINVAL;
  736. tio->c_iflag = 0;
  737. tio->c_oflag = 0;
  738. tio->c_lflag = 0;
  739. /* update oflag for console device */
  740. if (rt_console_get_device() == dev)
  741. tio->c_oflag = OPOST | ONLCR;
  742. /* set cflag */
  743. tio->c_cflag = 0;
  744. if (serial->config.data_bits == DATA_BITS_5)
  745. tio->c_cflag = CS5;
  746. else if (serial->config.data_bits == DATA_BITS_6)
  747. tio->c_cflag = CS6;
  748. else if (serial->config.data_bits == DATA_BITS_7)
  749. tio->c_cflag = CS7;
  750. else if (serial->config.data_bits == DATA_BITS_8)
  751. tio->c_cflag = CS8;
  752. if (serial->config.stop_bits == STOP_BITS_2)
  753. tio->c_cflag |= CSTOPB;
  754. if (serial->config.parity == PARITY_EVEN)
  755. tio->c_cflag |= PARENB;
  756. else if (serial->config.parity == PARITY_ODD)
  757. tio->c_cflag |= (PARODD | PARENB);
  758. cfsetospeed(tio, _get_speed(serial->config.baud_rate));
  759. }
  760. break;
  761. case TCSETAW:
  762. case TCSETAF:
  763. case TCSETA:
  764. {
  765. int baudrate;
  766. struct serial_configure config;
  767. struct termios *tio = (struct termios*)args;
  768. if (tio == RT_NULL) return -RT_EINVAL;
  769. config = serial->config;
  770. baudrate = _get_baudrate(cfgetospeed(tio));
  771. config.baud_rate = baudrate;
  772. if (tio->c_cflag & CS6) config.data_bits = DATA_BITS_6;
  773. else if (tio->c_cflag & CS7) config.data_bits = DATA_BITS_7;
  774. else if (tio->c_cflag & CS8) config.data_bits = DATA_BITS_8;
  775. else config.data_bits = DATA_BITS_5;
  776. if (tio->c_cflag & CSTOPB) config.data_bits = STOP_BITS_2;
  777. else config.data_bits = STOP_BITS_1;
  778. if (tio->c_cflag & PARENB)
  779. {
  780. if (tio->c_cflag & PARODD) config.parity = PARITY_ODD;
  781. else config.parity = PARITY_EVEN;
  782. }
  783. else config.parity = PARITY_NONE;
  784. serial->ops->configure(serial, &config);
  785. }
  786. break;
  787. case TCFLSH:
  788. break;
  789. case TCXONC:
  790. break;
  791. #endif
  792. default :
  793. /* control device */
  794. ret = serial->ops->control(serial, cmd, args);
  795. break;
  796. }
  797. return ret;
  798. }
  799. /*
  800. * serial register
  801. */
  802. rt_err_t rt_hw_serial_register(struct rt_serial_device *serial,
  803. const char *name,
  804. rt_uint32_t flag,
  805. void *data)
  806. {
  807. rt_err_t ret;
  808. struct rt_device *device;
  809. RT_ASSERT(serial != RT_NULL);
  810. device = &(serial->parent);
  811. device->type = RT_Device_Class_Char;
  812. device->rx_indicate = RT_NULL;
  813. device->tx_complete = RT_NULL;
  814. device->init = rt_serial_init;
  815. device->open = rt_serial_open;
  816. device->close = rt_serial_close;
  817. device->read = rt_serial_read;
  818. device->write = rt_serial_write;
  819. device->control = rt_serial_control;
  820. device->user_data = data;
  821. /* register a character device */
  822. ret = rt_device_register(device, name, flag);
  823. #if defined(RT_USING_POSIX)
  824. /* set fops */
  825. device->fops = &_serial_fops;
  826. #endif
  827. return ret;
  828. }
  829. /* ISR for serial interrupt */
  830. void rt_hw_serial_isr(struct rt_serial_device *serial, int event)
  831. {
  832. switch (event & 0xff)
  833. {
  834. case RT_SERIAL_EVENT_RX_IND:
  835. {
  836. int ch = -1;
  837. rt_base_t level;
  838. struct rt_serial_rx_fifo* rx_fifo;
  839. /* interrupt mode receive */
  840. rx_fifo = (struct rt_serial_rx_fifo*)serial->serial_rx;
  841. RT_ASSERT(rx_fifo != RT_NULL);
  842. while (1)
  843. {
  844. ch = serial->ops->getc(serial);
  845. if (ch == -1) break;
  846. /* disable interrupt */
  847. level = rt_hw_interrupt_disable();
  848. rx_fifo->buffer[rx_fifo->put_index] = ch;
  849. rx_fifo->put_index += 1;
  850. if (rx_fifo->put_index >= serial->config.bufsz) rx_fifo->put_index = 0;
  851. /* if the next position is read index, discard this 'read char' */
  852. if (rx_fifo->put_index == rx_fifo->get_index)
  853. {
  854. rx_fifo->get_index += 1;
  855. if (rx_fifo->get_index >= serial->config.bufsz) rx_fifo->get_index = 0;
  856. }
  857. /* enable interrupt */
  858. rt_hw_interrupt_enable(level);
  859. }
  860. /* invoke callback */
  861. if (serial->parent.rx_indicate != RT_NULL)
  862. {
  863. rt_size_t rx_length;
  864. /* get rx length */
  865. level = rt_hw_interrupt_disable();
  866. rx_length = (rx_fifo->put_index >= rx_fifo->get_index)? (rx_fifo->put_index - rx_fifo->get_index):
  867. (serial->config.bufsz - (rx_fifo->get_index - rx_fifo->put_index));
  868. rt_hw_interrupt_enable(level);
  869. serial->parent.rx_indicate(&serial->parent, rx_length);
  870. }
  871. break;
  872. }
  873. case RT_SERIAL_EVENT_TX_DONE:
  874. {
  875. struct rt_serial_tx_fifo* tx_fifo;
  876. tx_fifo = (struct rt_serial_tx_fifo*)serial->serial_tx;
  877. rt_completion_done(&(tx_fifo->completion));
  878. break;
  879. }
  880. case RT_SERIAL_EVENT_TX_DMADONE:
  881. {
  882. const void *data_ptr;
  883. rt_size_t data_size;
  884. const void *last_data_ptr;
  885. struct rt_serial_tx_dma* tx_dma;
  886. tx_dma = (struct rt_serial_tx_dma*) serial->serial_tx;
  887. rt_data_queue_pop(&(tx_dma->data_queue), &last_data_ptr, &data_size, 0);
  888. if (rt_data_queue_peak(&(tx_dma->data_queue), &data_ptr, &data_size) == RT_EOK)
  889. {
  890. /* transmit next data node */
  891. tx_dma->activated = RT_TRUE;
  892. serial->ops->dma_transmit(serial, (rt_uint8_t *)data_ptr, data_size, RT_SERIAL_DMA_TX);
  893. }
  894. else
  895. {
  896. tx_dma->activated = RT_FALSE;
  897. }
  898. /* invoke callback */
  899. if (serial->parent.tx_complete != RT_NULL)
  900. {
  901. serial->parent.tx_complete(&serial->parent, (void*)last_data_ptr);
  902. }
  903. break;
  904. }
  905. case RT_SERIAL_EVENT_RX_DMADONE:
  906. {
  907. int length;
  908. rt_base_t level;
  909. /* get DMA rx length */
  910. length = (event & (~0xff)) >> 8;
  911. if (serial->config.bufsz == 0)
  912. {
  913. struct rt_serial_rx_dma* rx_dma;
  914. rx_dma = (struct rt_serial_rx_dma*) serial->serial_rx;
  915. RT_ASSERT(rx_dma != RT_NULL);
  916. RT_ASSERT(serial->parent.rx_indicate != RT_NULL);
  917. serial->parent.rx_indicate(&(serial->parent), length);
  918. rx_dma->activated = RT_FALSE;
  919. }
  920. else
  921. {
  922. /* disable interrupt */
  923. level = rt_hw_interrupt_disable();
  924. /* update fifo put index */
  925. rt_dma_recv_update_put_index(serial, length);
  926. /* calculate received total length */
  927. length = rt_dma_calc_recved_len(serial);
  928. /* enable interrupt */
  929. rt_hw_interrupt_enable(level);
  930. /* invoke callback */
  931. if (serial->parent.rx_indicate != RT_NULL)
  932. {
  933. serial->parent.rx_indicate(&(serial->parent), length);
  934. }
  935. }
  936. break;
  937. }
  938. }
  939. }