kservice.c 32.6 KB
Newer Older
1
/*
mysterywolf's avatar
mysterywolf 已提交
2
 * Copyright (c) 2006-2021, RT-Thread Development Team
B
Bernard Xiong 已提交
3
 *
4
 * SPDX-License-Identifier: Apache-2.0
5 6 7 8 9 10
 *
 * Change Logs:
 * Date           Author       Notes
 * 2006-03-16     Bernard      the first version
 * 2006-05-25     Bernard      rewrite vsprintf
 * 2006-08-10     Bernard      add rt_show_version
B
bernard.xiong 已提交
11 12
 * 2010-03-17     Bernard      remove rt_strlcpy function
 *                             fix gcc compiling issue.
13
 * 2010-04-15     Bernard      remove weak definition on ICCM16C compiler
14
 * 2012-07-18     Arda         add the alignment display for signed integer
B
Bernard Xiong 已提交
15
 * 2012-11-23     Bernard      fix IAR compiler error.
16
 * 2012-12-22     Bernard      fix rt_kprintf issue, which found by Grissiom.
B
Bernard Xiong 已提交
17
 * 2013-06-24     Bernard      remove rt_kprintf if RT_USING_CONSOLE is not defined.
18
 * 2013-09-24     aozima       make sure the device is in STREAM mode when used by rt_kprintf.
19
 * 2015-07-06     Bernard      Add rt_assert_handler routine.
mysterywolf's avatar
mysterywolf 已提交
20
 * 2021-02-28     Meco Man     add RT_KSERVICE_USING_STDLIB
21 22 23 24 25
 */

#include <rtthread.h>
#include <rthw.h>

26 27
#ifdef RT_USING_MODULE
#include <dlmodule.h>
28
#endif /* RT_USING_MODULE */
29

30 31 32
/* use precision */
#define RT_PRINTF_PRECISION

33 34 35
/**
 * @addtogroup KernelService
 */
D
dzzxzz 已提交
36

D
dogandog 已提交
37
/**@{*/
38

D
dzzxzz 已提交
39
/* global errno in RT-Thread */
B
bernard 已提交
40
static volatile int __rt_errno;
41

42
#if defined(RT_USING_DEVICE) && defined(RT_USING_CONSOLE)
B
bernard.xiong 已提交
43
static rt_device_t _console_device = RT_NULL;
44
#endif
45 46 47

/*
 * This function will get errno
B
bernard.xiong 已提交
48
 *
49 50 51 52
 * @return errno
 */
rt_err_t rt_get_errno(void)
{
53
    rt_thread_t tid;
B
bernard.xiong 已提交
54

55 56 57
    if (rt_interrupt_get_nest() != 0)
    {
        /* it's in interrupt context */
B
bernard 已提交
58
        return __rt_errno;
59
    }
60

61 62
    tid = rt_thread_self();
    if (tid == RT_NULL)
B
bernard 已提交
63
        return __rt_errno;
B
bernard.xiong 已提交
64

65
    return tid->error;
66
}
67
RTM_EXPORT(rt_get_errno);
68 69 70 71 72 73 74 75

/*
 * This function will set errno
 *
 * @param error the errno shall be set
 */
void rt_set_errno(rt_err_t error)
{
76
    rt_thread_t tid;
B
bernard.xiong 已提交
77

78 79 80
    if (rt_interrupt_get_nest() != 0)
    {
        /* it's in interrupt context */
B
bernard 已提交
81
        __rt_errno = error;
D
dzzxzz 已提交
82

83 84
        return;
    }
85

86 87 88
    tid = rt_thread_self();
    if (tid == RT_NULL)
    {
B
bernard 已提交
89
        __rt_errno = error;
B
Bernard Xiong 已提交
90

91 92
        return;
    }
B
bernard.xiong 已提交
93

94
    tid->error = error;
95
}
96
RTM_EXPORT(rt_set_errno);
97

98 99 100 101 102 103 104
/**
 * This function returns errno.
 *
 * @return the errno in the system
 */
int *_rt_errno(void)
{
105
    rt_thread_t tid;
B
Bernard Xiong 已提交
106

107
    if (rt_interrupt_get_nest() != 0)
B
bernard 已提交
108
        return (int *)&__rt_errno;
109

110 111
    tid = rt_thread_self();
    if (tid != RT_NULL)
112
        return (int *) & (tid->error);
113

B
bernard 已提交
114
    return (int *)&__rt_errno;
115
}
116
RTM_EXPORT(_rt_errno);
117

118 119 120 121 122 123 124 125 126
/**
 * This function will set the content of memory to specified value
 *
 * @param s the address of source memory
 * @param c the value shall be set in content
 * @param count the copied length
 *
 * @return the address of source memory
 */
