at_client.c 25.3 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
Lawlieta's avatar
Lawlieta 已提交
11 12 13 14 15 16 17
 */

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

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

#ifdef AT_USING_CLIENT

Lawlieta's avatar
Lawlieta 已提交
23 24 25 26 27
#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"

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

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);

/**
35
 * Create response object.
Lawlieta's avatar
Lawlieta 已提交
36 37 38 39 40 41 42
 *
 * @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
 *
43
 * @return != RT_NULL: response object
Lawlieta's avatar
Lawlieta 已提交
44 45 46 47 48 49 50
 *          = 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));
51
    if (resp == RT_NULL)
Lawlieta's avatar
Lawlieta 已提交
52
    {
53
        LOG_E("AT create response object failed! No memory for response object!");
Lawlieta's avatar
Lawlieta 已提交
54 55 56 57
        return RT_NULL;
    }

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

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

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

/**
93
 * Set response object information
Lawlieta's avatar
Lawlieta 已提交
94
 *
95
 * @param resp response object
Lawlieta's avatar
Lawlieta 已提交
96 97 98 99 100 101
 * @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
 *
102
 * @return  != RT_NULL: response object
Lawlieta's avatar
Lawlieta 已提交
103 104 105 106
 *           = 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 已提交
107
    char *p_temp;
Lawlieta's avatar
Lawlieta 已提交
108 109
    RT_ASSERT(resp);

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

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

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

    return resp;
}

/**
 * Get one line AT response buffer by line number.
 *
135
 * @param resp response object
Lawlieta's avatar
Lawlieta 已提交
136 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
 * @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
 *
173
 * @param resp response object
Lawlieta's avatar
Lawlieta 已提交
174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189
 * @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++)
    {
190
        if (strstr(resp_buf, keyword))
Lawlieta's avatar
Lawlieta 已提交
191 192 193 194 195 196 197 198 199 200 201 202 203 204 205
        {
            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.
 *
206
 * @param resp response object
Lawlieta's avatar
Lawlieta 已提交
207 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
 * @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.
 *
240
 * @param resp response object
Lawlieta's avatar
Lawlieta 已提交
241 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
 * @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.
 *
274 275
 * @param client current AT client object
 * @param resp AT response object, using RT_NULL when you don't care response
Lawlieta's avatar
Lawlieta 已提交
276 277 278 279 280
 * @param cmd_expr AT commands expression
 *
 * @return 0 : success
 *        -1 : response status error
 *        -2 : wait timeout
281
 *        -7 : enter AT CLI mode
Lawlieta's avatar
Lawlieta 已提交
282
 */
