at_client.c 25.4 KB
Newer Older
Lawlieta's avatar
Lawlieta 已提交
1
/*
2
 * Copyright (c) 2006-2021, RT-Thread Development Team
Lawlieta's avatar
Lawlieta 已提交
3
 *
4
 * SPDX-License-Identifier: Apache-2.0
Lawlieta's avatar
Lawlieta 已提交
5 6 7 8 9
 *
 * Change Logs:
 * Date           Author       Notes
 * 2018-03-30     chenyong     first version
 * 2018-04-12     chenyong     add client implement
10
 * 2018-08-17     chenyong     multiple client support
mysterywolf's avatar
update  
mysterywolf 已提交
11
 * 2021-03-17     Meco Man     fix a buf of leaking memory 
Lawlieta's avatar
Lawlieta 已提交
12 13 14 15 16 17 18
 */

#include <at.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

19 20 21 22 23
#define LOG_TAG              "at.clnt"
#include <at_log.h>

#ifdef AT_USING_CLIENT

Lawlieta's avatar
Lawlieta 已提交
24 25 26 27 28
#define AT_RESP_END_OK                 "OK"
#define AT_RESP_END_ERROR              "ERROR"
#define AT_RESP_END_FAIL               "FAIL"
#define AT_END_CR_LF                   "\r\n"

29
static struct at_client at_client_table[AT_CLIENT_NUM_MAX] = { 0 };
Lawlieta's avatar
Lawlieta 已提交
30 31 32 33 34 35

extern rt_size_t at_vprintfln(rt_device_t device, const char *format, va_list args);
extern void at_print_raw_cmd(const char *type, const char *cmd, rt_size_t size);
extern const char *at_get_last_cmd(rt_size_t *cmd_size);

/**
36
 * Create response object.
Lawlieta's avatar
Lawlieta 已提交
37 38 39 40 41 42 43
 *
 * @param buf_size the maximum response buffer size
 * @param line_num the number of setting response lines
 *         = 0: the response data will auto return when received 'OK' or 'ERROR'
 *        != 0: the response data will return when received setting lines number data
 * @param timeout the maximum response time
 *
44
 * @return != RT_NULL: response object
Lawlieta's avatar
Lawlieta 已提交
45 46 47 48 49 50 51
 *          = RT_NULL: no memory
 */
at_response_t at_create_resp(rt_size_t buf_size, rt_size_t line_num, rt_int32_t timeout)
{
    at_response_t resp = RT_NULL;

    resp = (at_response_t) rt_calloc(1, sizeof(struct at_response));
52
    if (resp == RT_NULL)
Lawlieta's avatar
Lawlieta 已提交
53
    {
54
        LOG_E("AT create response object failed! No memory for response object!");
Lawlieta's avatar
Lawlieta 已提交
55 56 57 58
        return RT_NULL;
    }

    resp->buf = (char *) rt_calloc(1, buf_size);
59
    if (resp->buf == RT_NULL)
Lawlieta's avatar
Lawlieta 已提交
60
    {
61
        LOG_E("AT create response object failed! No memory for response buffer!");
Lawlieta's avatar
Lawlieta 已提交
62 63 64 65 66 67 68 69 70 71 72 73 74
        rt_free(resp);
        return RT_NULL;
    }

    resp->buf_size = buf_size;
    resp->line_num = line_num;
    resp->line_counts = 0;
    resp->timeout = timeout;

    return resp;
}

/**
75
 * Delete and free response object.
Lawlieta's avatar
Lawlieta 已提交
76
 *
77
 * @param resp response object
Lawlieta's avatar
Lawlieta 已提交
78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93
 */
void at_delete_resp(at_response_t resp)
{
    if (resp && resp->buf)
    {
        rt_free(resp->buf);
    }

    if (resp)
    {
        rt_free(resp);
        resp = RT_NULL;
    }
}

/**
94
 * Set response object information
Lawlieta's avatar
Lawlieta 已提交
95
 *
96
 * @param resp response object
Lawlieta's avatar
Lawlieta 已提交
97 98 99 100 101 102
 * @param buf_size the maximum response buffer size
 * @param line_num the number of setting response lines
 *         = 0: the response data will auto return when received 'OK' or 'ERROR'
 *        != 0: the response data will return when received setting lines number data
 * @param timeout the maximum response time
 *
103
 * @return  != RT_NULL: response object
Lawlieta's avatar
Lawlieta 已提交
104 105 106 107
 *           = RT_NULL: no memory
 */
