ymodem.c 17.9 KB
Newer Older
G
Grissiom 已提交
1
/*
2 3 4 5
 * COPYRIGHT (C) 2012, Real-Thread Information Technology Ltd
 * All rights reserved
 *
 * SPDX-License-Identifier: Apache-2.0
G
Grissiom 已提交
6 7 8 9
 *
 * Change Logs:
 * Date           Author       Notes
 * 2013-04-14     Grissiom     initial implementation
10
 * 2019-12-09     Steven Liu   add YMODEM send protocol
G
Grissiom 已提交
11 12 13 14 15
 */

#include <rthw.h>
#include "ymodem.h"

16
#ifdef YMODEM_USING_CRC_TABLE 
17 18
static const rt_uint16_t ccitt_table[256] =
{
G
Grissiom 已提交
19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51
    0x0000, 0x1021, 0x2042, 0x3063, 0x4084, 0x50A5, 0x60C6, 0x70E7,
    0x8108, 0x9129, 0xA14A, 0xB16B, 0xC18C, 0xD1AD, 0xE1CE, 0xF1EF,
    0x1231, 0x0210, 0x3273, 0x2252, 0x52B5, 0x4294, 0x72F7, 0x62D6,
    0x9339, 0x8318, 0xB37B, 0xA35A, 0xD3BD, 0xC39C, 0xF3FF, 0xE3DE,
    0x2462, 0x3443, 0x0420, 0x1401, 0x64E6, 0x74C7, 0x44A4, 0x5485,
    0xA56A, 0xB54B, 0x8528, 0x9509, 0xE5EE, 0xF5CF, 0xC5AC, 0xD58D,
    0x3653, 0x2672, 0x1611, 0x0630, 0x76D7, 0x66F6, 0x5695, 0x46B4,
    0xB75B, 0xA77A, 0x9719, 0x8738, 0xF7DF, 0xE7FE, 0xD79D, 0xC7BC,
    0x48C4, 0x58E5, 0x6886, 0x78A7, 0x0840, 0x1861, 0x2802, 0x3823,
    0xC9CC, 0xD9ED, 0xE98E, 0xF9AF, 0x8948, 0x9969, 0xA90A, 0xB92B,
    0x5AF5, 0x4AD4, 0x7AB7, 0x6A96, 0x1A71, 0x0A50, 0x3A33, 0x2A12,
    0xDBFD, 0xCBDC, 0xFBBF, 0xEB9E, 0x9B79, 0x8B58, 0xBB3B, 0xAB1A,
    0x6CA6, 0x7C87, 0x4CE4, 0x5CC5, 0x2C22, 0x3C03, 0x0C60, 0x1C41,
    0xEDAE, 0xFD8F, 0xCDEC, 0xDDCD, 0xAD2A, 0xBD0B, 0x8D68, 0x9D49,
    0x7E97, 0x6EB6, 0x5ED5, 0x4EF4, 0x3E13, 0x2E32, 0x1E51, 0x0E70,
    0xFF9F, 0xEFBE, 0xDFDD, 0xCFFC, 0xBF1B, 0xAF3A, 0x9F59, 0x8F78,
    0x9188, 0x81A9, 0xB1CA, 0xA1EB, 0xD10C, 0xC12D, 0xF14E, 0xE16F,
    0x1080, 0x00A1, 0x30C2, 0x20E3, 0x5004, 0x4025, 0x7046, 0x6067,
    0x83B9, 0x9398, 0xA3FB, 0xB3DA, 0xC33D, 0xD31C, 0xE37F, 0xF35E,
    0x02B1, 0x1290, 0x22F3, 0x32D2, 0x4235, 0x5214, 0x6277, 0x7256,
    0xB5EA, 0xA5CB, 0x95A8, 0x8589, 0xF56E, 0xE54F, 0xD52C, 0xC50D,
    0x34E2, 0x24C3, 0x14A0, 0x0481, 0x7466, 0x6447, 0x5424, 0x4405,
    0xA7DB, 0xB7FA, 0x8799, 0x97B8, 0xE75F, 0xF77E, 0xC71D, 0xD73C,
    0x26D3, 0x36F2, 0x0691, 0x16B0, 0x6657, 0x7676, 0x4615, 0x5634,
    0xD94C, 0xC96D, 0xF90E, 0xE92F, 0x99C8, 0x89E9, 0xB98A, 0xA9AB,
    0x5844, 0x4865, 0x7806, 0x6827, 0x18C0, 0x08E1, 0x3882, 0x28A3,
    0xCB7D, 0xDB5C, 0xEB3F, 0xFB1E, 0x8BF9, 0x9BD8, 0xABBB, 0xBB9A,
    0x4A75, 0x5A54, 0x6A37, 0x7A16, 0x0AF1, 0x1AD0, 0x2AB3, 0x3A92,
    0xFD2E, 0xED0F, 0xDD6C, 0xCD4D, 0xBDAA, 0xAD8B, 0x9DE8, 0x8DC9,
    0x7C26, 0x6C07, 0x5C64, 0x4C45, 0x3CA2, 0x2C83, 0x1CE0, 0x0CC1,
    0xEF1F, 0xFF3E, 0xCF5D, 0xDF7C, 0xAF9B, 0xBFBA, 0x8FD9, 0x9FF8,
    0x6E17, 0x7E36, 0x4E55, 0x5E74, 0x2E93, 0x3EB2, 0x0ED1, 0x1EF0
};
52
static rt_uint16_t CRC16(unsigned char *q, int len)
G
Grissiom 已提交
53 54 55 56 57 58 59
{
    rt_uint16_t crc = 0;

    while (len-- > 0)
        crc = (crc << 8) ^ ccitt_table[((crc >> 8) ^ *q++) & 0xff];
    return crc;
}
60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83
#else
static rt_uint16_t CRC16(unsigned char *q, int len)
{
    rt_uint16_t crc;
    char i;

    crc = 0;
    while (--len >= 0)
    {
        crc = crc ^ (int) * q++ << 8;
        i = 8;
        do
        {
            if (crc & 0x8000)
                crc = crc << 1 ^ 0x1021;
            else
                crc = crc << 1;
        }
        while (--i);
    }

    return (crc);
}
#endif
G
Grissiom 已提交
84 85 86 87 88 89 90 91 92 93 94 95 96 97 98