127
RT_WEAK void *rt_memset(void *s, int c, rt_ubase_t count)
128
{
mysterywolf's avatar
mysterywolf 已提交
129
#ifdef RT_KSERVICE_USING_TINY_SIZE
130
    char *xs = (char *)s;
131

132 133
    while (count--)
        *xs++ = c;
134

135
    return s;
136
#else
137 138
#define LBLOCKSIZE      (sizeof(long))
#define UNALIGNED(X)    ((long)X & (LBLOCKSIZE - 1))
D
dzzxzz 已提交
139
#define TOO_SMALL(LEN)  ((LEN) < LBLOCKSIZE)
140

141
    unsigned int i;
142
    char *m = (char *)s;
143 144 145 146
    unsigned long buffer;
    unsigned long *aligned_addr;
    unsigned int d = c & 0xff;  /* To avoid sign extension, copy C to an
                                unsigned variable.  */
147 148 149

    if (!TOO_SMALL(count) && !UNALIGNED(s))
    {
150
        /* If we get this far, we know that count is large and s is word-aligned. */
151
        aligned_addr = (unsigned long *)s;
152

153
        /* Store d into each char sized location in buffer so that
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 179 180 181 182 183 184 185 186 187 188 189 190 191 192
         * we can set large blocks quickly.
         */
        if (LBLOCKSIZE == 4)
        {
            buffer = (d << 8) | d;
            buffer |= (buffer << 16);
        }
        else
        {
            buffer = 0;
            for (i = 0; i < LBLOCKSIZE; i ++)
                buffer = (buffer << 8) | d;
        }

        while (count >= LBLOCKSIZE * 4)
        {
            *aligned_addr++ = buffer;
            *aligned_addr++ = buffer;
            *aligned_addr++ = buffer;
            *aligned_addr++ = buffer;
            count -= 4 * LBLOCKSIZE;
        }

        while (count >= LBLOCKSIZE)
        {
            *aligned_addr++ = buffer;
            count -= LBLOCKSIZE;
        }

        /* Pick up the remainder with a bytewise loop. */
        m = (char *)aligned_addr;
    }

    while (count--)
    {
        *m++ = (char)d;
    }

    return s;
193 194 195 196

#undef LBLOCKSIZE
#undef UNALIGNED
#undef TOO_SMALL
197
#endif /* RT_KSERVICE_USING_TINY_SIZE */
198
}
199
RTM_EXPORT(rt_memset);
200 201 202 203 204 205 206 207 208 209 210

/**
 * This function will copy memory content from source address to destination
 * address.
 *
 * @param dst the address of destination memory
 * @param src  the address of source memory
 * @param count the copied length
 *
 * @return the address of destination memory
 */
211
RT_WEAK void *rt_memcpy(void *dst, const void *src, rt_ubase_t count)
212
{
mysterywolf's avatar
mysterywolf 已提交
213
#ifdef RT_KSERVICE_USING_TINY_SIZE
214
    char *tmp = (char *)dst, *s = (char *)src;
215
    rt_ubase_t len;
E
emlslxl 已提交
216 217

    if (tmp <= s || tmp > (s + count))
218 219 220 221 222 223
    {
        while (count--)
            *tmp ++ = *s ++;
    }
    else
    {
E
emlslxl 已提交
224 225
        for (len = count; len > 0; len --)
            tmp[len - 1] = s[len - 1];
226
    }
227

E
emlslxl 已提交
228
    return dst;
229 230
#else

231 232 233 234
#define UNALIGNED(X, Y) \
    (((long)X & (sizeof (long) - 1)) | ((long)Y & (sizeof (long) - 1)))
#define BIGBLOCKSIZE    (sizeof (long) << 2)
#define LITTLEBLOCKSIZE (sizeof (long))
235 236
#define TOO_SMALL(LEN)  ((LEN) < BIGBLOCKSIZE)

237 238
    char *dst_ptr = (char *)dst;
    char *src_ptr = (char *)src;
239 240
    long *aligned_dst;
    long *aligned_src;
241 242 243 244 245 246
    int len = count;

    /* If the size is small, or either SRC or DST is unaligned,
    then punt into the byte copy loop.  This should be rare. */
    if (!TOO_SMALL(len) && !UNALIGNED(src_ptr, dst_ptr))
    {
247 248
        aligned_dst = (long *)dst_ptr;
        aligned_src = (long *)src_ptr;
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

        /* Copy 4X long words at a time if possible. */
        while (len >= BIGBLOCKSIZE)
        {
            *aligned_dst++ = *aligned_src++;
            *aligned_dst++ = *aligned_src++;
            *aligned_dst++ = *aligned_src++;
            *aligned_dst++ = *aligned_src++;
            len -= BIGBLOCKSIZE;
        }

        /* Copy one long word at a time if possible. */
        while (len >= LITTLEBLOCKSIZE)
        {
            *aligned_dst++ = *aligned_src++;
            len -= LITTLEBLOCKSIZE;
        }

        /* Pick up any residual with a byte copier. */
        dst_ptr = (char *)aligned_dst;
        src_ptr = (char *)aligned_src;
    }

    while (len--)
        *dst_ptr++ = *src_ptr++;

    return dst;
276 277 278 279
#undef UNALIGNED
#undef BIGBLOCKSIZE
#undef LITTLEBLOCKSIZE
#undef TOO_SMALL
280
#endif /* RT_KSERVICE_USING_TINY_SIZE */
281
}
282
RTM_EXPORT(rt_memcpy);
283

mysterywolf's avatar
mysterywolf 已提交
284
#ifndef RT_KSERVICE_USING_STDLIB
mysterywolf's avatar
mysterywolf 已提交
285

286 287 288 289 290 291 292 293 294 295
/**
 * This function will move memory content from source address to destination
 * address.
 *
 * @param dest the address of destination memory
 * @param src  the address of source memory
 * @param n the copied length
 *
 * @return the address of destination memory
 */
D
dzzxzz 已提交
296
void *rt_memmove(void *dest, const void *src, rt_ubase_t n)
297
{
298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314
    char *tmp = (char *)dest, *s = (char *)src;

    if (s < tmp && tmp < s + n)
    {
        tmp += n;
        s += n;

        while (n--)
            *(--tmp) = *(--s);
    }
    else
    {
        while (n--)
            *tmp++ = *s++;
    }

    return dest;
315
}
316
RTM_EXPORT(rt_memmove);
317 318

/**
D
dzzxzz 已提交
319 320 321
 * This function will compare two areas of memory
 *
 * @param cs one area of memory
B
Bernard Xiong 已提交
322
 * @param ct another area of memory
D
dzzxzz 已提交
323 324 325
 * @param count the size of the area
 *
 * @return the result
326
 */