at_response_t at_resp_set_info(at_response_t resp, rt_size_t buf_size, rt_size_t line_num, rt_int32_t timeout)
{
mysterywolf's avatar
mysterywolf 已提交
108
    char *p_temp;
Lawlieta's avatar
Lawlieta 已提交
109 110
    RT_ASSERT(resp);

111
    if (resp->buf_size != buf_size)
Lawlieta's avatar
Lawlieta 已提交
112 113 114
    {
        resp->buf_size = buf_size;

mysterywolf's avatar
mysterywolf 已提交
115 116
        p_temp = (char *) rt_realloc(resp->buf, buf_size);
        if (p_temp == RT_NULL)
Lawlieta's avatar
Lawlieta 已提交
117 118 119 120
        {
            LOG_D("No memory for realloc response buffer size(%d).", buf_size);
            return RT_NULL;
        }
mysterywolf's avatar
mysterywolf 已提交
121 122 123 124
        else
        {
            resp->buf = p_temp;
        }
Lawlieta's avatar
Lawlieta 已提交
125 126 127 128 129 130 131 132 133 134 135
    }

    resp->line_num = line_num;
    resp->timeout = timeout;

    return resp;
}

/**
 * Get one line AT response buffer by line number.
 *
136
 * @param resp response object
Lawlieta's avatar
Lawlieta 已提交
137 138 139 140 141 142 143 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 170 171 172 173
 * @param resp_line line number, start from '1'
 *
 * @return != RT_NULL: response line buffer
 *          = RT_NULL: input response line error
 */
const char *at_resp_get_line(at_response_t resp, rt_size_t resp_line)
{
    char *resp_buf = resp->buf;
    char *resp_line_buf = RT_NULL;
    rt_size_t line_num = 1;

    RT_ASSERT(resp);

    if (resp_line > resp->line_counts || resp_line <= 0)
    {
        LOG_E("AT response get line failed! Input response line(%d) error!", resp_line);
        return RT_NULL;
    }

    for (line_num = 1; line_num <= resp->line_counts; line_num++)
    {
        if (resp_line == line_num)
        {
            resp_line_buf = resp_buf;

            return resp_line_buf;
        }

        resp_buf += strlen(resp_buf) + 1;
    }

    return RT_NULL;
}

/**
 * Get one line AT response buffer by keyword
 *
174
 * @param resp response object
Lawlieta's avatar
Lawlieta 已提交
175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190
 * @param keyword query keyword
 *
 * @return != RT_NULL: response line buffer
 *          = RT_NULL: no matching data
 */
const char *at_resp_get_line_by_kw(at_response_t resp, const char *keyword)
{
    char *resp_buf = resp->buf;
    char *resp_line_buf = RT_NULL;
    rt_size_t line_num = 1;

    RT_ASSERT(resp);
    RT_ASSERT(keyword);

    for (line_num = 1; line_num <= resp->line_counts; line_num++)
    {
191
        if (strstr(resp_buf, keyword))
Lawlieta's avatar
Lawlieta 已提交
192 193 194 195 196 197 198 199 200 201 202 203 204 205 206
        {
            resp_line_buf = resp_buf;

            return resp_line_buf;
        }

        resp_buf += strlen(resp_buf) + 1;
    }

    return RT_NULL;
}

/**
 * Get and parse AT response buffer arguments by line number.
 *
207
 * @param resp response object
Lawlieta's avatar
Lawlieta 已提交
208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240
 * @param resp_line line number, start from '1'
 * @param resp_expr response buffer expression
 *
 * @return -1 : input response line number error or get line buffer error
 *          0 : parsed without match
 *         >0 : the number of arguments successfully parsed
 */
int at_resp_parse_line_args(at_response_t resp, rt_size_t resp_line, const char *resp_expr, ...)
{
    va_list args;
    int resp_args_num = 0;
    const char *resp_line_buf = RT_NULL;

    RT_ASSERT(resp);
    RT_ASSERT(resp_expr);

    if ((resp_line_buf = at_resp_get_line(resp, resp_line)) == RT_NULL)
    {
        return -1;
    }

    va_start(args, resp_expr);

    resp_args_num = vsscanf(resp_line_buf, resp_expr, args);

    va_end(args);

    return resp_args_num;
}

/**
 * Get and parse AT response buffer arguments by keyword.
 *
241
 * @param resp response object
Lawlieta's avatar
Lawlieta 已提交
242 243 244 245 246 247 248 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
 * @param keyword query keyword
 * @param resp_expr response buffer expression
 *
 * @return -1 : input keyword error or get line buffer error
 *          0 : parsed without match
 *         >0 : the number of arguments successfully parsed
 */
int at_resp_parse_line_args_by_kw(at_response_t resp, const char *keyword, const char *resp_expr, ...)
{
    va_list args;
    int resp_args_num = 0;
    const char *resp_line_buf = RT_NULL;

    RT_ASSERT(resp);
    RT_ASSERT(resp_expr);

    if ((resp_line_buf = at_resp_get_line_by_kw(resp, keyword)) == RT_NULL)
    {
        return -1;
    }

    va_start(args, resp_expr);

    resp_args_num = vsscanf(resp_line_buf, resp_expr, args);

    va_end(args);

    return resp_args_num;
}

