at_client.c 25.9 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
11
 * 2021-03-17     Meco Man     fix a buf of leaking memory
S
sszl 已提交
12
 * 2021-07-14     Sszl         fix a buf of leaking memory
Lawlieta's avatar
Lawlieta 已提交
13 14 15 16 17 18 19
 */

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

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

#ifdef AT_USING_CLIENT

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

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

Comeon_fly's avatar
Comeon_fly 已提交
32 33 34 35
extern rt_size_t at_utils_send(rt_device_t dev,
                               rt_off_t    pos,
                               const void *buffer,
                               rt_size_t   size);
Lawlieta's avatar
Lawlieta 已提交
36 37 38 39 40
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);

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

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

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

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

/**
99
 * Set response object information
Lawlieta's avatar
Lawlieta 已提交
100
 *
101
 * @param resp response object
Lawlieta's avatar
Lawlieta 已提交
102 103 104 105 106 107
 * @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
 *
108
 * @return  != RT_NULL: response object
Lawlieta's avatar
Lawlieta 已提交
109 110 111 112
 *           = 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 已提交
113
    char *p_temp;
Lawlieta's avatar
Lawlieta 已提交
114 115
    RT_ASSERT(resp);

116
    if (resp->buf_size != buf_size)
Lawlieta's avatar
Lawlieta 已提交
117 118 119
    {
        resp->buf_size = buf_size;

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

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

    return resp;
}

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

    RT_ASSERT(cmd_expr);

298 299 300 301 302 303
    if (client == RT_NULL)
    {
        LOG_E("input AT Client object is NULL, please create or get AT Client object!");
        return -RT_ERROR;
    }

304 305 306 307 308 309
    /* check AT CLI mode */
    if (client->status == AT_STATUS_CLI && resp)
    {
        return -RT_EBUSY;
    }

Lawlieta's avatar
Lawlieta 已提交
310 311 312 313
    rt_mutex_take(client->lock, RT_WAITING_FOREVER);

    client->resp_status = AT_RESP_OK;

314 315 316 317 318 319
    if (resp != RT_NULL)
    {
        resp->buf_len = 0;
        resp->line_counts = 0;
    }

Comeon_fly's avatar
Comeon_fly 已提交
320 321 322
    client->resp = resp;
    rt_sem_control(client->resp_notice, RT_IPC_CMD_RESET, RT_NULL);

Lawlieta's avatar
Lawlieta 已提交
323 324 325 326
    va_start(args, cmd_expr);
    at_vprintfln(client->device, cmd_expr, args);
    va_end(args);

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

371 372
    if (client == RT_NULL)
    {
373
        LOG_E("input AT client object is NULL, please create or get AT Client object!");
374
        return -RT_ERROR;
375 376
    }

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

    rt_mutex_take(client->lock, RT_WAITING_FOREVER);
    client->resp = resp;
Comeon_fly's avatar
Comeon_fly 已提交
386
    rt_sem_control(client->resp_notice, RT_IPC_CMD_RESET, RT_NULL);
