1*10465441SEvalZero #include <rtthread.h>
2*10465441SEvalZero #include <ymodem.h>
3*10465441SEvalZero #include <dfs_posix.h>
4*10465441SEvalZero #include <stdlib.h>
5*10465441SEvalZero
6*10465441SEvalZero #include <board.h>
7*10465441SEvalZero
8*10465441SEvalZero struct custom_ctx
9*10465441SEvalZero {
10*10465441SEvalZero struct rym_ctx parent;
11*10465441SEvalZero int fd;
12*10465441SEvalZero int flen;
13*10465441SEvalZero char fpath[256];
14*10465441SEvalZero };
15*10465441SEvalZero
_rym_bg(struct rym_ctx * ctx,rt_uint8_t * buf,rt_size_t len)16*10465441SEvalZero static enum rym_code _rym_bg(
17*10465441SEvalZero struct rym_ctx *ctx,
18*10465441SEvalZero rt_uint8_t *buf,
19*10465441SEvalZero rt_size_t len)
20*10465441SEvalZero {
21*10465441SEvalZero struct custom_ctx *cctx = (struct custom_ctx*)ctx;
22*10465441SEvalZero cctx->fpath[0] = '/';
23*10465441SEvalZero /* the buf should be the file name */
24*10465441SEvalZero strcpy(&(cctx->fpath[1]), (const char*)buf);
25*10465441SEvalZero cctx->fd = open(cctx->fpath, O_CREAT | O_WRONLY | O_TRUNC, 0);
26*10465441SEvalZero if (cctx->fd < 0)
27*10465441SEvalZero {
28*10465441SEvalZero rt_err_t err = rt_get_errno();
29*10465441SEvalZero rt_kprintf("error creating file: %d\n", err);
30*10465441SEvalZero rt_kprintf("abort transmission\n");
31*10465441SEvalZero return RYM_CODE_CAN;
32*10465441SEvalZero }
33*10465441SEvalZero
34*10465441SEvalZero cctx->flen = atoi((const char*)buf+strlen((const char*)buf)+1);
35*10465441SEvalZero if (cctx->flen == 0)
36*10465441SEvalZero cctx->flen = -1;
37*10465441SEvalZero return RYM_CODE_ACK;
38*10465441SEvalZero }
39*10465441SEvalZero
_rym_tof(struct rym_ctx * ctx,rt_uint8_t * buf,rt_size_t len)40*10465441SEvalZero static enum rym_code _rym_tof(
41*10465441SEvalZero struct rym_ctx *ctx,
42*10465441SEvalZero rt_uint8_t *buf,
43*10465441SEvalZero rt_size_t len)
44*10465441SEvalZero {
45*10465441SEvalZero struct custom_ctx *cctx = (struct custom_ctx*)ctx;
46*10465441SEvalZero RT_ASSERT(cctx->fd >= 0);
47*10465441SEvalZero if (cctx->flen == -1)
48*10465441SEvalZero {
49*10465441SEvalZero write(cctx->fd, buf, len);
50*10465441SEvalZero }
51*10465441SEvalZero else
52*10465441SEvalZero {
53*10465441SEvalZero int wlen = len > cctx->flen ? cctx->flen : len;
54*10465441SEvalZero write(cctx->fd, buf, wlen);
55*10465441SEvalZero cctx->flen -= wlen;
56*10465441SEvalZero }
57*10465441SEvalZero return RYM_CODE_ACK;
58*10465441SEvalZero }
59*10465441SEvalZero
_rym_end(struct rym_ctx * ctx,rt_uint8_t * buf,rt_size_t len)60*10465441SEvalZero static enum rym_code _rym_end(
61*10465441SEvalZero struct rym_ctx *ctx,
62*10465441SEvalZero rt_uint8_t *buf,
63*10465441SEvalZero rt_size_t len)
64*10465441SEvalZero {
65*10465441SEvalZero struct custom_ctx *cctx = (struct custom_ctx*)ctx;
66*10465441SEvalZero
67*10465441SEvalZero RT_ASSERT(cctx->fd >= 0);
68*10465441SEvalZero close(cctx->fd);
69*10465441SEvalZero cctx->fd = -1;
70*10465441SEvalZero
71*10465441SEvalZero return RYM_CODE_ACK;
72*10465441SEvalZero }
73*10465441SEvalZero
rym_write_to_file(rt_device_t idev)74*10465441SEvalZero rt_err_t rym_write_to_file(rt_device_t idev)
75*10465441SEvalZero {
76*10465441SEvalZero rt_err_t res;
77*10465441SEvalZero struct custom_ctx *ctx = rt_malloc(sizeof(*ctx));
78*10465441SEvalZero
79*10465441SEvalZero RT_ASSERT(idev);
80*10465441SEvalZero
81*10465441SEvalZero rt_kprintf("entering RYM mode\n");
82*10465441SEvalZero
83*10465441SEvalZero res = rym_recv_on_device(&ctx->parent, idev, RT_DEVICE_OFLAG_RDWR | RT_DEVICE_FLAG_INT_RX,
84*10465441SEvalZero _rym_bg, _rym_tof, _rym_end, 1000);
85*10465441SEvalZero
86*10465441SEvalZero /* there is no Ymodem traffic on the line so print out info. */
87*10465441SEvalZero rt_kprintf("leaving RYM mode with code %d\n", res);
88*10465441SEvalZero rt_kprintf("file %s has been created.\n", ctx->fpath);
89*10465441SEvalZero
90*10465441SEvalZero rt_free(ctx);
91*10465441SEvalZero
92*10465441SEvalZero return res;
93*10465441SEvalZero }
94*10465441SEvalZero
95*10465441SEvalZero #ifdef RT_USING_FINSH
96*10465441SEvalZero #include <finsh.h>
ry(char * dname)97*10465441SEvalZero rt_err_t ry(char *dname)
98*10465441SEvalZero {
99*10465441SEvalZero rt_err_t res;
100*10465441SEvalZero
101*10465441SEvalZero rt_device_t dev = rt_device_find(dname);
102*10465441SEvalZero if (!dev)
103*10465441SEvalZero {
104*10465441SEvalZero rt_kprintf("could not find device:%s\n", dname);
105*10465441SEvalZero return -RT_ERROR;
106*10465441SEvalZero }
107*10465441SEvalZero
108*10465441SEvalZero res = rym_write_to_file(dev);
109*10465441SEvalZero
110*10465441SEvalZero return res;
111*10465441SEvalZero }
112*10465441SEvalZero FINSH_FUNCTION_EXPORT(ry, receive files by ymodem protocol);
113*10465441SEvalZero #endif
114