/**
 * Send commands to AT server and wait response.
 *
275 276
 * @param client current AT client object
 * @param resp AT response object, using RT_NULL when you don't care response
Lawlieta's avatar
Lawlieta 已提交
277 278 279 280 281
 * @param cmd_expr AT commands expression
 *
 * @return 0 : success
 *        -1 : response status error
 *        -2 : wait timeout
282
 *        -7 : enter AT CLI mode
Lawlieta's avatar
Lawlieta 已提交
283
 */
284
int at_obj_exec_cmd(at_client_t client, at_response_t resp, const char *cmd_expr, ...)
Lawlieta's avatar
Lawlieta 已提交
285 286 287 288 289 290 291 292
{
    va_list args;
    rt_size_t cmd_size = 0;
    rt_err_t result = RT_EOK;
    const char *cmd = RT_NULL;

    RT_ASSERT(cmd_expr);

293 294 295 296 297 298
    if (client == RT_NULL)
    {
        LOG_E("input AT Client object is NULL, please create or get AT Client object!");
        return -RT_ERROR;
    }

299 300 301 302 303 304
    /* check AT CLI mode */
    if (client->status == AT_STATUS_CLI && resp)
    {
        return -RT_EBUSY;
    }

Lawlieta's avatar
Lawlieta 已提交
305 306 307 308 309
    rt_mutex_take(client->lock, RT_WAITING_FOREVER);

    client->resp_status = AT_RESP_OK;
    client->resp = resp;

310 311 312 313 314 315
    if (resp != RT_NULL)
    {
        resp->buf_len = 0;
        resp->line_counts = 0;
    }

Lawlieta's avatar
Lawlieta 已提交
316 317 318 319
    va_start(args, cmd_expr);
    at_vprintfln(client->device, cmd_expr, args);
    va_end(args);

320
    if (resp != RT_NULL)
Lawlieta's avatar
Lawlieta 已提交
321 322 323 324
    {
        if (rt_sem_take(client->resp_notice, resp->timeout) != RT_EOK)
        {
            cmd = at_get_last_cmd(&cmd_size);
325
            LOG_D("execute command (%.*s) timeout (%d ticks)!", cmd_size, cmd, resp->timeout);
Lawlieta's avatar
Lawlieta 已提交
326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346
            client->resp_status = AT_RESP_TIMEOUT;
            result = -RT_ETIMEOUT;
            goto __exit;
        }
        if (client->resp_status != AT_RESP_OK)
        {
            cmd = at_get_last_cmd(&cmd_size);
            LOG_E("execute command (%.*s) failed!", cmd_size, cmd);
            result = -RT_ERROR;
            goto __exit;
        }
    }

__exit:
    client->resp = RT_NULL;

    rt_mutex_release(client->lock);

    return result;
}

Z
zylx 已提交
347 348 349
/**
 * Waiting for connection to external devices.
 *
350
 * @param client current AT client object
Z
zylx 已提交
351 352 353 354 355 356
 * @param timeout millisecond for timeout
 *
 * @return 0 : success
 *        -2 : timeout
 *        -5 : no memory
 */
357
int at_client_obj_wait_connect(at_client_t client, rt_uint32_t timeout)
Z
zylx 已提交
358 359 360 361
{
    rt_err_t result = RT_EOK;
    at_response_t resp = RT_NULL;
    rt_tick_t start_time = 0;
362
    char *client_name = client->device->parent.name;
Z
zylx 已提交
363

364 365
    if (client == RT_NULL)
    {
366
        LOG_E("input AT client object is NULL, please create or get AT Client object!");
367
        return -RT_ERROR;
368 369
    }

370
    resp = at_create_resp(64, 0, rt_tick_from_millisecond(300));
371
    if (resp == RT_NULL)
Z
zylx 已提交
372
    {
373
        LOG_E("no memory for AT client(%s) response object.", client_name);
Z
zylx 已提交
374 375 376 377 378 379 380 381 382 383 384
        return -RT_ENOMEM;
    }

    rt_mutex_take(client->lock, RT_WAITING_FOREVER);
    client->resp = resp;

    start_time = rt_tick_get();

    while (1)
    {
        /* Check whether it is timeout */
385
        if (rt_tick_get() - start_time > rt_tick_from_millisecond(timeout))
Z
zylx 已提交
386
        {
387
            LOG_E("wait AT client(%s) connect timeout(%d tick).", client_name, timeout);
Z
zylx 已提交
388 389 390 391 392
            result = -RT_ETIMEOUT;
            break;
        }

        /* Check whether it is already connected */
393
        resp->buf_len = 0;
Z
zylx 已提交
394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411
        resp->line_counts = 0;
        rt_device_write(client->device, 0, "AT\r\n", 4);

        if (rt_sem_take(client->resp_notice, resp->timeout) != RT_EOK)
            continue;
        else
            break;
    }

    at_delete_resp(resp);

    client->resp = RT_NULL;

    rt_mutex_release(client->lock);

    return result;
}