327
RT_WEAK rt_int32_t rt_memcmp(const void *cs, const void *ct, rt_ubase_t count)
328
{
329 330
    const unsigned char *su1, *su2;
    int res = 0;
331

332
    for (su1 = (const unsigned char *)cs, su2 = (const unsigned char *)ct; 0 < count; ++su1, ++su2, count--)
333 334
        if ((res = *su1 - *su2) != 0)
            break;
D
dzzxzz 已提交
335

336
    return res;
337
}
338
RTM_EXPORT(rt_memcmp);
339 340 341 342 343 344 345 346 347

/**
 * This function will return the first occurrence of a string.
 *
 * @param s1 the source string
 * @param s2 the find string
 *
 * @return the first occurrence of a s2 in s1, or RT_NULL if no found.
 */
D
dzzxzz 已提交
348
char *rt_strstr(const char *s1, const char *s2)
349
{
350 351 352 353 354 355 356 357 358 359 360 361 362 363 364
    int l1, l2;

    l2 = rt_strlen(s2);
    if (!l2)
        return (char *)s1;
    l1 = rt_strlen(s1);
    while (l1 >= l2)
    {
        l1 --;
        if (!rt_memcmp(s1, s2, l2))
            return (char *)s1;
        s1 ++;
    }

    return RT_NULL;
365
}
366
RTM_EXPORT(rt_strstr);
367 368 369 370 371 372 373 374 375

/**
 * This function will compare two strings while ignoring differences in case
 *
 * @param a the string to be compared
 * @param b the string to be compared
 *
 * @return the result
 */
376
rt_int32_t rt_strcasecmp(const char *a, const char *b)
377
{
378 379 380 381 382 383 384 385 386 387 388 389 390 391
    int ca, cb;

    do
    {
        ca = *a++ & 0xff;
        cb = *b++ & 0xff;
        if (ca >= 'A' && ca <= 'Z')
            ca += 'a' - 'A';
        if (cb >= 'A' && cb <= 'Z')
            cb += 'a' - 'A';
    }
    while (ca == cb && ca != '\0');

    return ca - cb;
392
}
393
RTM_EXPORT(rt_strcasecmp);
394 395 396 397

/**
 * This function will copy string no more than n bytes.
 *
398
 * @param dst the string to copy
399 400 401 402 403
 * @param src the string to be copied
 * @param n the maximum copied length
 *
 * @return the result
 */
404
char *rt_strncpy(char *dst, const char *src, rt_ubase_t n)
405
{
406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423
    if (n != 0)
    {
        char *d = dst;
        const char *s = src;

        do
        {
            if ((*d++ = *s++) == 0)
            {
                /* NUL pad the remaining n-1 bytes */
                while (--n != 0)
                    *d++ = 0;
                break;
            }
        } while (--n != 0);
    }

    return (dst);
424
}
425
RTM_EXPORT(rt_strncpy);
426 427 428 429 430 431 432 433 434 435

/**
 * This function will compare two strings with specified maximum length
 *
 * @param cs the string to be compared
 * @param ct the string to be compared
 * @param count the maximum compare length
 *
 * @return the result
 */
436
rt_int32_t rt_strncmp(const char *cs, const char *ct, rt_ubase_t count)
437
{
438
    register signed char __res = 0;
439

440 441 442 443 444 445
    while (count)
    {
        if ((__res = *cs - *ct++) != 0 || !*cs++)
            break;
        count --;
    }
446

447
    return __res;
448
}
449
RTM_EXPORT(rt_strncmp);
450

451 452 453 454 455 456 457 458
/**
 * This function will compare two strings without specified length
 *
 * @param cs the string to be compared
 * @param ct the string to be compared
 *
 * @return the result
 */
459
rt_int32_t rt_strcmp(const char *cs, const char *ct)
460
{
461
    while (*cs && *cs == *ct)
mysterywolf's avatar
mysterywolf 已提交
462
    {
463 464 465
        cs++;
        ct++;
    }
D
dzzxzz 已提交
466

467
    return (*cs - *ct);
468
}
469
RTM_EXPORT(rt_strcmp);
470

471 472
/**
 * The  strnlen()  function  returns the number of characters in the
473 474 475
 * string pointed to by s, excluding the terminating null byte ('\0'),
 * but at most maxlen.  In doing this, strnlen() looks only at the
 * first maxlen characters in the string pointed to by s and never
476 477 478 479 480 481 482 483 484 485
 * beyond s+maxlen.
 *
 * @param s the string
 * @param maxlen the max size
 * @return the length of string
 */
rt_size_t rt_strnlen(const char *s, rt_ubase_t maxlen)
{
    const char *sc;

486
    for (sc = s; *sc != '\0' && (rt_ubase_t)(sc - s) < maxlen; ++sc) /* nothing */
487
        ;
488

489 490
    return sc - s;
}
491 492
RTM_EXPORT(rt_strnlen);

493 494 495 496 497 498 499 500
/**
 * This function will return the length of a string, which terminate will
 * null character.
 *
 * @param s the string
 *
 * @return the length of string
 */
501
rt_size_t rt_strlen(const char *s)
502
{
503
    const char *sc;
504

505 506
    for (sc = s; *sc != '\0'; ++sc) /* nothing */
        ;
507

508
    return sc - s;
509
}
510
RTM_EXPORT(rt_strlen);
511

512
#endif /* RT_KSERVICE_USING_STDLIB */
mysterywolf's avatar
mysterywolf 已提交
513

514 515 516 517 518 519 520 521 522 523
#ifdef RT_USING_HEAP
/**
 * This function will duplicate a string.
 *
 * @param s the string to be duplicated
 *
 * @return the duplicated string pointer
 */