// we could only use global varible because we could not use
// rt_device_t->user_data(it is used by the serial driver)...
static struct rym_ctx *_rym_the_ctx;

static rt_err_t _rym_rx_ind(rt_device_t dev, rt_size_t size)
{
    return rt_sem_release(&_rym_the_ctx->sem);
}

/* SOH/STX + seq + payload + crc */
#define _RYM_SOH_PKG_SZ (1+2+128+2)
#define _RYM_STX_PKG_SZ (1+2+1024+2)

static enum rym_code _rym_read_code(
99 100
    struct rym_ctx *ctx,
    rt_tick_t timeout)
G
Grissiom 已提交
101
{
G
Grissiom 已提交
102
    /* Fast path */
G
Grissiom 已提交
103
    if (rt_device_read(ctx->dev, 0, ctx->buf, 1) == 1)
104
        return (enum rym_code)(*ctx->buf);
G
Grissiom 已提交
105 106

    /* Slow path */
107 108
    do
    {
G
Grissiom 已提交
109 110 111 112 113 114 115 116 117
        rt_size_t rsz;

        /* No data yet, wait for one */
        if (rt_sem_take(&ctx->sem, timeout) != RT_EOK)
            return RYM_CODE_NONE;

        /* Try to read one */
        rsz = rt_device_read(ctx->dev, 0, ctx->buf, 1);
        if (rsz == 1)
118
            return (enum rym_code)(*ctx->buf);
119 120
    }
    while (1);
G
Grissiom 已提交
121 122 123 124
}

/* the caller should at least alloc _RYM_STX_PKG_SZ buffer */
static rt_size_t _rym_read_data(
125 126
    struct rym_ctx *ctx,
    rt_size_t len)
G
Grissiom 已提交
127 128 129 130 131 132 133 134
{
    /* we should already have had the code */
    rt_uint8_t *buf = ctx->buf + 1;
    rt_size_t readlen = 0;

    do
    {
        readlen += rt_device_read(ctx->dev,
135
                                  0, buf + readlen, len - readlen);
G
Grissiom 已提交
136 137
        if (readlen >= len)
            return readlen;
138 139
    }
    while (rt_sem_take(&ctx->sem, RYM_WAIT_CHR_TICK) == RT_EOK);
G
Grissiom 已提交
140 141 142 143

    return readlen;
}