Lawlieta's avatar
Lawlieta 已提交
412 413 414
/**
 * Send data to AT server, send data don't have end sign(eg: \r\n).
 *
415
 * @param client current AT client object
Lawlieta's avatar
Lawlieta 已提交
416 417 418
 * @param buf   send data buffer
 * @param size  send fixed data size
 *
419 420
 * @return >0: send data size
 *         =0: send failed
Lawlieta's avatar
Lawlieta 已提交
421
 */
422
rt_size_t at_client_obj_send(at_client_t client, const char *buf, rt_size_t size)
Lawlieta's avatar
Lawlieta 已提交
423 424 425
{
    RT_ASSERT(buf);

426 427 428 429 430 431
    if (client == RT_NULL)
    {
        LOG_E("input AT Client object is NULL, please create or get AT Client object!");
        return 0;
    }

Lawlieta's avatar
Lawlieta 已提交
432
#ifdef AT_PRINT_RAW_CMD
433
    at_print_raw_cmd("sendline", buf, size);
Lawlieta's avatar
Lawlieta 已提交
434 435 436 437 438
#endif

    return rt_device_write(client->device, 0, buf, size);
}

439
static rt_err_t at_client_getchar(at_client_t client, char *ch, rt_int32_t timeout)
Lawlieta's avatar
Lawlieta 已提交
440
{
441
    rt_err_t result = RT_EOK;
Lawlieta's avatar
Lawlieta 已提交
442

443
    while (rt_device_read(client->device, 0, ch, 1) == 0)
444
    {
445 446 447 448 449
        result = rt_sem_take(client->rx_notice, rt_tick_from_millisecond(timeout));
        if (result != RT_EOK)
        {
            return result;
        }
W
Wayne Lin 已提交
450 451

        rt_sem_control(client->rx_notice, RT_IPC_CMD_RESET, RT_NULL);
452
    }
453 454

    return RT_EOK;
Lawlieta's avatar
Lawlieta 已提交
455 456 457 458 459
}

/**
 * AT client receive fixed-length data.
 *
460
 * @param client current AT client object
Lawlieta's avatar
Lawlieta 已提交
461 462
 * @param buf   receive data buffer
 * @param size  receive fixed data size
463
 * @param timeout  receive data timeout (ms)
Lawlieta's avatar
Lawlieta 已提交
464 465 466
 *
 * @note this function can only be used in execution function of URC data
 *
467 468
 * @return >0: receive data size
 *         =0: receive failed
Lawlieta's avatar
Lawlieta 已提交
469
 */
470
rt_size_t at_client_obj_recv(at_client_t client, char *buf, rt_size_t size, rt_int32_t timeout)
Lawlieta's avatar
Lawlieta 已提交
471 472
{
    rt_size_t read_idx = 0;
473
    rt_err_t result = RT_EOK;
Lawlieta's avatar
Lawlieta 已提交
474 475 476 477
    char ch;

    RT_ASSERT(buf);

478 479 480 481 482 483
    if (client == RT_NULL)
    {
        LOG_E("input AT Client object is NULL, please create or get AT Client object!");
        return 0;
    }

Lawlieta's avatar
Lawlieta 已提交
484 485 486 487
    while (1)
    {
        if (read_idx < size)
        {
488 489 490 491 492 493
            result = at_client_getchar(client, &ch, timeout);
            if (result != RT_EOK)
            {
                LOG_E("AT Client receive failed, uart device get data error(%d)", result);
                return 0;
            }
Lawlieta's avatar
Lawlieta 已提交
494 495 496 497 498 499 500 501 502 503 504 505 506 507

            buf[read_idx++] = ch;
        }
        else
        {
            break;
        }
    }

#ifdef AT_PRINT_RAW_CMD
    at_print_raw_cmd("urc_recv", buf, size);
#endif

    return read_idx;
508 509 510 511 512 513 514 515 516 517
}

/**
 *  AT client set end sign.
 *
 * @param client current AT client object
 * @param ch the end sign, can not be used when it is '\0'
 */
void at_obj_set_end_sign(at_client_t client, char ch)
{
518 519 520 521 522
    if (client == RT_NULL)
    {
        LOG_E("input AT Client object is NULL, please create or get AT Client object!");
        return;
    }
Lawlieta's avatar
Lawlieta 已提交
523

524
    client->end_sign = ch;
Lawlieta's avatar
Lawlieta 已提交
525 526 527
}