Z
zylx 已提交
387 388 389 390 391 392

    start_time = rt_tick_get();

    while (1)
    {
        /* Check whether it is timeout */
393
        if (rt_tick_get() - start_time > rt_tick_from_millisecond(timeout))
Z
zylx 已提交
394
        {
395
            LOG_E("wait AT client(%s) connect timeout(%d tick).", client_name, timeout);
Z
zylx 已提交
396 397 398 399 400
            result = -RT_ETIMEOUT;
            break;
        }

        /* Check whether it is already connected */
401
        resp->buf_len = 0;
Z
zylx 已提交
402
        resp->line_counts = 0;
Comeon_fly's avatar
Comeon_fly 已提交
403
        at_utils_send(client->device, 0, "AT\r\n", 4);
Z
zylx 已提交
404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419

        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 已提交
420 421 422
/**
 * Send data to AT server, send data don't have end sign(eg: \r\n).
 *
423
 * @param client current AT client object
Lawlieta's avatar
Lawlieta 已提交
424 425 426
 * @param buf   send data buffer
 * @param size  send fixed data size
 *
427 428
 * @return >0: send data size
 *         =0: send failed
Lawlieta's avatar
Lawlieta 已提交
429
 */
430
rt_size_t at_client_obj_send(at_client_t client, const char *buf, rt_size_t size)
Lawlieta's avatar
Lawlieta 已提交
431 432 433
{
    RT_ASSERT(buf);

434 435 436 437 438 439
    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 已提交
440
#ifdef AT_PRINT_RAW_CMD
441
    at_print_raw_cmd("sendline", buf, size);
Lawlieta's avatar
Lawlieta 已提交
442 443
#endif

Comeon_fly's avatar
Comeon_fly 已提交
444 445 446 447 448 449 450
    rt_mutex_take(client->lock, RT_WAITING_FOREVER);

    rt_size_t len = at_utils_send(client->device, 0, buf, size);

    rt_mutex_release(client->lock);

    return len;
Lawlieta's avatar
Lawlieta 已提交
451 452
}

453
static rt_err_t at_client_getchar(at_client_t client, char *ch, rt_int32_t timeout)
Lawlieta's avatar
Lawlieta 已提交
454
{
455
    rt_err_t result = RT_EOK;
Lawlieta's avatar
Lawlieta 已提交
456

457
    while (rt_device_read(client->device, 0, ch, 1) == 0)
458
    {
459 460 461 462 463
        result = rt_sem_take(client->rx_notice, rt_tick_from_millisecond(timeout));
        if (result != RT_EOK)
        {
            return result;
        }
W
Wayne Lin 已提交
464 465

        rt_sem_control(client->rx_notice, RT_IPC_CMD_RESET, RT_NULL);
466
    }
467 468

    return RT_EOK;
Lawlieta's avatar
Lawlieta 已提交
469 470 471 472 473
}

/**
 * AT client receive fixed-length data.
 *
474
 * @param client current AT client object
Lawlieta's avatar
Lawlieta 已提交
475 476
 * @param buf   receive data buffer
 * @param size  receive fixed data size
477
 * @param timeout  receive data timeout (ms)
Lawlieta's avatar
Lawlieta 已提交
478 479 480
 *
 * @note this function can only be used in execution function of URC data
 *
481 482
 * @return >0: receive data size
 *         =0: receive failed
Lawlieta's avatar
Lawlieta 已提交
483
 */
484
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 已提交
485
{
Comeon_fly's avatar
Comeon_fly 已提交
486
    rt_size_t len = 0;
Lawlieta's avatar
Lawlieta 已提交
487 488 489

    RT_ASSERT(buf);

490 491 492 493 494 495
    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 已提交
496 497
    while (1)
    {
Comeon_fly's avatar
Comeon_fly 已提交
498 499 500 501
        rt_sem_control(client->rx_notice, RT_IPC_CMD_RESET, RT_NULL);

        rt_size_t read_len = rt_device_read(client->device, 0, buf + len, size);
        if(read_len > 0)
Lawlieta's avatar
Lawlieta 已提交
502
        {
Comeon_fly's avatar
Comeon_fly 已提交
503 504 505 506 507 508
            len += read_len;
            size -= read_len;
            if(size == 0)
                break;

            continue;
Lawlieta's avatar
Lawlieta 已提交
509
        }
Comeon_fly's avatar
Comeon_fly 已提交
510 511

        if(rt_sem_take(client->rx_notice, rt_tick_from_millisecond(timeout)) != RT_EOK)
Lawlieta's avatar
Lawlieta 已提交
512 513 514 515
            break;
    }

#ifdef AT_PRINT_RAW_CMD
Comeon_fly's avatar
Comeon_fly 已提交
516
    at_print_raw_cmd("urc_recv", buf, len);
Lawlieta's avatar
Lawlieta 已提交
517 518
#endif

Comeon_fly's avatar
Comeon_fly 已提交
519
    return len;
520 521 522 523 524 525 526 527 528 529
}

/**
 *  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)
{
530 531 532 533 534
    if (client == RT_NULL)
    {
        LOG_E("input AT Client object is NULL, please create or get AT Client object!");
        return;
    }
Lawlieta's avatar
Lawlieta 已提交
535

536
    client->end_sign = ch;
Lawlieta's avatar
Lawlieta 已提交
537 538 539
}

/**
540
 * set URC(Unsolicited Result Code) table
Lawlieta's avatar
Lawlieta 已提交
541
 *
542 543 544
 * @param client current AT client object
 * @param table URC table
 * @param size table size
Lawlieta's avatar
Lawlieta 已提交
545
 */
546
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 已提交
547
{
548 549
    rt_size_t idx;

550 551 552
    if (client == RT_NULL)
    {
        LOG_E("input AT Client object is NULL, please create or get AT Client object!");
553
        return -RT_ERROR;
554 555
    }

556 557 558 559 560
    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 已提交
561

562 563 564 565 566 567 568 569 570 571 572 573 574 575
    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
    {
S
sszl 已提交
576
        struct at_urc_table *new_urc_table = RT_NULL;
577

578
        /* realloc urc table space */
S
sszl 已提交
579 580
        new_urc_table = (struct at_urc_table *) rt_realloc(client->urc_table,client->urc_table_size * sizeof(struct at_urc_table) + sizeof(struct at_urc_table));
        if (new_urc_table == RT_NULL)
581 582 583
        {
            return -RT_ENOMEM;
        }
S
sszl 已提交
584
        client->urc_table = new_urc_table;
585 586 587
        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++;
588

589 590 591
    }

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

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

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

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

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

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

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

    return RT_NULL;
}

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

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

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

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

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

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

    return read_len;
}

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

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

