mem.S 1.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394
  1. /*
  2. * String handling functions.
  3. *
  4. * Copyright IBM Corp. 2012
  5. */
  6. #include <linux/linkage.h>
  7. #include <asm/nospec-insn.h>
  8. GEN_BR_THUNK %r14
  9. /*
  10. * memset implementation
  11. *
  12. * This code corresponds to the C construct below. We do distinguish
  13. * between clearing (c == 0) and setting a memory array (c != 0) simply
  14. * because nearly all memset invocations in the kernel clear memory and
  15. * the xc instruction is preferred in such cases.
  16. *
  17. * void *memset(void *s, int c, size_t n)
  18. * {
  19. * if (likely(c == 0))
  20. * return __builtin_memset(s, 0, n);
  21. * return __builtin_memset(s, c, n);
  22. * }
  23. */
  24. ENTRY(memset)
  25. ltgr %r4,%r4
  26. jz .Lmemset_exit
  27. ltgr %r3,%r3
  28. jnz .Lmemset_fill
  29. aghi %r4,-1
  30. srlg %r3,%r4,8
  31. ltgr %r3,%r3
  32. lgr %r1,%r2
  33. jz .Lmemset_clear_rest
  34. .Lmemset_clear_loop:
  35. xc 0(256,%r1),0(%r1)
  36. la %r1,256(%r1)
  37. brctg %r3,.Lmemset_clear_loop
  38. .Lmemset_clear_rest:
  39. larl %r3,.Lmemset_xc
  40. ex %r4,0(%r3)
  41. .Lmemset_exit:
  42. BR_EX %r14
  43. .Lmemset_fill:
  44. stc %r3,0(%r2)
  45. cghi %r4,1
  46. lgr %r1,%r2
  47. je .Lmemset_fill_exit
  48. aghi %r4,-2
  49. srlg %r3,%r4,8
  50. ltgr %r3,%r3
  51. jz .Lmemset_fill_rest
  52. .Lmemset_fill_loop:
  53. mvc 1(256,%r1),0(%r1)
  54. la %r1,256(%r1)
  55. brctg %r3,.Lmemset_fill_loop
  56. .Lmemset_fill_rest:
  57. larl %r3,.Lmemset_mvc
  58. ex %r4,0(%r3)
  59. .Lmemset_fill_exit:
  60. BR_EX %r14
  61. .Lmemset_xc:
  62. xc 0(1,%r1),0(%r1)
  63. .Lmemset_mvc:
  64. mvc 1(1,%r1),0(%r1)
  65. /*
  66. * memcpy implementation
  67. *
  68. * void *memcpy(void *dest, const void *src, size_t n)
  69. */
  70. ENTRY(memcpy)
  71. ltgr %r4,%r4
  72. jz .Lmemcpy_exit
  73. aghi %r4,-1
  74. srlg %r5,%r4,8
  75. ltgr %r5,%r5
  76. lgr %r1,%r2
  77. jnz .Lmemcpy_loop
  78. .Lmemcpy_rest:
  79. larl %r5,.Lmemcpy_mvc
  80. ex %r4,0(%r5)
  81. .Lmemcpy_exit:
  82. BR_EX %r14
  83. .Lmemcpy_loop:
  84. mvc 0(256,%r1),0(%r3)
  85. la %r1,256(%r1)
  86. la %r3,256(%r3)
  87. brctg %r5,.Lmemcpy_loop
  88. j .Lmemcpy_rest
  89. .Lmemcpy_mvc:
  90. mvc 0(1,%r1),0(%r3)