/**
528
 * set URC(Unsolicited Result Code) table
Lawlieta's avatar
Lawlieta 已提交
529
 *
530 531 532
 * @param client current AT client object
 * @param table URC table
 * @param size table size
Lawlieta's avatar
Lawlieta 已提交
533
 */
534
int at_obj_set_urc_table(at_client_t client, const struct at_urc *urc_table, rt_size_t table_sz)
Lawlieta's avatar
Lawlieta 已提交
535
{
536 537
    rt_size_t idx;

538 539 540
    if (client == RT_NULL)
    {
        LOG_E("input AT Client object is NULL, please create or get AT Client object!");
541
        return -RT_ERROR;
542 543
    }

544 545 546 547 548
    for (idx = 0; idx < table_sz; idx++)
    {
        RT_ASSERT(urc_table[idx].cmd_prefix);
        RT_ASSERT(urc_table[idx].cmd_suffix);
    }
Lawlieta's avatar
Lawlieta 已提交
549

550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585
    if (client->urc_table_size == 0)
    {
        client->urc_table = (struct at_urc_table *) rt_calloc(1, sizeof(struct at_urc_table));
        if (client->urc_table == RT_NULL)
        {
            return -RT_ENOMEM;
        }

        client->urc_table[0].urc = urc_table;
        client->urc_table[0].urc_size = table_sz;
        client->urc_table_size++;
    }
    else
    {
        struct at_urc_table *old_urc_table = RT_NULL;
        size_t old_table_size = client->urc_table_size * sizeof(struct at_urc_table);

        old_urc_table = (struct at_urc_table *) rt_malloc(old_table_size);
        if (old_urc_table == RT_NULL)
        {
            return -RT_ENOMEM;
        }
        rt_memcpy(old_urc_table, client->urc_table, old_table_size);

        /* realloc urc table space */
        client->urc_table = (struct at_urc_table *) rt_realloc(client->urc_table,
                old_table_size + sizeof(struct at_urc_table));
        if (client->urc_table == RT_NULL)
        {
            rt_free(old_urc_table);
            return -RT_ENOMEM;
        }
        rt_memcpy(client->urc_table, old_urc_table, old_table_size);
        client->urc_table[client->urc_table_size].urc = urc_table;
        client->urc_table[client->urc_table_size].urc_size = table_sz;
        client->urc_table_size++;
586

587 588 589 590
        rt_free(old_urc_table);
    }

    return RT_EOK;
Lawlieta's avatar
Lawlieta 已提交
591 592 593
}

/**
594
 * get AT client object by AT device name.
Lawlieta's avatar
Lawlieta 已提交
595
 *
596
 * @dev_name AT client device name
Lawlieta's avatar
Lawlieta 已提交
597
 *
598
 * @return AT client object
Lawlieta's avatar
Lawlieta 已提交
599
 */
600
at_client_t at_client_get(const char *dev_name)
Lawlieta's avatar
Lawlieta 已提交
601
{
602
    int idx = 0;
Lawlieta's avatar
Lawlieta 已提交
603

604 605 606 607 608 609 610 611 612 613 614
    RT_ASSERT(dev_name);

    for (idx = 0; idx < AT_CLIENT_NUM_MAX; idx++)
    {
        if (rt_strcmp(at_client_table[idx].device->parent.name, dev_name) == 0)
        {
            return &at_client_table[idx];
        }
    }

    return RT_NULL;
Lawlieta's avatar
Lawlieta 已提交
615 616
}

617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632
/**
 * get first AT client object in the table.
 *
 * @return AT client object
 */
at_client_t at_client_get_first(void)
{
    if (at_client_table[0].device == RT_NULL)
    {
        return RT_NULL;
    }

    return &at_client_table[0];
}

static const struct at_urc *get_urc_obj(at_client_t client)
Lawlieta's avatar
Lawlieta 已提交
633
{
634 635
    rt_size_t i, j, prefix_len, suffix_len;
    rt_size_t bufsz;
636
    char *buffer = RT_NULL;
637 638
    const struct at_urc *urc = RT_NULL;
    struct at_urc_table *urc_table = RT_NULL;
Lawlieta's avatar
Lawlieta 已提交
639 640 641 642 643 644

    if (client->urc_table == RT_NULL)
    {
        return RT_NULL;
    }

645 646
    buffer = client->recv_line_buf;
    bufsz = client->recv_line_len;
647

Lawlieta's avatar
Lawlieta 已提交
648 649
    for (i = 0; i < client->urc_table_size; i++)
    {
650
        for (j = 0; j < client->urc_table[i].urc_size; j++)
Lawlieta's avatar
Lawlieta 已提交
651
        {
652 653 654 655 656 657 658 659 660 661 662 663 664 665
            urc_table = client->urc_table + i;
            urc = urc_table->urc + j;

            prefix_len = rt_strlen(urc->cmd_prefix);
            suffix_len = rt_strlen(urc->cmd_suffix);
            if (bufsz < prefix_len + suffix_len)
            {
                continue;
            }
            if ((prefix_len ? !rt_strncmp(buffer, urc->cmd_prefix, prefix_len) : 1)
                    && (suffix_len ? !rt_strncmp(buffer + bufsz - suffix_len, urc->cmd_suffix, suffix_len) : 1))
            {
                return urc;
            }
Lawlieta's avatar
Lawlieta 已提交
666 667 668 669 670 671
        }
    }

    return RT_NULL;
}