144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169
static rt_err_t _rym_send_packet(
    struct rym_ctx *ctx,
    enum rym_code code,
    rt_uint8_t index)
{
    rt_uint16_t send_crc;
    rt_uint8_t index_inv = ~index;
    rt_size_t writelen = 0;

    send_crc = CRC16(ctx->buf + 3, _RYM_SOH_PKG_SZ - 5);
    ctx->buf[0] = code;
    ctx->buf[1] = index;
    ctx->buf[2] = index_inv;
    ctx->buf[131] = (rt_uint8_t)(send_crc >> 8);
    ctx->buf[132] = (rt_uint8_t)send_crc & 0xff;

    do
    {
        writelen += rt_device_write(ctx->dev, 0, ctx->buf + writelen,
                                    _RYM_SOH_PKG_SZ - writelen);
    }
    while (writelen < _RYM_SOH_PKG_SZ);

    return RT_EOK;
}

G
Grissiom 已提交
170 171 172 173 174 175
static rt_size_t _rym_putchar(struct rym_ctx *ctx, rt_uint8_t code)
{
    rt_device_write(ctx->dev, 0, &code, sizeof(code));
    return 1;
}

176 177 178 179 180 181 182 183 184 185 186
static rt_size_t _rym_getchar(struct rym_ctx *ctx)
{
    rt_uint8_t getc_ack;

    while (rt_device_read(ctx->dev, 0, &getc_ack, 1) != 1)
    {
        rt_sem_take(&ctx->sem, RT_WAITING_FOREVER);
    }
    return getc_ack;
}

G
Grissiom 已提交
187
static rt_err_t _rym_do_handshake(
188 189
    struct rym_ctx *ctx,
    int tm_sec)
G
Grissiom 已提交
190 191 192 193
{
    enum rym_code code;
    rt_size_t i;
    rt_uint16_t recv_crc, cal_crc;
194
    rt_size_t data_sz;
195
    rt_tick_t tick;
G
Grissiom 已提交
196 197 198 199 200 201 202

    ctx->stage = RYM_STAGE_ESTABLISHING;
    /* send C every second, so the sender could know we are waiting for it. */
    for (i = 0; i < tm_sec; i++)
    {
        _rym_putchar(ctx, RYM_CODE_C);
        code = _rym_read_code(ctx,
203
                              RYM_CHD_INTV_TICK);
G
Grissiom 已提交
204
        if (code == RYM_CODE_SOH)
205 206
        {
            data_sz = _RYM_SOH_PKG_SZ;
G
Grissiom 已提交
207
            break;
208
        }
209
        else if (code == RYM_CODE_STX)
210 211 212 213
        {
            data_sz = _RYM_STX_PKG_SZ;
            break;
        }
G
Grissiom 已提交
214 215
    }
    if (i == tm_sec)
216
    {
G
Grissiom 已提交
217
        return -RYM_ERR_TMO;
218
    }
G
Grissiom 已提交
219

220 221 222 223
    /* receive all data */
    i = 0;
    /* automatic exit after receiving specified length data, timeout: 100ms */
    tick = rt_tick_get();
224
    while (rt_tick_get() <= (tick + rt_tick_from_millisecond(100)) && i < (data_sz - 1))
225 226 227 228 229 230
    {
        i += _rym_read_data(ctx, data_sz - 1);
        rt_thread_mdelay(5);
    }

    if (i != (data_sz - 1))
G
Grissiom 已提交
231 232 233 234 235 236
        return -RYM_ERR_DSZ;

    /* sanity check */
    if (ctx->buf[1] != 0 || ctx->buf[2] != 0xFF)
        return -RYM_ERR_SEQ;

237 238
    recv_crc = (rt_uint16_t)(*(ctx->buf + data_sz - 2) << 8) | *(ctx->buf + data_sz - 1);
    cal_crc = CRC16(ctx->buf + 3, data_sz - 5);
G
Grissiom 已提交
239 240 241 242
    if (recv_crc != cal_crc)
        return -RYM_ERR_CRC;

    /* congratulations, check passed. */
243
    if (ctx->on_begin && ctx->on_begin(ctx, ctx->buf + 3, data_sz - 5) != RYM_CODE_ACK)
G
Grissiom 已提交
244 245 246 247 248
        return -RYM_ERR_CAN;

    return RT_EOK;
}