283
int at_obj_exec_cmd(at_client_t client, at_response_t resp, const char *cmd_expr, ...)
Lawlieta's avatar
Lawlieta 已提交
284 285 286 287 288 289 290 291
{
    va_list args;
    rt_size_t cmd_size = 0;
    rt_err_t result = RT_EOK;
    const char *cmd = RT_NULL;

    RT_ASSERT(cmd_expr);

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

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

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

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

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

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

319
    if (resp != RT_NULL)
Lawlieta's avatar
Lawlieta 已提交
320 321 322 323
    {
        if (rt_sem_take(client->resp_notice, resp->timeout) != RT_EOK)
        {
            cmd = at_get_last_cmd(&cmd_size);
324
            LOG_D("execute command (%.*s) timeout (%d ticks)!", cmd_size, cmd, resp->timeout);
Lawlieta's avatar
Lawlieta 已提交
325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345
            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 已提交
346 347 348
/**
 * Waiting for connection to external devices.
 *
349
 * @param client current AT client object
Z
zylx 已提交
350 351 352 353 354 355
 * @param timeout millisecond for timeout
 *
 * @return 0 : success
 *        -2 : timeout
 *        -5 : no memory
 */
356
int at_client_obj_wait_connect(at_client_t client, rt_uint32_t timeout)
Z
zylx 已提交
357 358 359 360
{
    rt_err_t result = RT_EOK;
    at_response_t resp = RT_NULL;
    rt_tick_t start_time = 0;
361
    char *client_name = client->device->parent.name;
Z
zylx 已提交
362

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

369
    resp = at_create_resp(64, 0, rt_tick_from_millisecond(300));
370
    if (resp == RT_NULL)
Z
zylx 已提交
371
    {
372
        LOG_E("no memory for AT client(%s) response object.", client_name);
Z
zylx 已提交
373 374 375 376 377 378 379 380 381 382 383
        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 */
384
        if (rt_tick_get() - start_time > rt_tick_from_millisecond(timeout))
Z
zylx 已提交
385
        {
386
            LOG_E("wait AT client(%s) connect timeout(%d tick).", client_name, timeout);
Z
zylx 已提交
387 388 389 390 391
            result = -RT_ETIMEOUT;
            break;
        }

        /* Check whether it is already connected */
392
        resp->buf_len = 0;
Z
zylx 已提交
393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410
        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 已提交
411 412 413
/**
 * Send data to AT server, send data don't have end sign(eg: \r\n).
 *
414
 * @param client current AT client object
Lawlieta's avatar
Lawlieta 已提交
415 416 417
 * @param buf   send data buffer
 * @param size  send fixed data size
 *
418 419
 * @return >0: send data size
 *         =0: send failed
Lawlieta's avatar
Lawlieta 已提交
420
 */
421
rt_size_t at_client_obj_send(at_client_t client, const char *buf, rt_size_t size)
Lawlieta's avatar
Lawlieta 已提交
422 423 424
{
    RT_ASSERT(buf);

425 426 427 428 429 430
    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 已提交
431
#ifdef AT_PRINT_RAW_CMD
432
    at_print_raw_cmd("sendline", buf, size);
Lawlieta's avatar
Lawlieta 已提交
433 434 435 436 437
#endif

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

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

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

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

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

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

    RT_ASSERT(buf);

477 478 479 480 481 482
    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 已提交
483 484 485 486
    while (1)
    {
        if (read_idx < size)
        {
487 488 489 490 491 492
            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 已提交
493 494 495 496 497 498 499 500 501 502 503 504 505 506

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

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

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

/**
 *  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)
{
517 518 519 520 521
    if (client == RT_NULL)
    {
        LOG_E("input AT Client object is NULL, please create or get AT Client object!");
        return;
    }
Lawlieta's avatar
Lawlieta 已提交
522

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

/**
527
 * set URC(Unsolicited Result Code) table
Lawlieta's avatar
Lawlieta 已提交
528
 *
529 530 531
 * @param client current AT client object
 * @param table URC table
 * @param size table size
Lawlieta's avatar
Lawlieta 已提交
532
 */
533
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 已提交
534
{
535 536
    rt_size_t idx;

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

543 544 545 546 547
    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 已提交
548

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
    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++;
585

586 587 588 589
        rt_free(old_urc_table);
    }

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

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

603 604 605 606 607 608 609 610 611 612 613
    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 已提交
614 615
}

616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631
/**
 * 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 已提交
632
{
633 634
    rt_size_t i, j, prefix_len, suffix_len;
    rt_size_t bufsz;
635
    char *buffer = RT_NULL;
636 637
    const struct at_urc *urc = RT_NULL;
    struct at_urc_table *urc_table = RT_NULL;
Lawlieta's avatar
Lawlieta 已提交
638 639 640 641 642 643

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

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

Lawlieta's avatar
Lawlieta 已提交
647 648
    for (i = 0; i < client->urc_table_size; i++)
    {
649
        for (j = 0; j < client->urc_table[i].urc_size; j++)
Lawlieta's avatar
Lawlieta 已提交
650
        {
651 652 653 654 655 656 657 658 659 660 661 662 663 664
            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 已提交
665 666 667 668 669 670
        }
    }

    return RT_NULL;
}

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

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

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

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

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

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

    return read_len;
}

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

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

Lawlieta's avatar
Lawlieta 已提交
737
                /* current receive is response */
738 739
                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 已提交
740 741
                {
                    /* copy response lines, separated by '\0' */
742
                    rt_memcpy(resp->buf + resp->buf_len, client->recv_line_buf, client->recv_line_len);
Lawlieta's avatar
Lawlieta 已提交
743

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

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

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

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

801 802
/* initialize the client object parameters */
static int at_client_para_init(at_client_t client)
Lawlieta's avatar
Lawlieta 已提交
803
{
804 805 806 807
#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 已提交
808 809

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

813 814
    client->status = AT_STATUS_UNINITIALIZED;

815 816 817
    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 已提交
818
    {
H
HubretXie 已提交
819
        LOG_E("AT client initialize failed! No memory for receive buffer.");
820 821
        result = -RT_ENOMEM;
        goto __exit;
Lawlieta's avatar
Lawlieta 已提交
822 823
    }

824 825 826
    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 已提交
827
    {
828 829
        LOG_E("AT client initialize failed! at_client_recv_lock create failed!");
        result = -RT_ENOMEM;
Lawlieta's avatar
Lawlieta 已提交
830 831 832
        goto __exit;
    }

833 834 835
    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 已提交
836
    {
837
        LOG_E("AT client initialize failed! at_client_notice semaphore create failed!");
Lawlieta's avatar
Lawlieta 已提交
838 839 840 841
        result = -RT_ENOMEM;
        goto __exit;
    }

842 843 844
    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 已提交
845
    {
846
        LOG_E("AT client initialize failed! at_client_resp semaphore create failed!");
Lawlieta's avatar
Lawlieta 已提交
847 848 849 850
        result = -RT_ENOMEM;
        goto __exit;
    }

851 852 853 854 855 856 857 858 859 860 861
    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 已提交
862 863 864 865 866
    {
        result = -RT_ENOMEM;
        goto __exit;
    }

867 868
__exit:
    if (result != RT_EOK)
Lawlieta's avatar
Lawlieta 已提交
869
    {
870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889
        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);
        }

890
        if (client->recv_line_buf)
891
        {
892
            rt_free(client->recv_line_buf);
893 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
        }

        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);

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

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

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

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

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

948 949 950 951 952
    /* 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 已提交
953

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

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

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

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

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

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