672
static int at_recv_readline(at_client_t client)
Lawlieta's avatar
Lawlieta 已提交
673 674 675 676 677
{
    rt_size_t read_len = 0;
    char ch = 0, last_ch = 0;
    rt_bool_t is_full = RT_FALSE;

678 679
    rt_memset(client->recv_line_buf, 0x00, client->recv_bufsz);
    client->recv_line_len = 0;
Lawlieta's avatar
Lawlieta 已提交
680 681 682

    while (1)
    {
683
        at_client_getchar(client, &ch, RT_WAITING_FOREVER);
Lawlieta's avatar
Lawlieta 已提交
684

685
        if (read_len < client->recv_bufsz)
Lawlieta's avatar
Lawlieta 已提交
686
        {
687 688
            client->recv_line_buf[read_len++] = ch;
            client->recv_line_len = read_len;
Lawlieta's avatar
Lawlieta 已提交
689 690 691 692 693 694 695
        }
        else
        {
            is_full = RT_TRUE;
        }

        /* is newline or URC data */
696 697
        if ((ch == '\n' && last_ch == '\r') || (client->end_sign != 0 && ch == client->end_sign)
                || get_urc_obj(client))
Lawlieta's avatar
Lawlieta 已提交
698 699 700
        {
            if (is_full)
            {
701
                LOG_E("read line failed. The line data length is out of buffer size(%d)!", client->recv_bufsz);
702 703
                rt_memset(client->recv_line_buf, 0x00, client->recv_bufsz);
                client->recv_line_len = 0;
Lawlieta's avatar
Lawlieta 已提交
704 705 706 707 708 709 710 711
                return -RT_EFULL;
            }
            break;
        }
        last_ch = ch;
    }

#ifdef AT_PRINT_RAW_CMD
712
    at_print_raw_cmd("recvline", client->recv_line_buf, read_len);
Lawlieta's avatar
Lawlieta 已提交
713 714 715 716 717 718 719 720 721 722 723
#endif

    return read_len;
}

static void client_parser(at_client_t client)
{
    const struct at_urc *urc;

    while(1)
    {
724
        if (at_recv_readline(client) > 0)
Lawlieta's avatar
Lawlieta 已提交
725
        {
726
            if ((urc = get_urc_obj(client)) != RT_NULL)
Lawlieta's avatar
Lawlieta 已提交
727 728 729 730
            {
                /* current receive is request, try to execute related operations */
                if (urc->func != RT_NULL)
                {
731
                    urc->func(client, client->recv_line_buf, client->recv_line_len);
Lawlieta's avatar
Lawlieta 已提交
732 733 734 735
                }
            }
            else if (client->resp != RT_NULL)
            {
736 737
                at_response_t resp = client->resp;

Lawlieta's avatar
Lawlieta 已提交
738
                /* current receive is response */
739 740
                client->recv_line_buf[client->recv_line_len - 1] = '\0';
                if (resp->buf_len + client->recv_line_len < resp->buf_size)
Lawlieta's avatar
Lawlieta 已提交
741 742
                {
                    /* copy response lines, separated by '\0' */
743
                    rt_memcpy(resp->buf + resp->buf_len, client->recv_line_buf, client->recv_line_len);
Lawlieta's avatar
Lawlieta 已提交
744

745 746 747
                    /* update the current response information */
                    resp->buf_len += client->recv_line_len;
                    resp->line_counts++;
Lawlieta's avatar
Lawlieta 已提交
748 749 750 751
                }
                else
                {
                    client->resp_status = AT_RESP_BUFF_FULL;
752
                    LOG_E("Read response buffer failed. The Response buffer size is out of buffer size(%d)!", resp->buf_size);
Lawlieta's avatar
Lawlieta 已提交
753 754
                }
                /* check response result */
755 756
                if (rt_memcmp(client->recv_line_buf, AT_RESP_END_OK, rt_strlen(AT_RESP_END_OK)) == 0
                        && resp->line_num == 0)
Lawlieta's avatar
Lawlieta 已提交
757 758 759 760
                {
                    /* get the end data by response result, return response state END_OK. */
                    client->resp_status = AT_RESP_OK;
                }
761 762
                else if (rt_strstr(client->recv_line_buf, AT_RESP_END_ERROR)
                        || (rt_memcmp(client->recv_line_buf, AT_RESP_END_FAIL, rt_strlen(AT_RESP_END_FAIL)) == 0))
Lawlieta's avatar
Lawlieta 已提交
763 764 765
                {
                    client->resp_status = AT_RESP_ERROR;
                }
766
                else if (resp->line_counts == resp->line_num && resp->line_num)
Lawlieta's avatar
Lawlieta 已提交
767 768 769 770 771 772 773 774 775 776 777 778 779 780
                {
                    /* get the end data by response line, return response state END_OK.*/
                    client->resp_status = AT_RESP_OK;
                }
                else
                {
                    continue;
                }

                client->resp = RT_NULL;
                rt_sem_release(client->resp_notice);
            }
            else
            {
781
//                log_d("unrecognized line: %.*s", client->recv_line_len, client->recv_line_buf);
Lawlieta's avatar
Lawlieta 已提交
782 783 784 785 786 787 788
            }
        }
    }
}

