| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 1 | /* Copyright 2002 Andi Kleen, SuSE Labs */ | 
| Jan Beulich | 8d379da | 2006-09-26 10:52:32 +0200 | [diff] [blame] | 2 |  | 
| Jan Beulich | 8d379da | 2006-09-26 10:52:32 +0200 | [diff] [blame] | 3 | #include <linux/linkage.h> | 
|  | 4 | #include <asm/dwarf2.h> | 
|  | 5 |  | 
| Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 6 | /* | 
|  | 7 | * ISO C memset - set a memory block to a byte value. | 
|  | 8 | * | 
|  | 9 | * rdi   destination | 
|  | 10 | * rsi   value (char) | 
|  | 11 | * rdx   count (bytes) | 
|  | 12 | * | 
|  | 13 | * rax   original destination | 
|  | 14 | */ | 
| Jan Beulich | 8d379da | 2006-09-26 10:52:32 +0200 | [diff] [blame] | 15 | ALIGN | 
|  | 16 | memset_c: | 
|  | 17 | CFI_STARTPROC | 
|  | 18 | movq %rdi,%r9 | 
|  | 19 | movl %edx,%r8d | 
|  | 20 | andl $7,%r8d | 
|  | 21 | movl %edx,%ecx | 
|  | 22 | shrl $3,%ecx | 
|  | 23 | /* expand byte value  */ | 
|  | 24 | movzbl %sil,%esi | 
|  | 25 | movabs $0x0101010101010101,%rax | 
|  | 26 | mulq %rsi		/* with rax, clobbers rdx */ | 
|  | 27 | rep stosq | 
|  | 28 | movl %r8d,%ecx | 
|  | 29 | rep stosb | 
|  | 30 | movq %r9,%rax | 
|  | 31 | ret | 
|  | 32 | CFI_ENDPROC | 
|  | 33 | ENDPROC(memset_c) | 
|  | 34 |  | 
|  | 35 | ENTRY(memset) | 
|  | 36 | ENTRY(__memset) | 
|  | 37 | CFI_STARTPROC | 
| Andi Kleen | 7bcd3f3 | 2006-02-03 21:51:02 +0100 | [diff] [blame] | 38 | movq %rdi,%r10 | 
|  | 39 | movq %rdx,%r11 | 
|  | 40 |  | 
|  | 41 | /* expand byte value  */ | 
|  | 42 | movzbl %sil,%ecx | 
|  | 43 | movabs $0x0101010101010101,%rax | 
|  | 44 | mul    %rcx		/* with rax, clobbers rdx */ | 
|  | 45 |  | 
|  | 46 | /* align dst */ | 
|  | 47 | movl  %edi,%r9d | 
|  | 48 | andl  $7,%r9d | 
|  | 49 | jnz  .Lbad_alignment | 
| Jan Beulich | 8d379da | 2006-09-26 10:52:32 +0200 | [diff] [blame] | 50 | CFI_REMEMBER_STATE | 
| Andi Kleen | 7bcd3f3 | 2006-02-03 21:51:02 +0100 | [diff] [blame] | 51 | .Lafter_bad_alignment: | 
|  | 52 |  | 
|  | 53 | movl %r11d,%ecx | 
|  | 54 | shrl $6,%ecx | 
|  | 55 | jz	 .Lhandle_tail | 
|  | 56 |  | 
|  | 57 | .p2align 4 | 
|  | 58 | .Lloop_64: | 
|  | 59 | decl   %ecx | 
|  | 60 | movq  %rax,(%rdi) | 
|  | 61 | movq  %rax,8(%rdi) | 
|  | 62 | movq  %rax,16(%rdi) | 
|  | 63 | movq  %rax,24(%rdi) | 
|  | 64 | movq  %rax,32(%rdi) | 
|  | 65 | movq  %rax,40(%rdi) | 
|  | 66 | movq  %rax,48(%rdi) | 
|  | 67 | movq  %rax,56(%rdi) | 
|  | 68 | leaq  64(%rdi),%rdi | 
|  | 69 | jnz    .Lloop_64 | 
|  | 70 |  | 
|  | 71 | /* Handle tail in loops. The loops should be faster than hard | 
|  | 72 | to predict jump tables. */ | 
|  | 73 | .p2align 4 | 
|  | 74 | .Lhandle_tail: | 
|  | 75 | movl	%r11d,%ecx | 
|  | 76 | andl    $63&(~7),%ecx | 
|  | 77 | jz 		.Lhandle_7 | 
|  | 78 | shrl	$3,%ecx | 
|  | 79 | .p2align 4 | 
|  | 80 | .Lloop_8: | 
|  | 81 | decl   %ecx | 
|  | 82 | movq  %rax,(%rdi) | 
|  | 83 | leaq  8(%rdi),%rdi | 
|  | 84 | jnz    .Lloop_8 | 
|  | 85 |  | 
|  | 86 | .Lhandle_7: | 
|  | 87 | movl	%r11d,%ecx | 
|  | 88 | andl	$7,%ecx | 
|  | 89 | jz      .Lende | 
|  | 90 | .p2align 4 | 
|  | 91 | .Lloop_1: | 
|  | 92 | decl    %ecx | 
|  | 93 | movb 	%al,(%rdi) | 
|  | 94 | leaq	1(%rdi),%rdi | 
|  | 95 | jnz     .Lloop_1 | 
|  | 96 |  | 
|  | 97 | .Lende: | 
|  | 98 | movq	%r10,%rax | 
|  | 99 | ret | 
|  | 100 |  | 
| Jan Beulich | 8d379da | 2006-09-26 10:52:32 +0200 | [diff] [blame] | 101 | CFI_RESTORE_STATE | 
| Andi Kleen | 7bcd3f3 | 2006-02-03 21:51:02 +0100 | [diff] [blame] | 102 | .Lbad_alignment: | 
|  | 103 | cmpq $7,%r11 | 
|  | 104 | jbe	.Lhandle_7 | 
|  | 105 | movq %rax,(%rdi)	/* unaligned store */ | 
|  | 106 | movq $8,%r8 | 
|  | 107 | subq %r9,%r8 | 
|  | 108 | addq %r8,%rdi | 
|  | 109 | subq %r8,%r11 | 
|  | 110 | jmp .Lafter_bad_alignment | 
| Jan Beulich | 8d379da | 2006-09-26 10:52:32 +0200 | [diff] [blame] | 111 | .Lfinal: | 
|  | 112 | CFI_ENDPROC | 
|  | 113 | ENDPROC(memset) | 
|  | 114 | ENDPROC(__memset) | 
| Andi Kleen | 7bcd3f3 | 2006-02-03 21:51:02 +0100 | [diff] [blame] | 115 |  | 
|  | 116 | /* Some CPUs run faster using the string instructions. | 
|  | 117 | It is also a lot simpler. Use this when possible */ | 
|  | 118 |  | 
|  | 119 | #include <asm/cpufeature.h> | 
|  | 120 |  | 
| Jan Beulich | 8d379da | 2006-09-26 10:52:32 +0200 | [diff] [blame] | 121 | .section .altinstr_replacement,"ax" | 
|  | 122 | 1:	.byte 0xeb				/* jmp <disp8> */ | 
|  | 123 | .byte (memset_c - memset) - (2f - 1b)	/* offset */ | 
|  | 124 | 2: | 
|  | 125 | .previous | 
| Andi Kleen | 7bcd3f3 | 2006-02-03 21:51:02 +0100 | [diff] [blame] | 126 | .section .altinstructions,"a" | 
|  | 127 | .align 8 | 
| Jan Beulich | 8d379da | 2006-09-26 10:52:32 +0200 | [diff] [blame] | 128 | .quad memset | 
|  | 129 | .quad 1b | 
|  | 130 | .byte X86_FEATURE_REP_GOOD | 
|  | 131 | .byte .Lfinal - memset | 
|  | 132 | .byte 2b - 1b | 
| Andi Kleen | 7bcd3f3 | 2006-02-03 21:51:02 +0100 | [diff] [blame] | 133 | .previous |