249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303
static rt_err_t _rym_do_send_handshake(
    struct rym_ctx *ctx,
    int tm_sec)
{
    enum rym_code code;
    rt_size_t i;
    rt_size_t data_sz;
    rt_uint8_t index = 0;
    rt_uint8_t getc_ack;

    ctx->stage = RYM_STAGE_ESTABLISHING;
    data_sz = _RYM_SOH_PKG_SZ;

    /* receive C every second */
    for (i = 0; i < tm_sec; i++)
    {
        code = _rym_read_code(ctx,
                              RYM_CHD_INTV_TICK);
        if (code == RYM_CODE_C)
        {
            break;
        }
    }
    if (i == tm_sec)
    {
        return -RYM_ERR_TMO;
    }

    /* congratulations, check passed. */
    if (ctx->on_begin && ctx->on_begin(ctx, ctx->buf + 3, data_sz - 5) != RYM_CODE_SOH)
        return -RYM_ERR_CODE;

    code = RYM_CODE_SOH;
    _rym_send_packet(ctx, code, index);

    rt_device_set_rx_indicate(ctx->dev, _rym_rx_ind);
    getc_ack = _rym_getchar(ctx);

    if (getc_ack != RYM_CODE_ACK)
    {
        return -RYM_ERR_ACK;
    }

    getc_ack = _rym_getchar(ctx);

    if (getc_ack != RYM_CODE_C)
    {
        return -RYM_ERR_ACK;
    }

    ctx->stage = RYM_STAGE_ESTABLISHED;

    return RT_EOK;
}

G
Grissiom 已提交
304
static rt_err_t _rym_trans_data(
305 306 307
    struct rym_ctx *ctx,
    rt_size_t data_sz,
    enum rym_code *code)
G
Grissiom 已提交
308
{
309
    const rt_size_t tsz = 2 + data_sz + 2;
G
Grissiom 已提交
310 311 312 313 314 315 316 317 318 319 320 321
    rt_uint16_t recv_crc;

    /* seq + data + crc */
    rt_size_t i = _rym_read_data(ctx, tsz);
    if (i != tsz)
        return -RYM_ERR_DSZ;

    if ((ctx->buf[1] + ctx->buf[2]) != 0xFF)
    {
        return -RYM_ERR_SEQ;
    }

322 323 324 325 326 327 328 329 330 331 332 333
    /* As we are sending C continuously, there is a chance that the
     * sender(remote) receive an C after sending the first handshake package.
     * So the sender will interpret it as NAK and re-send the package. So we
     * just ignore it and proceed. */
    if (ctx->stage == RYM_STAGE_ESTABLISHED && ctx->buf[1] == 0x00)
    {
        *code = RYM_CODE_NONE;
        return RT_EOK;
    }

    ctx->stage = RYM_STAGE_TRANSMITTING;

G
Grissiom 已提交
334
    /* sanity check */
335 336
    recv_crc = (rt_uint16_t)(*(ctx->buf + tsz - 1) << 8) | *(ctx->buf + tsz);
    if (recv_crc != CRC16(ctx->buf + 3, data_sz))
G
Grissiom 已提交
337 338 339 340
        return -RYM_ERR_CRC;

    /* congratulations, check passed. */
    if (ctx->on_data)
341
        *code = ctx->on_data(ctx, ctx->buf + 3, data_sz);
G
Grissiom 已提交
342 343 344 345 346 347 348 349 350 351 352 353 354 355 356
    else
        *code = RYM_CODE_ACK;
    return RT_EOK;
}