static rt_err_t at_client_rx_ind(rt_device_t dev, rt_size_t size)
{
789 790 791
    int idx = 0;

    for (idx = 0; idx < AT_CLIENT_NUM_MAX; idx++)
792
    {
793 794 795 796
        if (at_client_table[idx].device == dev && size > 0)
        {
            rt_sem_release(at_client_table[idx].rx_notice);
        }
797 798
    }

Lawlieta's avatar
Lawlieta 已提交
799 800 801
    return RT_EOK;
}

802 803
/* initialize the client object parameters */
static int at_client_para_init(at_client_t client)
Lawlieta's avatar
Lawlieta 已提交
804
{
805 806 807 808
#define AT_CLIENT_LOCK_NAME            "at_c"
#define AT_CLIENT_SEM_NAME             "at_cs"
#define AT_CLIENT_RESP_NAME            "at_cr"
#define AT_CLIENT_THREAD_NAME          "at_clnt"
Lawlieta's avatar
Lawlieta 已提交
809 810

    int result = RT_EOK;
811 812
    static int at_client_num = 0;
    char name[RT_NAME_MAX];
Lawlieta's avatar
Lawlieta 已提交
813

814 815
    client->status = AT_STATUS_UNINITIALIZED;

816 817 818
    client->recv_line_len = 0;
    client->recv_line_buf = (char *) rt_calloc(1, client->recv_bufsz);
    if (client->recv_line_buf == RT_NULL)
Lawlieta's avatar
Lawlieta 已提交
819
    {
H
HubretXie 已提交
820
        LOG_E("AT client initialize failed! No memory for receive buffer.");
821 822
        result = -RT_ENOMEM;
        goto __exit;
Lawlieta's avatar
Lawlieta 已提交
823 824
    }

825 826 827
    rt_snprintf(name, RT_NAME_MAX, "%s%d", AT_CLIENT_LOCK_NAME, at_client_num);
    client->lock = rt_mutex_create(name, RT_IPC_FLAG_FIFO);
    if (client->lock == RT_NULL)
Lawlieta's avatar
Lawlieta 已提交
828
    {
829 830
        LOG_E("AT client initialize failed! at_client_recv_lock create failed!");
        result = -RT_ENOMEM;
Lawlieta's avatar
Lawlieta 已提交
831 832 833
        goto __exit;
    }

834 835 836
    rt_snprintf(name, RT_NAME_MAX, "%s%d", AT_CLIENT_SEM_NAME, at_client_num);
    client->rx_notice = rt_sem_create(name, 0, RT_IPC_FLAG_FIFO);
    if (client->rx_notice == RT_NULL)
Lawlieta's avatar
Lawlieta 已提交
837
    {
838
        LOG_E("AT client initialize failed! at_client_notice semaphore create failed!");
Lawlieta's avatar
Lawlieta 已提交
839 840 841 842
        result = -RT_ENOMEM;
        goto __exit;
    }

843 844 845
    rt_snprintf(name, RT_NAME_MAX, "%s%d", AT_CLIENT_RESP_NAME, at_client_num);
    client->resp_notice = rt_sem_create(name, 0, RT_IPC_FLAG_FIFO);
    if (client->resp_notice == RT_NULL)
Lawlieta's avatar
Lawlieta 已提交
846
    {
847
        LOG_E("AT client initialize failed! at_client_resp semaphore create failed!");
Lawlieta's avatar
Lawlieta 已提交
848 849 850 851
        result = -RT_ENOMEM;
        goto __exit;
    }

852 853 854 855 856 857 858 859 860 861 862
    client->urc_table = RT_NULL;
    client->urc_table_size = 0;

    rt_snprintf(name, RT_NAME_MAX, "%s%d", AT_CLIENT_THREAD_NAME, at_client_num);
    client->parser = rt_thread_create(name,
                                     (void (*)(void *parameter))client_parser,
                                     client,
                                     1024 + 512,
                                     RT_THREAD_PRIORITY_MAX / 3 - 1,
                                     5);
    if (client->parser == RT_NULL)
Lawlieta's avatar
Lawlieta 已提交
863 864 865 866 867
    {
        result = -RT_ENOMEM;
        goto __exit;
    }

868 869
__exit:
    if (result != RT_EOK)
Lawlieta's avatar
Lawlieta 已提交
870
    {
871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890
        if (client->lock)
        {
            rt_mutex_delete(client->lock);
        }

        if (client->rx_notice)
        {
            rt_sem_delete(client->rx_notice);
        }

        if (client->resp_notice)
        {
            rt_sem_delete(client->resp_notice);
        }

        if (client->device)
        {
            rt_device_close(client->device);
        }

891
        if (client->recv_line_buf)
892
        {
893
            rt_free(client->recv_line_buf);
894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925
        }

        rt_memset(client, 0x00, sizeof(struct at_client));
    }
    else
    {
        at_client_num++;
    }

    return result;
}

