Commit | Line | Data |
---|---|---|
eea9507a LFT |
1 | /* |
2 | * Copyright (C) 2011 Tobias Klauser <tklauser@distanz.ch> | |
3 | * Copyright (C) 2004 Microtronix Datacom Ltd | |
4 | * | |
5 | * This file is subject to the terms and conditions of the GNU General Public | |
6 | * License. See the file "COPYING" in the main directory of this archive | |
7 | * for more details. | |
8 | */ | |
9 | ||
10 | #include <linux/types.h> | |
11 | #include <linux/string.h> | |
12 | ||
eea9507a LFT |
13 | void *memset(void *s, int c, size_t count) |
14 | { | |
15 | int destptr, charcnt, dwordcnt, fill8reg, wrkrega; | |
16 | ||
17 | if (!count) | |
18 | return s; | |
19 | ||
20 | c &= 0xFF; | |
21 | ||
22 | if (count <= 8) { | |
23 | char *xs = (char *) s; | |
24 | ||
25 | while (count--) | |
26 | *xs++ = c; | |
27 | return s; | |
28 | } | |
29 | ||
30 | __asm__ __volatile__ ( | |
31 | /* fill8 %3, %5 (c & 0xff) */ | |
32 | " slli %4, %5, 8\n" | |
33 | " or %4, %4, %5\n" | |
34 | " slli %3, %4, 16\n" | |
35 | " or %3, %3, %4\n" | |
36 | /* Word-align %0 (s) if necessary */ | |
37 | " andi %4, %0, 0x01\n" | |
38 | " beq %4, zero, 1f\n" | |
39 | " addi %1, %1, -1\n" | |
40 | " stb %3, 0(%0)\n" | |
41 | " addi %0, %0, 1\n" | |
42 | "1: mov %2, %1\n" | |
43 | /* Dword-align %0 (s) if necessary */ | |
44 | " andi %4, %0, 0x02\n" | |
45 | " beq %4, zero, 2f\n" | |
46 | " addi %1, %1, -2\n" | |
47 | " sth %3, 0(%0)\n" | |
48 | " addi %0, %0, 2\n" | |
49 | " mov %2, %1\n" | |
50 | /* %1 and %2 are how many more bytes to set */ | |
51 | "2: srli %2, %2, 2\n" | |
52 | /* %2 is how many dwords to set */ | |
53 | "3: stw %3, 0(%0)\n" | |
54 | " addi %0, %0, 4\n" | |
55 | " addi %2, %2, -1\n" | |
56 | " bne %2, zero, 3b\n" | |
57 | /* store residual word and/or byte if necessary */ | |
58 | " andi %4, %1, 0x02\n" | |
59 | " beq %4, zero, 4f\n" | |
60 | " sth %3, 0(%0)\n" | |
61 | " addi %0, %0, 2\n" | |
62 | /* store residual byte if necessary */ | |
63 | "4: andi %4, %1, 0x01\n" | |
64 | " beq %4, zero, 5f\n" | |
65 | " stb %3, 0(%0)\n" | |
66 | "5:\n" | |
67 | : "=r" (destptr), /* %0 Output */ | |
68 | "=r" (charcnt), /* %1 Output */ | |
69 | "=r" (dwordcnt), /* %2 Output */ | |
70 | "=r" (fill8reg), /* %3 Output */ | |
a8950e49 | 71 | "=&r" (wrkrega) /* %4 Output only */ |
eea9507a LFT |
72 | : "r" (c), /* %5 Input */ |
73 | "0" (s), /* %0 Input/Output */ | |
74 | "1" (count) /* %1 Input/Output */ | |
75 | : "memory" /* clobbered */ | |
76 | ); | |
77 | ||
78 | return s; | |
79 | } |