739 740
                char end_ch = client->recv_line_buf[client->recv_line_len - 1];

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

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

static rt_err_t at_client_rx_ind(rt_device_t dev, rt_size_t size)
{
797 798 799
    int idx = 0;

    for (idx = 0; idx < AT_CLIENT_NUM_MAX; idx++)
800
    {
801 802 803 804
        if (at_client_table[idx].device == dev && size > 0)
        {
            rt_sem_release(at_client_table[idx].rx_notice);
        }
805 806
    }

Lawlieta's avatar
Lawlieta 已提交
807 808 809
    return RT_EOK;
}

810 811
/* initialize the client object parameters */
static int at_client_para_init(at_client_t client)
Lawlieta's avatar
Lawlieta 已提交
812
{
813 814 815 816
#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 已提交
817 818

    int result = RT_EOK;
819 820
    static int at_client_num = 0;
    char name[RT_NAME_MAX];
Lawlieta's avatar
Lawlieta 已提交
821

822 823
    client->status = AT_STATUS_UNINITIALIZED;

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

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

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

851 852 853
    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 已提交
854
    {
855
        LOG_E("AT client initialize failed! at_client_resp semaphore create failed!");
Lawlieta's avatar
Lawlieta 已提交
856 857 858 859
        result = -RT_ENOMEM;
        goto __exit;
    }

860 861 862 863 864 865 866 867 868 869 870
    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 已提交
871 872 873 874 875
    {
        result = -RT_ENOMEM;
        goto __exit;
    }

876 877
__exit:
    if (result != RT_EOK)
Lawlieta's avatar
Lawlieta 已提交
878
    {
879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898
        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);
        }

899
        if (client->recv_line_buf)
900
        {
901
            rt_free(client->recv_line_buf);
902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933
        }

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

934 935 936 937 938
    if (at_client_get(dev_name) != RT_NULL)
    {
        return result;
    }

939 940 941 942
    for (idx = 0; idx < AT_CLIENT_NUM_MAX && at_client_table[idx].device; idx++);

    if (idx >= AT_CLIENT_NUM_MAX)
    {
L
luofanlu 已提交
943
        LOG_E("AT client initialize failed! Check the maximum number(%d) of AT client.", AT_CLIENT_NUM_MAX);
944 945 946 947 948 949 950
        result = -RT_EFULL;
        goto __exit;
    }

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

951 952 953 954 955 956
    result = at_client_para_init(client);
    if (result != RT_EOK)
    {
        goto __exit;
    }

957 958 959 960 961
    /* 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 已提交
962

963
        /* using DMA mode first */
964
        open_result = rt_device_open(client->device, RT_DEVICE_OFLAG_RDWR | RT_DEVICE_FLAG_DMA_RX);
965 966 967
        /* using interrupt mode when DMA mode not supported */
        if (open_result == -RT_EIO)
        {
968
            open_result = rt_device_open(client->device, RT_DEVICE_OFLAG_RDWR | RT_DEVICE_FLAG_INT_RX);
969 970
        }
        RT_ASSERT(open_result == RT_EOK);
Lawlieta's avatar
Lawlieta 已提交
971

972
        rt_device_set_rx_indicate(client->device, at_client_rx_ind);
Lawlieta's avatar
Lawlieta 已提交
973 974 975
    }
    else
    {
976
        LOG_E("AT client initialize failed! Not find the device(%s).", dev_name);
Lawlieta's avatar
Lawlieta 已提交
977 978 979 980 981
        result = -RT_ERROR;
        goto __exit;
    }

__exit:
982
    if (result == RT_EOK)
Lawlieta's avatar
Lawlieta 已提交
983
    {
984
        client->status = AT_STATUS_INITIALIZED;
Lawlieta's avatar
Lawlieta 已提交
985

986
        rt_thread_startup(client->parser);
Lawlieta's avatar
Lawlieta 已提交
987

988
        LOG_I("AT client(V%s) on device %s initialize success.", AT_SW_VERSION, dev_name);
Lawlieta's avatar
Lawlieta 已提交
989 990 991
    }
    else
    {
992
        LOG_E("AT client(V%s) on device %s initialize failed(%d).", AT_SW_VERSION, dev_name, result);
Lawlieta's avatar
Lawlieta 已提交
993 994 995 996
    }

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