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 107 108
 *           = 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)
{
    RT_ASSERT(resp);

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

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

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

    return resp;
}

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

    RT_ASSERT(cmd_expr);

287 288 289 290 291 292
    if (client == RT_NULL)
    {
        LOG_E("input AT Client object is NULL, please create or get AT Client object!");
        return -RT_ERROR;
    }

293 294 295 296 297 298
    /* check AT CLI mode */
    if (client->status == AT_STATUS_CLI && resp)
    {
        return -RT_EBUSY;
    }

Lawlieta's avatar
Lawlieta 已提交
299 300 301 302 303
    rt_mutex_take(client->lock, RT_WAITING_FOREVER);

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

304 305 306 307 308 309
    if (resp != RT_NULL)
    {
        resp->buf_len = 0;
        resp->line_counts = 0;
    }

Lawlieta's avatar
Lawlieta 已提交
310 311 312 313
    va_start(args, cmd_expr);
    at_vprintfln(client->device, cmd_expr, args);
    va_end(args);

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

358 359
    if (client == RT_NULL)
    {
360
        LOG_E("input AT client object is NULL, please create or get AT Client object!");
361
        return -RT_ERROR;
362 363
    }

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

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

420 421 422 423 424 425
    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 已提交
426
#ifdef AT_PRINT_RAW_CMD
427
    at_print_raw_cmd("sendline", buf, size);
Lawlieta's avatar
Lawlieta 已提交
428 429 430 431 432
#endif

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

433
static rt_err_t at_client_getchar(at_client_t client, char *ch, rt_int32_t timeout)
Lawlieta's avatar
Lawlieta 已提交
434
{
435
    rt_err_t result = RT_EOK;
Lawlieta's avatar
Lawlieta 已提交
436

437
    while (rt_device_read(client->device, 0, ch, 1) == 0)
438
    {
439 440 441 442 443
        result = rt_sem_take(client->rx_notice, rt_tick_from_millisecond(timeout));
        if (result != RT_EOK)
        {
            return result;
        }
W
Wayne Lin 已提交
444 445

        rt_sem_control(client->rx_notice, RT_IPC_CMD_RESET, RT_NULL);
446
    }
447 448

    return RT_EOK;
Lawlieta's avatar
Lawlieta 已提交
449 450 451 452 453
}

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

    RT_ASSERT(buf);

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

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

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

    return read_idx;
502 503 504 505 506 507 508 509 510 511
}

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

518
    client->end_sign = ch;
Lawlieta's avatar
Lawlieta 已提交
519 520 521
}

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

532 533 534
    if (client == RT_NULL)
    {
        LOG_E("input AT Client object is NULL, please create or get AT Client object!");
535
        return -RT_ERROR;
536 537
    }

538 539 540 541 542
    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 已提交
543

544 545 546 547 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
    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++;
580

581 582 583 584
        rt_free(old_urc_table);
    }

    return RT_EOK;
Lawlieta's avatar
Lawlieta 已提交
585 586 587
}

/**
588
 * get AT client object by AT device name.
Lawlieta's avatar
Lawlieta 已提交
589
 *
590
 * @dev_name AT client device name
Lawlieta's avatar
Lawlieta 已提交
591
 *
592
 * @return AT client object
Lawlieta's avatar
Lawlieta 已提交
593
 */
594
at_client_t at_client_get(const char *dev_name)
Lawlieta's avatar
Lawlieta 已提交
595
{
596
    int idx = 0;
Lawlieta's avatar
Lawlieta 已提交
597

598 599 600 601 602 603 604 605 606 607 608
    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 已提交
609 610
}

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

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

639 640
    buffer = client->recv_line_buf;
    bufsz = client->recv_line_len;
641

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

    return RT_NULL;
}