char *rt_strdup(const char *s)
{
524 525
    rt_size_t len = rt_strlen(s) + 1;
    char *tmp = (char *)rt_malloc(len);
526

527 528
    if (!tmp)
        return RT_NULL;
529

530
    rt_memcpy(tmp, s, len);
D
dzzxzz 已提交
531

532
    return tmp;
533
}
534
RTM_EXPORT(rt_strdup);
B
Bernard Xiong 已提交
535
#if defined(__CC_ARM) || defined(__CLANG_ARM)
B
Bernard Xiong 已提交
536 537
char *strdup(const char *s) __attribute__((alias("rt_strdup")));
#endif
538
#endif /* RT_USING_HEAP */
539 540 541 542

/**
 * This function will show the version of rt-thread rtos
 */
D
dzzxzz 已提交
543
void rt_show_version(void)
544
{
545 546 547 548
    rt_kprintf("\n \\ | /\n");
    rt_kprintf("- RT -     Thread Operating System\n");
    rt_kprintf(" / | \\     %d.%d.%d build %s\n",
               RT_VERSION, RT_SUBVERSION, RT_REVISION, __DATE__);
549
    rt_kprintf(" 2006 - 2021 Copyright by rt-thread team\n");
550
}
551
RTM_EXPORT(rt_show_version);
552 553

/* private function */
554
#define _ISDIGIT(c)  ((unsigned)((c) - '0') < 10)
555

H
HubretXie 已提交
556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575
#ifdef RT_PRINTF_LONGLONG
rt_inline int divide(long long *n, int base)
{
    int res;

    /* optimized for processor which does not support divide instructions. */
    if (base == 10)
    {
        res = (int)(((unsigned long long)*n) % 10U);
        *n = (long long)(((unsigned long long)*n) / 10U);
    }
    else
    {
        res = (int)(((unsigned long long)*n) % 16U);
        *n = (long long)(((unsigned long long)*n) / 16U);
    }

    return res;
}
#else
576
rt_inline int divide(long *n, int base)
577
{
578
    int res;
579 580 581 582

    /* optimized for processor which does not support divide instructions. */
    if (base == 10)
    {
583 584
        res = (int)(((unsigned long)*n) % 10U);
        *n = (long)(((unsigned long)*n) / 10U);
585 586 587
    }
    else
    {
588 589
        res = (int)(((unsigned long)*n) % 16U);
        *n = (long)(((unsigned long)*n) / 16U);
590 591 592
    }

    return res;
593
}
594
#endif /* RT_PRINTF_LONGLONG */
595 596 597

rt_inline int skip_atoi(const char **s)
{
598
    register int i = 0;
599
    while (_ISDIGIT(**s))
600
        i = i * 10 + *((*s)++) - '0';
601

602
    return i;
603 604
}

605 606 607 608 609 610 611
#define ZEROPAD     (1 << 0)    /* pad with zero */
#define SIGN        (1 << 1)    /* unsigned/signed long */
#define PLUS        (1 << 2)    /* show plus */
#define SPACE       (1 << 3)    /* space if plus */
#define LEFT        (1 << 4)    /* left justified */
#define SPECIAL     (1 << 5)    /* 0x */
#define LARGE       (1 << 6)    /* use 'ABCDEF' instead of 'abcdef' */
612 613

#ifdef RT_PRINTF_PRECISION
614 615
static char *print_number(char *buf,
                          char *end,
H
HubretXie 已提交
616 617 618
#ifdef RT_PRINTF_LONGLONG
                          long long  num,
#else
619
                          long  num,
620
#endif /* RT_PRINTF_LONGLONG */
621 622 623 624
                          int   base,
                          int   s,
                          int   precision,
                          int   type)
625
#else
626 627
static char *print_number(char *buf,
                          char *end,
H
HubretXie 已提交
628 629 630
#ifdef RT_PRINTF_LONGLONG
                          long long  num,
#else
631
                          long  num,
632
#endif /* RT_PRINTF_LONGLONG */
633 634 635
                          int   base,
                          int   s,
                          int   type)
636
#endif /* RT_PRINTF_PRECISION */
637
{
638
    char c, sign;
639
#ifdef RT_PRINTF_LONGLONG
640
    char tmp[32];
641
#else
642
    char tmp[16];
643
#endif /* RT_PRINTF_LONGLONG */
644
    int precision_bak = precision;
645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672
    const char *digits;
    static const char small_digits[] = "0123456789abcdef";
    static const char large_digits[] = "0123456789ABCDEF";
    register int i;
    register int size;

    size = s;

    digits = (type & LARGE) ? large_digits : small_digits;
    if (type & LEFT)
        type &= ~ZEROPAD;

    c = (type & ZEROPAD) ? '0' : ' ';

    /* get sign */
    sign = 0;
    if (type & SIGN)
    {
        if (num < 0)
        {
            sign = '-';
            num = -num;
        }
        else if (type & PLUS)
            sign = '+';
        else if (type & SPACE)
            sign = ' ';
    }
673 674

#ifdef RT_PRINTF_SPECIAL
675 676 677 678 679 680 681
    if (type & SPECIAL)
    {
        if (base == 16)
            size -= 2;
        else if (base == 8)
            size--;
    }
682
#endif /* RT_PRINTF_SPECIAL */
683

684 685
    i = 0;
    if (num == 0)
686
        tmp[i++] = '0';
687 688 689 690 691
    else
    {
        while (num != 0)
            tmp[i++] = digits[divide(&num, base)];
    }
692 693

#ifdef RT_PRINTF_PRECISION
694 695 696
    if (i > precision)
        precision = i;
    size -= precision;
697
#else
698
    size -= i;
699
#endif /* RT_PRINTF_PRECISION */
700

701
    if (!(type & (ZEROPAD | LEFT)))
702
    {
703
        if ((sign) && (size > 0))
704 705
            size--;

706
        while (size-- > 0)
707
        {
708
            if (buf < end)
709 710 711 712 713 714 715
                *buf = ' ';
            ++ buf;
        }
    }

    if (sign)
    {
716
        if (buf < end)
717 718 719
        {
            *buf = sign;
        }
720
        -- size;
721 722
        ++ buf;
    }
723 724

#ifdef RT_PRINTF_SPECIAL
725 726
    if (type & SPECIAL)
    {
727
        if (base == 8)
728
        {
729
            if (buf < end)
730 731 732 733 734
                *buf = '0';
            ++ buf;
        }
        else if (base == 16)
        {
735
            if (buf < end)
736 737
                *buf = '0';
            ++ buf;
738
            if (buf < end)
739
            {
740
                *buf = type & LARGE ? 'X' : 'x';
741 742 743 744
            }
            ++ buf;
        }
    }
745
#endif /* RT_PRINTF_SPECIAL */
746

747 748 749 750 751
    /* no align to the left */
    if (!(type & LEFT))
    {
        while (size-- > 0)
        {
752
            if (buf < end)
753 754 755 756
                *buf = c;
            ++ buf;
        }
    }
757 758

#ifdef RT_PRINTF_PRECISION
759 760
    while (i < precision--)
    {
761
        if (buf < end)
762 763 764
            *buf = '0';
        ++ buf;
    }
765
#endif /* RT_PRINTF_PRECISION */
766

767
    /* put number in the temporary buffer */
768
    while (i-- > 0 && (precision_bak != 0))
769
    {
770
        if (buf < end)
771 772 773 774 775 776
            *buf = tmp[i];
        ++ buf;
    }

    while (size-- > 0)
    {
777
        if (buf < end)
778 779 780 781 782
            *buf = ' ';
        ++ buf;
    }

    return buf;
783 784
}