/**
 * AT client initialize.
 *
 * @param dev_name AT client device name
 * @param recv_bufsz the maximum number of receive buffer length
 *
 * @return 0 : initialize success
 *        -1 : initialize failed
 *        -5 : no memory
 */
int at_client_init(const char *dev_name,  rt_size_t recv_bufsz)
{
    int idx = 0;
    int result = RT_EOK;
    rt_err_t open_result = RT_EOK;
    at_client_t client = RT_NULL;

    RT_ASSERT(dev_name);
    RT_ASSERT(recv_bufsz > 0);

926 927 928 929 930
    if (at_client_get(dev_name) != RT_NULL)
    {
        return result;
    }

931 932 933 934
    for (idx = 0; idx < AT_CLIENT_NUM_MAX && at_client_table[idx].device; idx++);

    if (idx >= AT_CLIENT_NUM_MAX)
    {
L
luofanlu 已提交
935
        LOG_E("AT client initialize failed! Check the maximum number(%d) of AT client.", AT_CLIENT_NUM_MAX);
936 937 938 939 940 941 942
        result = -RT_EFULL;
        goto __exit;
    }

    client = &at_client_table[idx];
    client->recv_bufsz = recv_bufsz;

943 944 945 946 947 948
    result = at_client_para_init(client);
    if (result != RT_EOK)
    {
        goto __exit;
    }

949 950 951 952 953
    /* find and open command device */
    client->device = rt_device_find(dev_name);
    if (client->device)
    {
        RT_ASSERT(client->device->type == RT_Device_Class_Char);
Lawlieta's avatar
Lawlieta 已提交
954

955
        /* using DMA mode first */
956
        open_result = rt_device_open(client->device, RT_DEVICE_OFLAG_RDWR | RT_DEVICE_FLAG_DMA_RX);
957 958 959
        /* using interrupt mode when DMA mode not supported */
        if (open_result == -RT_EIO)
        {
960
            open_result = rt_device_open(client->device, RT_DEVICE_OFLAG_RDWR | RT_DEVICE_FLAG_INT_RX);
961 962
        }
        RT_ASSERT(open_result == RT_EOK);
Lawlieta's avatar
Lawlieta 已提交
963

964
        rt_device_set_rx_indicate(client->device, at_client_rx_ind);
Lawlieta's avatar
Lawlieta 已提交
965 966 967
    }
    else
    {
968
        LOG_E("AT client initialize failed! Not find the device(%s).", dev_name);
Lawlieta's avatar
Lawlieta 已提交
969 970 971 972 973
        result = -RT_ERROR;
        goto __exit;
    }

__exit:
974
    if (result == RT_EOK)
Lawlieta's avatar
Lawlieta 已提交
975
    {
976
        client->status = AT_STATUS_INITIALIZED;
Lawlieta's avatar
Lawlieta 已提交
977

978
        rt_thread_startup(client->parser);
Lawlieta's avatar
Lawlieta 已提交
979

980
        LOG_I("AT client(V%s) on device %s initialize success.", AT_SW_VERSION, dev_name);
Lawlieta's avatar
Lawlieta 已提交
981 982 983
    }
    else
    {
984
        LOG_E("AT client(V%s) on device %s initialize failed(%d).", AT_SW_VERSION, dev_name, result);
Lawlieta's avatar
Lawlieta 已提交
985 986 987 988
    }

    return result;
}
989
#endif /* AT_USING_CLIENT */