1*19c3b8c2SApple OSS Distributions /*-
2*19c3b8c2SApple OSS Distributions * Copyright (c) 2009 Xin LI <[email protected]>
3*19c3b8c2SApple OSS Distributions * All rights reserved.
4*19c3b8c2SApple OSS Distributions *
5*19c3b8c2SApple OSS Distributions * Redistribution and use in source and binary forms, with or without
6*19c3b8c2SApple OSS Distributions * modification, are permitted provided that the following conditions
7*19c3b8c2SApple OSS Distributions * are met:
8*19c3b8c2SApple OSS Distributions * 1. Redistributions of source code must retain the above copyright
9*19c3b8c2SApple OSS Distributions * notice, this list of conditions and the following disclaimer.
10*19c3b8c2SApple OSS Distributions * 2. Redistributions in binary form must reproduce the above copyright
11*19c3b8c2SApple OSS Distributions * notice, this list of conditions and the following disclaimer in the
12*19c3b8c2SApple OSS Distributions * documentation and/or other materials provided with the distribution.
13*19c3b8c2SApple OSS Distributions *
14*19c3b8c2SApple OSS Distributions * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15*19c3b8c2SApple OSS Distributions * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16*19c3b8c2SApple OSS Distributions * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17*19c3b8c2SApple OSS Distributions * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18*19c3b8c2SApple OSS Distributions * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19*19c3b8c2SApple OSS Distributions * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20*19c3b8c2SApple OSS Distributions * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21*19c3b8c2SApple OSS Distributions * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22*19c3b8c2SApple OSS Distributions * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23*19c3b8c2SApple OSS Distributions * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24*19c3b8c2SApple OSS Distributions * SUCH DAMAGE.
25*19c3b8c2SApple OSS Distributions */
26*19c3b8c2SApple OSS Distributions
27*19c3b8c2SApple OSS Distributions #include "string.h"
28*19c3b8c2SApple OSS Distributions #include <limits.h>
29*19c3b8c2SApple OSS Distributions
30*19c3b8c2SApple OSS Distributions /*
31*19c3b8c2SApple OSS Distributions * Portable strlen() for 32-bit and 64-bit systems.
32*19c3b8c2SApple OSS Distributions *
33*19c3b8c2SApple OSS Distributions * Rationale: it is generally much more efficient to do word length
34*19c3b8c2SApple OSS Distributions * operations and avoid branches on modern computer systems, as
35*19c3b8c2SApple OSS Distributions * compared to byte-length operations with a lot of branches.
36*19c3b8c2SApple OSS Distributions *
37*19c3b8c2SApple OSS Distributions * The expression:
38*19c3b8c2SApple OSS Distributions *
39*19c3b8c2SApple OSS Distributions * ((x - 0x01....01) & ~x & 0x80....80)
40*19c3b8c2SApple OSS Distributions *
41*19c3b8c2SApple OSS Distributions * would evaluate to a non-zero value iff any of the bytes in the
42*19c3b8c2SApple OSS Distributions * original word is zero. However, we can further reduce ~1/3 of
43*19c3b8c2SApple OSS Distributions * time if we consider that strlen() usually operate on 7-bit ASCII
44*19c3b8c2SApple OSS Distributions * by employing the following expression, which allows false positive
45*19c3b8c2SApple OSS Distributions * when high bit of 1 and use the tail case to catch these case:
46*19c3b8c2SApple OSS Distributions *
47*19c3b8c2SApple OSS Distributions * ((x - 0x01....01) & 0x80....80)
48*19c3b8c2SApple OSS Distributions *
49*19c3b8c2SApple OSS Distributions * This is more than 5.2 times as fast as the raw implementation on
50*19c3b8c2SApple OSS Distributions * Intel T7300 under long mode for strings longer than word length.
51*19c3b8c2SApple OSS Distributions */
52*19c3b8c2SApple OSS Distributions
53*19c3b8c2SApple OSS Distributions /* Magic numbers for the algorithm */
54*19c3b8c2SApple OSS Distributions #if LONG_BIT == 32
55*19c3b8c2SApple OSS Distributions static const unsigned long mask01 = 0x01010101;
56*19c3b8c2SApple OSS Distributions static const unsigned long mask80 = 0x80808080;
57*19c3b8c2SApple OSS Distributions #elif LONG_BIT == 64
58*19c3b8c2SApple OSS Distributions static const unsigned long mask01 = 0x0101010101010101;
59*19c3b8c2SApple OSS Distributions static const unsigned long mask80 = 0x8080808080808080;
60*19c3b8c2SApple OSS Distributions #else
61*19c3b8c2SApple OSS Distributions #error Unsupported word size
62*19c3b8c2SApple OSS Distributions #endif
63*19c3b8c2SApple OSS Distributions
64*19c3b8c2SApple OSS Distributions #define LONGPTR_MASK (sizeof(long) - 1)
65*19c3b8c2SApple OSS Distributions
66*19c3b8c2SApple OSS Distributions /*
67*19c3b8c2SApple OSS Distributions * Helper macro to return string length if we caught the zero
68*19c3b8c2SApple OSS Distributions * byte.
69*19c3b8c2SApple OSS Distributions */
70*19c3b8c2SApple OSS Distributions #define testbyte(x) \
71*19c3b8c2SApple OSS Distributions do { \
72*19c3b8c2SApple OSS Distributions if (p[x] == '\0') \
73*19c3b8c2SApple OSS Distributions return (p - str + x); \
74*19c3b8c2SApple OSS Distributions } while (0)
75*19c3b8c2SApple OSS Distributions
76*19c3b8c2SApple OSS Distributions __attribute__((visibility("hidden")))
77*19c3b8c2SApple OSS Distributions size_t
_libkernel_strlen(const char * str)78*19c3b8c2SApple OSS Distributions _libkernel_strlen(const char *str)
79*19c3b8c2SApple OSS Distributions {
80*19c3b8c2SApple OSS Distributions const char *p;
81*19c3b8c2SApple OSS Distributions const unsigned long *lp;
82*19c3b8c2SApple OSS Distributions
83*19c3b8c2SApple OSS Distributions /* Skip the first few bytes until we have an aligned p */
84*19c3b8c2SApple OSS Distributions for (p = str; (uintptr_t)p & LONGPTR_MASK; p++) {
85*19c3b8c2SApple OSS Distributions if (*p == '\0') {
86*19c3b8c2SApple OSS Distributions return p - str;
87*19c3b8c2SApple OSS Distributions }
88*19c3b8c2SApple OSS Distributions }
89*19c3b8c2SApple OSS Distributions
90*19c3b8c2SApple OSS Distributions /* Scan the rest of the string using word sized operation */
91*19c3b8c2SApple OSS Distributions for (lp = (const unsigned long *)p;; lp++) {
92*19c3b8c2SApple OSS Distributions if ((*lp - mask01) & mask80) {
93*19c3b8c2SApple OSS Distributions p = (const char *)(lp);
94*19c3b8c2SApple OSS Distributions testbyte(0);
95*19c3b8c2SApple OSS Distributions testbyte(1);
96*19c3b8c2SApple OSS Distributions testbyte(2);
97*19c3b8c2SApple OSS Distributions testbyte(3);
98*19c3b8c2SApple OSS Distributions #if (LONG_BIT >= 64)
99*19c3b8c2SApple OSS Distributions testbyte(4);
100*19c3b8c2SApple OSS Distributions testbyte(5);
101*19c3b8c2SApple OSS Distributions testbyte(6);
102*19c3b8c2SApple OSS Distributions testbyte(7);
103*19c3b8c2SApple OSS Distributions #endif
104*19c3b8c2SApple OSS Distributions }
105*19c3b8c2SApple OSS Distributions }
106*19c3b8c2SApple OSS Distributions
107*19c3b8c2SApple OSS Distributions /* NOTREACHED */
108*19c3b8c2SApple OSS Distributions return 0;
109*19c3b8c2SApple OSS Distributions }
110