785 786 787 788
rt_int32_t rt_vsnprintf(char       *buf,
                        rt_size_t   size,
                        const char *fmt,
                        va_list     args)
789 790
{
#ifdef RT_PRINTF_LONGLONG
791
    unsigned long long num;
792
#else
793
    rt_uint32_t num;
794
#endif /* RT_PRINTF_LONGLONG */
795 796 797
    int i, len;
    char *str, *end, c;
    const char *s;
798

799 800 801 802
    rt_uint8_t base;            /* the base of number */
    rt_uint8_t flags;           /* flags to print number */
    rt_uint8_t qualifier;       /* 'h', 'l', or 'L' for integer fields */
    rt_int32_t field_width;     /* width of output field */
803 804

#ifdef RT_PRINTF_PRECISION
805
    int precision;      /* min. # of digits for integers and max for a string */
806
#endif /* RT_PRINTF_PRECISION */
807

808
    str = buf;
809
    end = buf + size;
810 811 812 813

    /* Make sure end is always >= buf */
    if (end < buf)
    {
814
        end  = ((char *) - 1);
815 816 817 818 819 820 821
        size = end - buf;
    }

    for (; *fmt ; ++fmt)
    {
        if (*fmt != '%')
        {
822
            if (str < end)
823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844
                *str = *fmt;
            ++ str;
            continue;
        }

        /* process flags */
        flags = 0;

        while (1)
        {
            /* skips the first '%' also */
            ++ fmt;
            if (*fmt == '-') flags |= LEFT;
            else if (*fmt == '+') flags |= PLUS;
            else if (*fmt == ' ') flags |= SPACE;
            else if (*fmt == '#') flags |= SPECIAL;
            else if (*fmt == '0') flags |= ZEROPAD;
            else break;
        }

        /* get field width */
        field_width = -1;
845
        if (_ISDIGIT(*fmt)) field_width = skip_atoi(&fmt);
846 847 848 849 850 851 852 853 854 855 856
        else if (*fmt == '*')
        {
            ++ fmt;
            /* it's the next argument */
            field_width = va_arg(args, int);
            if (field_width < 0)
            {
                field_width = -field_width;
                flags |= LEFT;
            }
        }
857 858

#ifdef RT_PRINTF_PRECISION
859 860 861 862 863
        /* get the precision */
        precision = -1;
        if (*fmt == '.')
        {
            ++ fmt;
864
            if (_ISDIGIT(*fmt)) precision = skip_atoi(&fmt);
865 866 867 868 869 870 871 872
            else if (*fmt == '*')
            {
                ++ fmt;
                /* it's the next argument */
                precision = va_arg(args, int);
            }
            if (precision < 0) precision = 0;
        }
873
#endif /* RT_PRINTF_PRECISION */
874 875
        /* get the conversion qualifier */
        qualifier = 0;
876
#ifdef RT_PRINTF_LONGLONG
877
        if (*fmt == 'h' || *fmt == 'l' || *fmt == 'L')
878
#else
879
        if (*fmt == 'h' || *fmt == 'l')
880
#endif /* RT_PRINTF_LONGLONG */
881 882 883
        {
            qualifier = *fmt;
            ++ fmt;
884
#ifdef RT_PRINTF_LONGLONG
885 886 887 888 889
            if (qualifier == 'l' && *fmt == 'l')
            {
                qualifier = 'L';
                ++ fmt;
            }
890
#endif /* RT_PRINTF_LONGLONG */
891 892 893 894 895 896 897 898 899 900 901 902
        }

        /* the default base */
        base = 10;

        switch (*fmt)
        {
        case 'c':
            if (!(flags & LEFT))
            {
                while (--field_width > 0)
                {
903
                    if (str < end) *str = ' ';
904 905 906 907 908 909
                    ++ str;
                }
            }

            /* get character */
            c = (rt_uint8_t)va_arg(args, int);
910
            if (str < end) *str = c;
911 912 913 914 915
            ++ str;

            /* put width */
            while (--field_width > 0)
            {
916
                if (str < end) *str = ' ';
917 918 919 920 921 922 923 924
                ++ str;
            }
            continue;

        case 's':
            s = va_arg(args, char *);
            if (!s) s = "(NULL)";

925
            for (len = 0; (len != field_width) && (s[len] != '\0'); len++);
926
#ifdef RT_PRINTF_PRECISION
927
            if (precision > 0 && len > precision) len = precision;
928
#endif /* RT_PRINTF_PRECISION */
929

930 931 932 933
            if (!(flags & LEFT))
            {
                while (len < field_width--)
                {
934
                    if (str < end) *str = ' ';
935 936 937 938 939 940
                    ++ str;
                }
            }

            for (i = 0; i < len; ++i)
            {
941
                if (str < end) *str = *s;
942 943 944 945 946 947
                ++ str;
                ++ s;
            }

            while (len < field_width--)
            {
948
                if (str < end) *str = ' ';
949 950 951 952 953 954 955 956 957 958
                ++ str;
            }
            continue;

        case 'p':
            if (field_width == -1)
            {
                field_width = sizeof(void *) << 1;
                flags |= ZEROPAD;
            }
959
#ifdef RT_PRINTF_PRECISION
960 961 962
            str = print_number(str, end,
                               (long)va_arg(args, void *),
                               16, field_width, precision, flags);
963
#else
964 965 966
            str = print_number(str, end,
                               (long)va_arg(args, void *),
                               16, field_width, flags);
967
#endif /* RT_PRINTF_PRECISION */
968 969 970
            continue;

        case '%':
971
            if (str < end) *str = '%';
972 973 974
            ++ str;
            continue;

975
        /* integer number formats - set up the flags and "break" */
976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992
        case 'o':
            base = 8;
            break;

        case 'X':
            flags |= LARGE;
        case 'x':
            base = 16;
            break;

        case 'd':
        case 'i':
            flags |= SIGN;
        case 'u':
            break;

        default:
993
            if (str < end) *str = '%';
994 995 996 997
            ++ str;

            if (*fmt)
            {
998
                if (str < end) *str = *fmt;
999 1000 1001 1002 1003 1004 1005 1006
                ++ str;
            }
            else
            {
                -- fmt;
            }
            continue;
        }
1007 1008

#ifdef RT_PRINTF_LONGLONG
1009 1010
        if (qualifier == 'L') num = va_arg(args, long long);
        else if (qualifier == 'l')
1011
#else
1012
        if (qualifier == 'l')
1013
#endif /* RT_PRINTF_LONGLONG */
1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027
        {
            num = va_arg(args, rt_uint32_t);
            if (flags & SIGN) num = (rt_int32_t)num;
        }
        else if (qualifier == 'h')
        {
            num = (rt_uint16_t)va_arg(args, rt_int32_t);
            if (flags & SIGN) num = (rt_int16_t)num;
        }
        else
        {
            num = va_arg(args, rt_uint32_t);
            if (flags & SIGN) num = (rt_int32_t)num;
        }
1028
#ifdef RT_PRINTF_PRECISION
1029
        str = print_number(str, end, num, base, field_width, precision, flags);
1030
#else
1031
        str = print_number(str, end, num, base, field_width, flags);
1032
#endif /* RT_PRINTF_PRECISION */
1033
    }
1034

1035 1036 1037 1038 1039 1040 1041 1042
    if (size > 0)
    {
        if (str < end) *str = '\0';
        else
        {
            end[-1] = '\0';
        }
    }
1043

1044 1045 1046 1047
    /* the trailing null byte doesn't count towards the total
    * ++str;
    */
    return str - buf;
1048
}
1049
RTM_EXPORT(rt_vsnprintf);
1050 1051 1052 1053 1054 1055 1056 1057 1058 1059