666
static int at_recv_readline(at_client_t client)
Lawlieta's avatar
Lawlieta 已提交
667 668 669 670 671
{
    rt_size_t read_len = 0;
    char ch = 0, last_ch = 0;
    rt_bool_t is_full = RT_FALSE;

672 673
    rt_memset(client->recv_line_buf, 0x00, client->recv_bufsz);
    client->recv_line_len = 0;
Lawlieta's avatar
Lawlieta 已提交
674 675 676

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

679
        if (read_len < client->recv_bufsz)
Lawlieta's avatar
Lawlieta 已提交
680
        {
681 682
            client->recv_line_buf[read_len++] = ch;
            client->recv_line_len = read_len;
Lawlieta's avatar
Lawlieta 已提交
683 684 685 686 687 688 689
        }
        else
        {
            is_full = RT_TRUE;
        }

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

#ifdef AT_PRINT_RAW_CMD
706
    at_print_raw_cmd("recvline", client->recv_line_buf, read_len);
Lawlieta's avatar
Lawlieta 已提交
707 708 709 710 711 712 713 714 715 716 717
#endif

    return read_len;
}

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

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

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

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

static rt_err_t at_client_rx_ind(rt_device_t dev, rt_size_t size)
{
783 784 785
    int idx = 0;

    for (idx = 0; idx < AT_CLIENT_NUM_MAX; idx++)
786
    {
787 788 789 790
        if (at_client_table[idx].device == dev && size > 0)
        {
            rt_sem_release(at_client_table[idx].rx_notice);
        }
791 792
    }

Lawlieta's avatar
Lawlieta 已提交
793 794 795
    return RT_EOK;
}

796 797
/* initialize the client object parameters */
static int at_client_para_init(at_client_t client)
Lawlieta's avatar
Lawlieta 已提交
798
{
799 800 801 802
#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 已提交
803 804

    int result = RT_EOK;
805 806
    static int at_client_num = 0;
    char name[RT_NAME_MAX];
Lawlieta's avatar
Lawlieta 已提交
807

808 809
    client->status = AT_STATUS_UNINITIALIZED;

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

819 820 821
    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 已提交
822
    {
823 824
        LOG_E("AT client initialize failed! at_client_recv_lock create failed!");
        result = -RT_ENOMEM;
Lawlieta's avatar
Lawlieta 已提交
825 826 827
        goto __exit;
    }

828 829 830
    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 已提交
831
    {
832
        LOG_E("AT client initialize failed! at_client_notice semaphore create failed!");
Lawlieta's avatar
Lawlieta 已提交
833 834 835 836
        result = -RT_ENOMEM;
        goto __exit;
    }

837 838 839
    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 已提交
840
    {
841
        LOG_E("AT client initialize failed! at_client_resp semaphore create failed!");
Lawlieta's avatar
Lawlieta 已提交
842 843 844 845
        result = -RT_ENOMEM;
        goto __exit;
    }

846 847 848 849 850 851 852 853 854 855 856
    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 已提交
857 858 859 860 861
    {
        result = -RT_ENOMEM;
        goto __exit;
    }

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

885
        if (client->recv_line_buf)
886
        {
887
            rt_free(client->recv_line_buf);
888 889 890 891 892 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
        }

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

920 921 922 923 924
    if (at_client_get(dev_name) != RT_NULL)
    {
        return result;
    }

925 926 927 928
    for (idx = 0; idx < AT_CLIENT_NUM_MAX && at_client_table[idx].device; idx++);

    if (idx >= AT_CLIENT_NUM_MAX)
    {
L
luofanlu 已提交
929
        LOG_E("AT client initialize failed! Check the maximum number(%d) of AT client.", AT_CLIENT_NUM_MAX);
930 931 932 933 934 935 936
        result = -RT_EFULL;
        goto __exit;
    }

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

937 938 939 940 941 942
    result = at_client_para_init(client);
    if (result != RT_EOK)
    {
        goto __exit;
    }

943 944 945 946 947
    /* 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 已提交
948

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

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

__exit:
968
    if (result == RT_EOK)
Lawlieta's avatar
Lawlieta 已提交
969
    {
970
        client->status = AT_STATUS_INITIALIZED;
Lawlieta's avatar
Lawlieta 已提交
971

972
        rt_thread_startup(client->parser);
Lawlieta's avatar
Lawlieta 已提交
973

974
        LOG_I("AT client(V%s) on device %s initialize success.", AT_SW_VERSION, dev_name);
Lawlieta's avatar
Lawlieta 已提交
975 976 977
    }
    else
    {
978
        LOG_E("AT client(V%s) on device %s initialize failed(%d).", AT_SW_VERSION, dev_name, result);
Lawlieta's avatar
Lawlieta 已提交
979 980 981 982
    }

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