static rt_err_t _rym_do_trans(struct rym_ctx *ctx)
{
    _rym_putchar(ctx, RYM_CODE_ACK);
    _rym_putchar(ctx, RYM_CODE_C);
    ctx->stage = RYM_STAGE_ESTABLISHED;

    while (1)
    {
        rt_err_t err;
        enum rym_code code;
357
        rt_size_t data_sz, i;
G
Grissiom 已提交
358 359

        code = _rym_read_code(ctx,
360
                              RYM_WAIT_PKG_TICK);
G
Grissiom 已提交
361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381
        switch (code)
        {
        case RYM_CODE_SOH:
            data_sz = 128;
            break;
        case RYM_CODE_STX:
            data_sz = 1024;
            break;
        case RYM_CODE_EOT:
            return RT_EOK;
        default:
            return -RYM_ERR_CODE;
        };

        err = _rym_trans_data(ctx, data_sz, &code);
        if (err != RT_EOK)
            return err;
        switch (code)
        {
        case RYM_CODE_CAN:
            /* the spec require multiple CAN */
382 383
            for (i = 0; i < RYM_END_SESSION_SEND_CAN_NUM; i++)
            {
384 385
                _rym_putchar(ctx, RYM_CODE_CAN);
            }
G
Grissiom 已提交
386 387 388 389 390 391 392 393 394 395 396
            return -RYM_ERR_CAN;
        case RYM_CODE_ACK:
            _rym_putchar(ctx, RYM_CODE_ACK);
            break;
        default:
            // wrong code
            break;
        };
    }
}

397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431
static rt_err_t _rym_do_send_trans(struct rym_ctx *ctx)
{
    ctx->stage = RYM_STAGE_TRANSMITTING;
    enum rym_code code;
    rt_size_t data_sz;
    rt_uint32_t index = 1;
    rt_uint8_t getc_ack;

    data_sz = _RYM_SOH_PKG_SZ;

    while (1)
    {
        if (ctx->on_data && ctx->on_data(ctx, ctx->buf + 3, data_sz - 5) != RYM_CODE_SOH)
            return -RYM_ERR_CODE;

        code = RYM_CODE_SOH;

        _rym_send_packet(ctx, code, index);
        index++;
        rt_device_set_rx_indicate(ctx->dev, _rym_rx_ind);

        getc_ack = _rym_getchar(ctx);

        if (getc_ack != RYM_CODE_ACK)
        {
            return -RYM_ERR_ACK;
        }

        if (ctx->stage == RYM_STAGE_FINISHING)
            break;
    }

    return RT_EOK;
}

G
Grissiom 已提交
432 433 434 435 436
static rt_err_t _rym_do_fin(struct rym_ctx *ctx)
{
    enum rym_code code;
    rt_uint16_t recv_crc;
    rt_size_t i;
L
Lyn 已提交
437
    rt_size_t data_sz;
G
Grissiom 已提交
438 439 440 441 442

    ctx->stage = RYM_STAGE_FINISHING;
    /* we already got one EOT in the caller. invoke the callback if there is
     * one. */
    if (ctx->on_end)
443
        ctx->on_end(ctx, ctx->buf + 3, 128);
G
Grissiom 已提交
444 445 446 447 448 449 450 451 452 453

    _rym_putchar(ctx, RYM_CODE_NAK);
    code = _rym_read_code(ctx, RYM_WAIT_PKG_TICK);
    if (code != RYM_CODE_EOT)
        return -RYM_ERR_CODE;

    _rym_putchar(ctx, RYM_CODE_ACK);
    _rym_putchar(ctx, RYM_CODE_C);

    code = _rym_read_code(ctx, RYM_WAIT_PKG_TICK);
L
Lyn 已提交
454 455 456 457
    if (code == RYM_CODE_SOH)
    {
        data_sz = _RYM_SOH_PKG_SZ;
    }
458
    else if (code == RYM_CODE_STX)
L
Lyn 已提交
459 460 461 462
    {
        data_sz = _RYM_STX_PKG_SZ;
    }
    else
G
Grissiom 已提交
463 464
        return -RYM_ERR_CODE;

465 466
    i = _rym_read_data(ctx, _RYM_SOH_PKG_SZ - 1);
    if (i != (_RYM_SOH_PKG_SZ - 1))
G
Grissiom 已提交
467 468 469 470 471 472 473
        return -RYM_ERR_DSZ;

    /* sanity check
     */
    if (ctx->buf[1] != 0 || ctx->buf[2] != 0xFF)
        return -RYM_ERR_SEQ;

474 475
    recv_crc = (rt_uint16_t)(*(ctx->buf + _RYM_SOH_PKG_SZ - 2) << 8) | *(ctx->buf + _RYM_SOH_PKG_SZ - 1);
    if (recv_crc != CRC16(ctx->buf + 3, _RYM_SOH_PKG_SZ - 5))
G
Grissiom 已提交
476 477
        return -RYM_ERR_CRC;

L
Lyn 已提交
478 479 480
    /*next file transmission*/
    if (ctx->buf[3] != 0)
    {
481
        if (ctx->on_begin && ctx->on_begin(ctx, ctx->buf + 3, data_sz - 5) != RYM_CODE_ACK)
L
Lyn 已提交
482 483 484 485
            return -RYM_ERR_CAN;
        return RT_EOK;
    }

G
Grissiom 已提交
486 487 488 489 490 491 492 493 494
    /* congratulations, check passed. */
    ctx->stage = RYM_STAGE_FINISHED;

    /* put the last ACK */
    _rym_putchar(ctx, RYM_CODE_ACK);

    return RT_EOK;
}