/**
 * This function will fill a formatted string to buffer
 *
 * @param buf the buffer to save formatted string
 * @param size the size of buffer
 * @param fmt the format
 */
rt_int32_t rt_snprintf(char *buf, rt_size_t size, const char *fmt, ...)
{
1060 1061
    rt_int32_t n;
    va_list args;
1062

1063
    va_start(args, fmt);
1064
    n = rt_vsnprintf(buf, size, fmt, args);
1065
    va_end(args);
1066

1067
    return n;
1068
}
1069
RTM_EXPORT(rt_snprintf);
1070 1071 1072 1073 1074

/**
 * This function will fill a formatted string to buffer
 *
 * @param buf the buffer to save formatted string
B
bernard.xiong 已提交
1075
 * @param arg_ptr the arg_ptr
1076 1077 1078 1079
 * @param format the format
 */
rt_int32_t rt_vsprintf(char *buf, const char *format, va_list arg_ptr)
{
1080
    return rt_vsnprintf(buf, (rt_size_t) - 1, format, arg_ptr);
1081
}
1082
RTM_EXPORT(rt_vsprintf);
1083 1084 1085 1086 1087 1088 1089

/**
 * This function will fill a formatted string to buffer
 *
 * @param buf the buffer to save formatted string
 * @param format the format
 */
D
dzzxzz 已提交
1090
rt_int32_t rt_sprintf(char *buf, const char *format, ...)
1091
{
1092 1093
    rt_int32_t n;
    va_list arg_ptr;
B
bernard.xiong 已提交
1094

1095
    va_start(arg_ptr, format);
1096
    n = rt_vsprintf(buf, format, arg_ptr);
1097
    va_end(arg_ptr);
B
bernard.xiong 已提交
1098

1099
    return n;
1100
}
1101
RTM_EXPORT(rt_sprintf);
B
bernard.xiong 已提交
1102

1103 1104
#ifdef RT_USING_CONSOLE