495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539
static rt_err_t _rym_do_send_fin(struct rym_ctx *ctx)
{
    enum rym_code code;
    rt_size_t data_sz;
    rt_uint8_t index = 0;
    rt_uint8_t getc_ack;

    data_sz = _RYM_SOH_PKG_SZ;
    rt_device_set_rx_indicate(ctx->dev, _rym_rx_ind);

    _rym_putchar(ctx, RYM_CODE_EOT);
    getc_ack = _rym_getchar(ctx);

    if (getc_ack != RYM_CODE_NAK)
    {
        return -RYM_ERR_ACK;
    }

    _rym_putchar(ctx, RYM_CODE_EOT);
    getc_ack = _rym_getchar(ctx);

    if (getc_ack != RYM_CODE_ACK)
    {
        return -RYM_ERR_ACK;
    }

    getc_ack = _rym_getchar(ctx);

    if (getc_ack != RYM_CODE_C)
    {
        return -RYM_ERR_ACK;
    }

    if (ctx->on_end && ctx->on_end(ctx, ctx->buf + 3, data_sz - 5) != RYM_CODE_SOH)
        return -RYM_ERR_CODE;

    code = RYM_CODE_SOH;

    _rym_send_packet(ctx, code, index);

    ctx->stage = RYM_STAGE_FINISHED;

    return RT_EOK;
}

G
Grissiom 已提交
540
static rt_err_t _rym_do_recv(
541 542
    struct rym_ctx *ctx,
    int handshake_timeout)
G
Grissiom 已提交
543 544 545 546 547 548 549 550 551 552 553
{
    rt_err_t err;

    ctx->stage = RYM_STAGE_NONE;

    ctx->buf = rt_malloc(_RYM_STX_PKG_SZ);
    if (ctx->buf == RT_NULL)
        return -RT_ENOMEM;

    err = _rym_do_handshake(ctx, handshake_timeout);
    if (err != RT_EOK)
554 555
    {
        rt_free(ctx->buf);
G
Grissiom 已提交
556
        return err;
557
    }
L
Lyn 已提交
558 559 560 561
    while (1)
    {
        err = _rym_do_trans(ctx);
        if (err != RT_EOK)
562 563
        {
            rt_free(ctx->buf);
L
Lyn 已提交
564
            return err;
565
        }
G
Grissiom 已提交
566

L
Lyn 已提交
567 568
        err = _rym_do_fin(ctx);
        if (err != RT_EOK)
569 570
        {
            rt_free(ctx->buf);
L
Lyn 已提交
571
            return err;
572 573
        }

L
Lyn 已提交
574 575 576
        if (ctx->stage == RYM_STAGE_FINISHED)
            break;
    }
577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614
    rt_free(ctx->buf);
    return err;
}

static rt_err_t _rym_do_send(
    struct rym_ctx *ctx,
    int handshake_timeout)
{
    rt_err_t err;

    ctx->stage = RYM_STAGE_NONE;

    ctx->buf = rt_malloc(_RYM_STX_PKG_SZ);
    if (ctx->buf == RT_NULL)
        return -RT_ENOMEM;

    err = _rym_do_send_handshake(ctx, handshake_timeout);
    if (err != RT_EOK)
    {
        rt_free(ctx->buf);
        return err;
    }

    err = _rym_do_send_trans(ctx);
    if (err != RT_EOK)
    {
        rt_free(ctx->buf);
        return err;
    }

    err = _rym_do_send_fin(ctx);
    if (err != RT_EOK)
    {
        rt_free(ctx->buf);
        return err;
    }

    rt_free(ctx->buf);
L
Lyn 已提交
615
    return err;
G
Grissiom 已提交
616 617 618
}