1105
#ifdef RT_USING_DEVICE
1106 1107 1108 1109 1110 1111 1112
/**
 * This function returns the device using in console.
 *
 * @return the device using in console or RT_NULL
 */
rt_device_t rt_console_get_device(void)
{
1113
    return _console_device;
1114
}
1115
RTM_EXPORT(rt_console_get_device);
1116

1117
/**
B
bernard.xiong@gmail.com 已提交
1118
 * This function will set a device as console device.
1119
 * After set a device to console, all output of rt_kprintf will be
B
bernard.xiong@gmail.com 已提交
1120
 * redirected to this new device.
1121
 *
B
bernard.xiong@gmail.com 已提交
1122
 * @param name the name of new console device
1123
 *
1124
 * @return the old console device handler on successful, or RT_NULL on failure.
1125
 */
D
dzzxzz 已提交
1126
rt_device_t rt_console_set_device(const char *name)
1127
{
1128
    rt_device_t new_device, old_device;
1129 1130

    /* save old device */
1131
    old_device = _console_device;
1132 1133

    /* find new console device */
1134
    new_device = rt_device_find(name);
mysterywolf's avatar
mysterywolf 已提交
1135

1136 1137
    /* check whether it's a same device */
    if (new_device == old_device) return RT_NULL;
mysterywolf's avatar
mysterywolf 已提交
1138

1139
    if (new_device != RT_NULL)
1140 1141 1142 1143 1144 1145 1146 1147
    {
        if (_console_device != RT_NULL)
        {
            /* close old console device */
            rt_device_close(_console_device);
        }

        /* set new console device */
1148 1149
        rt_device_open(new_device, RT_DEVICE_OFLAG_RDWR | RT_DEVICE_FLAG_STREAM);
        _console_device = new_device;
1150 1151
    }

1152
    return old_device;
B
bernard.xiong 已提交
1153
}
1154
RTM_EXPORT(rt_console_set_device);
1155
#endif /* RT_USING_DEVICE */
1156

X
xieyangrun 已提交
1157
RT_WEAK void rt_hw_console_output(const char *str)
1158
{
1159
    /* empty console output */
1160
}
1161
RTM_EXPORT(rt_hw_console_output);
1162

B
bernard 已提交
1163 1164 1165 1166 1167 1168 1169
/**
 * This function will put string to the console.
 *
 * @param str the string output to the console.
 */
void rt_kputs(const char *str)
{
B
bernard 已提交
1170 1171
    if (!str) return;

B
bernard 已提交
1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186
#ifdef RT_USING_DEVICE
    if (_console_device == RT_NULL)
    {
        rt_hw_console_output(str);
    }
    else
    {
        rt_uint16_t old_flag = _console_device->open_flag;

        _console_device->open_flag |= RT_DEVICE_FLAG_STREAM;
        rt_device_write(_console_device, 0, str, rt_strlen(str));
        _console_device->open_flag = old_flag;
    }
#else
    rt_hw_console_output(str);
1187
#endif /* RT_USING_DEVICE */
B
bernard 已提交
1188 1189
}

1190 1191 1192 1193 1194
/**
 * This function will print a formatted string on system console
 *
 * @param fmt the format
 */
1195
RT_WEAK void rt_kprintf(const char *fmt, ...)
1196
{
1197 1198 1199 1200 1201 1202 1203 1204 1205 1206
    va_list args;
    rt_size_t length;
    static char rt_log_buf[RT_CONSOLEBUF_SIZE];

    va_start(args, fmt);
    /* the return value of vsnprintf is the number of bytes that would be
     * written to buffer had if the size of the buffer been sufficiently
     * large excluding the terminating null byte. If the output string
     * would be larger than the rt_log_buf, we have to adjust the output
     * length. */
1207
    length = rt_vsnprintf(rt_log_buf, sizeof(rt_log_buf) - 1, fmt, args);
1208 1209
    if (length > RT_CONSOLEBUF_SIZE - 1)
        length = RT_CONSOLEBUF_SIZE - 1;
1210
#ifdef RT_USING_DEVICE
1211 1212 1213 1214 1215 1216
    if (_console_device == RT_NULL)
    {
        rt_hw_console_output(rt_log_buf);
    }
    else
    {
1217
        rt_uint16_t old_flag = _console_device->open_flag;
1218

1219
        _console_device->open_flag |= RT_DEVICE_FLAG_STREAM;
1220
        rt_device_write(_console_device, 0, rt_log_buf, length);
1221
        _console_device->open_flag = old_flag;
1222
    }
1223
#else
1224
    rt_hw_console_output(rt_log_buf);
1225
#endif /* RT_USING_DEVICE */
1226
    va_end(args);
1227
}
1228
RTM_EXPORT(rt_kprintf);
1229
#endif /* RT_USING_CONSOLE */
1230

1231 1232 1233 1234 1235 1236 1237 1238 1239 1240
#ifdef RT_USING_HEAP
/**
 * This function allocates a memory block, which address is aligned to the
 * specified alignment size.
 *
 * @param size the allocated memory block size
 * @param align the alignment size
 *
 * @return the allocated memory block on successful, otherwise returns RT_NULL
 */
1241
RT_WEAK void *rt_malloc_align(rt_size_t size, rt_size_t align)
1242
{
1243
    void *ptr;
B
Bernard Xiong 已提交
1244 1245
    void *align_ptr;
    int uintptr_size;
1246 1247
    rt_size_t align_size;

B
Bernard Xiong 已提交
1248 1249 1250 1251 1252 1253
    /* sizeof pointer */
    uintptr_size = sizeof(void*);
    uintptr_size -= 1;

    /* align the alignment size to uintptr size byte */
    align = ((align + uintptr_size) & ~uintptr_size);
1254 1255

    /* get total aligned size */
B
Bernard Xiong 已提交
1256
    align_size = ((size + uintptr_size) & ~uintptr_size) + align;
1257 1258 1259 1260
    /* allocate memory block from heap */
    ptr = rt_malloc(align_size);
    if (ptr != RT_NULL)
    {
1261
        /* the allocated memory block is aligned */
B
Bernard Xiong 已提交
1262
        if (((rt_ubase_t)ptr & (align - 1)) == 0)
1263
        {
B
Bernard Xiong 已提交
1264
            align_ptr = (void *)((rt_ubase_t)ptr + align);
1265 1266 1267
        }
        else
        {
B
Bernard Xiong 已提交
1268
            align_ptr = (void *)(((rt_ubase_t)ptr + (align - 1)) & ~(align - 1));
1269 1270 1271
        }

        /* set the pointer before alignment pointer to the real pointer */
B
Bernard Xiong 已提交
1272
        *((rt_ubase_t *)((rt_ubase_t)align_ptr - sizeof(void *))) = (rt_ubase_t)ptr;
1273 1274 1275 1276 1277

        ptr = align_ptr;
    }

    return ptr;
1278
}
1279
RTM_EXPORT(rt_malloc_align);
1280 1281

/**
1282 1283
 * This function release the memory block, which is allocated by
 * rt_malloc_align function and address is aligned.
1284 1285 1286
 *
 * @param ptr the memory block pointer
 */
1287
RT_WEAK void rt_free_align(void *ptr)
1288
{
1289
    void *real_ptr;
1290

B
Bernard Xiong 已提交
1291
    real_ptr = (void *) * (rt_ubase_t *)((rt_ubase_t)ptr - sizeof(void *));
1292
    rt_free(real_ptr);
1293
}
1294
RTM_EXPORT(rt_free_align);
1295
#endif /* RT_USING_HEAP */
1296

1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318
#ifndef RT_USING_CPU_FFS
const rt_uint8_t __lowest_bit_bitmap[] =
{
    /* 00 */ 0, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0,
    /* 10 */ 4, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0,
    /* 20 */ 5, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0,
    /* 30 */ 4, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0,
    /* 40 */ 6, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0,
    /* 50 */ 4, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0,
    /* 60 */ 5, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0,
    /* 70 */ 4, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0,
    /* 80 */ 7, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0,
    /* 90 */ 4, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0,
    /* A0 */ 5, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0,
    /* B0 */ 4, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0,
    /* C0 */ 6, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0,
    /* D0 */ 4, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0,
    /* E0 */ 5, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0,
    /* F0 */ 4, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0
};

/**
B
Bernard Xiong 已提交
1319
 * This function finds the first bit set (beginning with the least significant bit)
1320 1321
 * in value and return the index of that bit.
 *
B
Bernard Xiong 已提交
1322
 * Bits are numbered starting at 1 (the least significant bit).  A return value of
1323
 * zero from any of these functions means that the argument was zero.
B
Bernard Xiong 已提交
1324 1325
 *
 * @return return the index of the first bit set. If value is 0, then this function
1326 1327
 * shall return 0.
 */
B
bernard 已提交
1328
int __rt_ffs(int value)
1329 1330 1331 1332 1333 1334 1335 1336
{
    if (value == 0) return 0;

    if (value & 0xff)
        return __lowest_bit_bitmap[value & 0xff] + 1;

    if (value & 0xff00)
        return __lowest_bit_bitmap[(value & 0xff00) >> 8] + 9;
B
Bernard Xiong 已提交
1337

1338 1339
    if (value & 0xff0000)
        return __lowest_bit_bitmap[(value & 0xff0000) >> 16] + 17;
B
Bernard Xiong 已提交
1340

1341 1342
    return __lowest_bit_bitmap[(value & 0xff000000) >> 24] + 25;
}
1343
#endif /* RT_USING_CPU_FFS */
1344

armink_ztl's avatar
armink_ztl 已提交
1345 1346
#ifdef RT_DEBUG
/* RT_ASSERT(EX)'s hook */
1347

1348
void (*rt_assert_hook)(const char *ex, const char *func, rt_size_t line);
1349

armink_ztl's avatar
armink_ztl 已提交
1350 1351 1352 1353 1354
/**
 * This function will set a hook function to RT_ASSERT(EX). It will run when the expression is false.
 *
 * @param hook the hook function
 */
1355 1356
void rt_assert_set_hook(void (*hook)(const char *ex, const char *func, rt_size_t line))
{
armink_ztl's avatar
armink_ztl 已提交
1357 1358
    rt_assert_hook = hook;
}
1359 1360 1361 1362 1363 1364 1365 1366

/**
 * The RT_ASSERT function.
 *
 * @param ex the assertion condition string
 * @param func the function name when assertion.
 * @param line the file line number when assertion.
 */
1367
void rt_assert_handler(const char *ex_string, const char *func, rt_size_t line)
1368 1369 1370 1371 1372 1373
{
    volatile char dummy = 0;

    if (rt_assert_hook == RT_NULL)
    {
#ifdef RT_USING_MODULE
1374
        if (dlmodule_self())
1375
        {
1376 1377
            /* close assertion module */
            dlmodule_exit(-1);
1378 1379
        }
        else
1380
#endif /*RT_USING_MODULE*/
1381 1382 1383 1384
        {
            rt_kprintf("(%s) assertion failed at function:%s, line number:%d \n", ex_string, func, line);
            while (dummy == 0);
        }
1385
    }
1386 1387
    else
    {
1388
        rt_assert_hook(ex_string, func, line);
J
Jason Pan 已提交
1389
    }
1390 1391
}
RTM_EXPORT(rt_assert_handler);
armink_ztl's avatar
armink_ztl 已提交
1392 1393
#endif /* RT_DEBUG */

D
dogandog 已提交
1394
/**@}*/