rt_err_t rym_recv_on_device(
619 620 621 622 623 624 625
    struct rym_ctx *ctx,
    rt_device_t dev,
    rt_uint16_t oflag,
    rym_callback on_begin,
    rym_callback on_data,
    rym_callback on_end,
    int handshake_timeout)
G
Grissiom 已提交
626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646
{
    rt_err_t res;
    rt_err_t (*odev_rx_ind)(rt_device_t dev, rt_size_t size);
    rt_uint16_t odev_flag;
    int int_lvl;

    RT_ASSERT(_rym_the_ctx == 0);
    _rym_the_ctx = ctx;

    ctx->on_begin = on_begin;
    ctx->on_data  = on_data;
    ctx->on_end   = on_end;
    ctx->dev      = dev;
    rt_sem_init(&ctx->sem, "rymsem", 0, RT_IPC_FLAG_FIFO);

    odev_rx_ind = dev->rx_indicate;
    /* no data should be received before the device has been fully setted up.
     */
    int_lvl = rt_hw_interrupt_disable();
    rt_device_set_rx_indicate(dev, _rym_rx_ind);

647
    odev_flag = dev->open_flag;
G
Grissiom 已提交
648
    /* make sure the device don't change the content. */
649
    dev->open_flag &= ~RT_DEVICE_FLAG_STREAM;
G
Grissiom 已提交
650 651
    rt_hw_interrupt_enable(int_lvl);

652
    res = rt_device_open(dev, oflag);
G
Grissiom 已提交
653 654 655 656 657 658 659 660 661 662 663 664 665
    if (res != RT_EOK)
        goto __exit;

    res = _rym_do_recv(ctx, handshake_timeout);

    rt_device_close(dev);

__exit:
    /* no rx_ind should be called before the callback has been fully detached.
     */
    int_lvl = rt_hw_interrupt_disable();
    rt_sem_detach(&ctx->sem);

666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722
    dev->open_flag = odev_flag;
    rt_device_set_rx_indicate(dev, odev_rx_ind);
    rt_hw_interrupt_enable(int_lvl);

    _rym_the_ctx = RT_NULL;

    return res;
}

rt_err_t rym_send_on_device(
    struct rym_ctx *ctx,
    rt_device_t dev,
    rt_uint16_t oflag,
    rym_callback on_begin,
    rym_callback on_data,
    rym_callback on_end,
    int handshake_timeout)
{
    rt_err_t res = 0;
    rt_err_t (*odev_rx_ind)(rt_device_t dev, rt_size_t size);
    rt_uint16_t odev_flag;
    int int_lvl;

    RT_ASSERT(_rym_the_ctx == 0);
    _rym_the_ctx = ctx;

    ctx->on_begin = on_begin;
    ctx->on_data  = on_data;
    ctx->on_end   = on_end;
    ctx->dev      = dev;
    rt_sem_init(&ctx->sem, "rymsem", 0, RT_IPC_FLAG_FIFO);

    odev_rx_ind = dev->rx_indicate;
    /* no data should be received before the device has been fully setted up.
     */
    int_lvl = rt_hw_interrupt_disable();
    rt_device_set_rx_indicate(dev, _rym_rx_ind);

    odev_flag = dev->open_flag;
    /* make sure the device don't change the content. */
    dev->open_flag &= ~RT_DEVICE_FLAG_STREAM;
    rt_hw_interrupt_enable(int_lvl);

    res = rt_device_open(dev, oflag);
    if (res != RT_EOK)
        goto __exit;

    res = _rym_do_send(ctx, handshake_timeout);

    rt_device_close(dev);

__exit:

    int_lvl = rt_hw_interrupt_disable();
    rt_sem_detach(&ctx->sem);

    dev->open_flag = odev_flag;
G
Grissiom 已提交
723 724 725 726 727 728 729
    rt_device_set_rx_indicate(dev, odev_rx_ind);
    rt_hw_interrupt_enable(int_lvl);

    _rym_the_ctx = RT_NULL;

    return res;
}
730