memcpy_32.S 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616
  1. /*
  2. * Copyright 2010 Tilera Corporation. All Rights Reserved.
  3. *
  4. * This program is free software; you can redistribute it and/or
  5. * modify it under the terms of the GNU General Public License
  6. * as published by the Free Software Foundation, version 2.
  7. *
  8. * This program is distributed in the hope that it will be useful, but
  9. * WITHOUT ANY WARRANTY; without even the implied warranty of
  10. * MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE, GOOD TITLE or
  11. * NON INFRINGEMENT. See the GNU General Public License for
  12. * more details.
  13. *
  14. * This file shares the implementation of the userspace memcpy and
  15. * the kernel's memcpy, copy_to_user and copy_from_user.
  16. */
  17. #include <arch/chip.h>
  18. #include <linux/linkage.h>
  19. /* On TILE64, we wrap these functions via arch/tile/lib/memcpy_tile64.c */
  20. #if !CHIP_HAS_COHERENT_LOCAL_CACHE()
  21. #define memcpy __memcpy_asm
  22. #define __copy_to_user_inatomic __copy_to_user_inatomic_asm
  23. #define __copy_from_user_inatomic __copy_from_user_inatomic_asm
  24. #define __copy_from_user_zeroing __copy_from_user_zeroing_asm
  25. #endif
  26. #define IS_MEMCPY 0
  27. #define IS_COPY_FROM_USER 1
  28. #define IS_COPY_FROM_USER_ZEROING 2
  29. #define IS_COPY_TO_USER -1
  30. .section .text.memcpy_common, "ax"
  31. .align 64
  32. /* Use this to preface each bundle that can cause an exception so
  33. * the kernel can clean up properly. The special cleanup code should
  34. * not use these, since it knows what it is doing.
  35. */
  36. #define EX \
  37. .pushsection __ex_table, "a"; \
  38. .word 9f, memcpy_common_fixup; \
  39. .popsection; \
  40. 9
  41. /* __copy_from_user_inatomic takes the kernel target address in r0,
  42. * the user source in r1, and the bytes to copy in r2.
  43. * It returns the number of uncopiable bytes (hopefully zero) in r0.
  44. */
  45. ENTRY(__copy_from_user_inatomic)
  46. .type __copy_from_user_inatomic, @function
  47. FEEDBACK_ENTER_EXPLICIT(__copy_from_user_inatomic, \
  48. .text.memcpy_common, \
  49. .Lend_memcpy_common - __copy_from_user_inatomic)
  50. { movei r29, IS_COPY_FROM_USER; j memcpy_common }
  51. .size __copy_from_user_inatomic, . - __copy_from_user_inatomic
  52. /* __copy_from_user_zeroing is like __copy_from_user_inatomic, but
  53. * any uncopiable bytes are zeroed in the target.
  54. */
  55. ENTRY(__copy_from_user_zeroing)
  56. .type __copy_from_user_zeroing, @function
  57. FEEDBACK_REENTER(__copy_from_user_inatomic)
  58. { movei r29, IS_COPY_FROM_USER_ZEROING; j memcpy_common }
  59. .size __copy_from_user_zeroing, . - __copy_from_user_zeroing
  60. /* __copy_to_user_inatomic takes the user target address in r0,
  61. * the kernel source in r1, and the bytes to copy in r2.
  62. * It returns the number of uncopiable bytes (hopefully zero) in r0.
  63. */
  64. ENTRY(__copy_to_user_inatomic)
  65. .type __copy_to_user_inatomic, @function
  66. FEEDBACK_REENTER(__copy_from_user_inatomic)
  67. { movei r29, IS_COPY_TO_USER; j memcpy_common }
  68. .size __copy_to_user_inatomic, . - __copy_to_user_inatomic
  69. ENTRY(memcpy)
  70. .type memcpy, @function
  71. FEEDBACK_REENTER(__copy_from_user_inatomic)
  72. { movei r29, IS_MEMCPY }
  73. .size memcpy, . - memcpy
  74. /* Fall through */
  75. .type memcpy_common, @function
  76. memcpy_common:
  77. /* On entry, r29 holds one of the IS_* macro values from above. */
  78. /* r0 is the dest, r1 is the source, r2 is the size. */
  79. /* Save aside original dest so we can return it at the end. */
  80. { sw sp, lr; move r23, r0; or r4, r0, r1 }
  81. /* Check for an empty size. */
  82. { bz r2, .Ldone; andi r4, r4, 3 }
  83. /* Save aside original values in case of a fault. */
  84. { move r24, r1; move r25, r2 }
  85. move r27, lr
  86. /* Check for an unaligned source or dest. */
  87. { bnz r4, .Lcopy_unaligned_maybe_many; addli r4, r2, -256 }
  88. .Lcheck_aligned_copy_size:
  89. /* If we are copying < 256 bytes, branch to simple case. */
  90. { blzt r4, .Lcopy_8_check; slti_u r8, r2, 8 }
  91. /* Copying >= 256 bytes, so jump to complex prefetching loop. */
  92. { andi r6, r1, 63; j .Lcopy_many }
  93. /*
  94. *
  95. * Aligned 4 byte at a time copy loop
  96. *
  97. */
  98. .Lcopy_8_loop:
  99. /* Copy two words at a time to hide load latency. */
  100. EX: { lw r3, r1; addi r1, r1, 4; slti_u r8, r2, 16 }
  101. EX: { lw r4, r1; addi r1, r1, 4 }
  102. EX: { sw r0, r3; addi r0, r0, 4; addi r2, r2, -4 }
  103. EX: { sw r0, r4; addi r0, r0, 4; addi r2, r2, -4 }
  104. .Lcopy_8_check:
  105. { bzt r8, .Lcopy_8_loop; slti_u r4, r2, 4 }
  106. /* Copy odd leftover word, if any. */
  107. { bnzt r4, .Lcheck_odd_stragglers }
  108. EX: { lw r3, r1; addi r1, r1, 4 }
  109. EX: { sw r0, r3; addi r0, r0, 4; addi r2, r2, -4 }
  110. .Lcheck_odd_stragglers:
  111. { bnz r2, .Lcopy_unaligned_few }
  112. .Ldone:
  113. /* For memcpy return original dest address, else zero. */
  114. { mz r0, r29, r23; jrp lr }
  115. /*
  116. *
  117. * Prefetching multiple cache line copy handler (for large transfers).
  118. *
  119. */
  120. /* Copy words until r1 is cache-line-aligned. */
  121. .Lalign_loop:
  122. EX: { lw r3, r1; addi r1, r1, 4 }
  123. { andi r6, r1, 63 }
  124. EX: { sw r0, r3; addi r0, r0, 4; addi r2, r2, -4 }
  125. .Lcopy_many:
  126. { bnzt r6, .Lalign_loop; addi r9, r0, 63 }
  127. { addi r3, r1, 60; andi r9, r9, -64 }
  128. #if CHIP_HAS_WH64()
  129. /* No need to prefetch dst, we'll just do the wh64
  130. * right before we copy a line.
  131. */
  132. #endif
  133. EX: { lw r5, r3; addi r3, r3, 64; movei r4, 1 }
  134. /* Intentionally stall for a few cycles to leave L2 cache alone. */
  135. { bnzt zero, .; move r27, lr }
  136. EX: { lw r6, r3; addi r3, r3, 64 }
  137. /* Intentionally stall for a few cycles to leave L2 cache alone. */
  138. { bnzt zero, . }
  139. EX: { lw r7, r3; addi r3, r3, 64 }
  140. #if !CHIP_HAS_WH64()
  141. /* Prefetch the dest */
  142. /* Intentionally stall for a few cycles to leave L2 cache alone. */
  143. { bnzt zero, . }
  144. /* Use a real load to cause a TLB miss if necessary. We aren't using
  145. * r28, so this should be fine.
  146. */
  147. EX: { lw r28, r9; addi r9, r9, 64 }
  148. /* Intentionally stall for a few cycles to leave L2 cache alone. */
  149. { bnzt zero, . }
  150. { prefetch r9; addi r9, r9, 64 }
  151. /* Intentionally stall for a few cycles to leave L2 cache alone. */
  152. { bnzt zero, . }
  153. { prefetch r9; addi r9, r9, 64 }
  154. #endif
  155. /* Intentionally stall for a few cycles to leave L2 cache alone. */
  156. { bz zero, .Lbig_loop2 }
  157. /* On entry to this loop:
  158. * - r0 points to the start of dst line 0
  159. * - r1 points to start of src line 0
  160. * - r2 >= (256 - 60), only the first time the loop trips.
  161. * - r3 contains r1 + 128 + 60 [pointer to end of source line 2]
  162. * This is our prefetch address. When we get near the end
  163. * rather than prefetching off the end this is changed to point
  164. * to some "safe" recently loaded address.
  165. * - r5 contains *(r1 + 60) [i.e. last word of source line 0]
  166. * - r6 contains *(r1 + 64 + 60) [i.e. last word of source line 1]
  167. * - r9 contains ((r0 + 63) & -64)
  168. * [start of next dst cache line.]
  169. */
  170. .Lbig_loop:
  171. { jal .Lcopy_line2; add r15, r1, r2 }
  172. .Lbig_loop2:
  173. /* Copy line 0, first stalling until r5 is ready. */
  174. EX: { move r12, r5; lw r16, r1 }
  175. { bz r4, .Lcopy_8_check; slti_u r8, r2, 8 }
  176. /* Prefetch several lines ahead. */
  177. EX: { lw r5, r3; addi r3, r3, 64 }
  178. { jal .Lcopy_line }
  179. /* Copy line 1, first stalling until r6 is ready. */
  180. EX: { move r12, r6; lw r16, r1 }
  181. { bz r4, .Lcopy_8_check; slti_u r8, r2, 8 }
  182. /* Prefetch several lines ahead. */
  183. EX: { lw r6, r3; addi r3, r3, 64 }
  184. { jal .Lcopy_line }
  185. /* Copy line 2, first stalling until r7 is ready. */
  186. EX: { move r12, r7; lw r16, r1 }
  187. { bz r4, .Lcopy_8_check; slti_u r8, r2, 8 }
  188. /* Prefetch several lines ahead. */
  189. EX: { lw r7, r3; addi r3, r3, 64 }
  190. /* Use up a caches-busy cycle by jumping back to the top of the
  191. * loop. Might as well get it out of the way now.
  192. */
  193. { j .Lbig_loop }
  194. /* On entry:
  195. * - r0 points to the destination line.
  196. * - r1 points to the source line.
  197. * - r3 is the next prefetch address.
  198. * - r9 holds the last address used for wh64.
  199. * - r12 = WORD_15
  200. * - r16 = WORD_0.
  201. * - r17 == r1 + 16.
  202. * - r27 holds saved lr to restore.
  203. *
  204. * On exit:
  205. * - r0 is incremented by 64.
  206. * - r1 is incremented by 64, unless that would point to a word
  207. * beyond the end of the source array, in which case it is redirected
  208. * to point to an arbitrary word already in the cache.
  209. * - r2 is decremented by 64.
  210. * - r3 is unchanged, unless it points to a word beyond the
  211. * end of the source array, in which case it is redirected
  212. * to point to an arbitrary word already in the cache.
  213. * Redirecting is OK since if we are that close to the end
  214. * of the array we will not come back to this subroutine
  215. * and use the contents of the prefetched address.
  216. * - r4 is nonzero iff r2 >= 64.
  217. * - r9 is incremented by 64, unless it points beyond the
  218. * end of the last full destination cache line, in which
  219. * case it is redirected to a "safe address" that can be
  220. * clobbered (sp - 64)
  221. * - lr contains the value in r27.
  222. */
  223. /* r26 unused */
  224. .Lcopy_line:
  225. /* TODO: when r3 goes past the end, we would like to redirect it
  226. * to prefetch the last partial cache line (if any) just once, for the
  227. * benefit of the final cleanup loop. But we don't want to
  228. * prefetch that line more than once, or subsequent prefetches
  229. * will go into the RTF. But then .Lbig_loop should unconditionally
  230. * branch to top of loop to execute final prefetch, and its
  231. * nop should become a conditional branch.
  232. */
  233. /* We need two non-memory cycles here to cover the resources
  234. * used by the loads initiated by the caller.
  235. */
  236. { add r15, r1, r2 }
  237. .Lcopy_line2:
  238. { slt_u r13, r3, r15; addi r17, r1, 16 }
  239. /* NOTE: this will stall for one cycle as L1 is busy. */
  240. /* Fill second L1D line. */
  241. EX: { lw r17, r17; addi r1, r1, 48; mvz r3, r13, r1 } /* r17 = WORD_4 */
  242. #if CHIP_HAS_WH64()
  243. /* Prepare destination line for writing. */
  244. EX: { wh64 r9; addi r9, r9, 64 }
  245. #else
  246. /* Prefetch dest line */
  247. { prefetch r9; addi r9, r9, 64 }
  248. #endif
  249. /* Load seven words that are L1D hits to cover wh64 L2 usage. */
  250. /* Load the three remaining words from the last L1D line, which
  251. * we know has already filled the L1D.
  252. */
  253. EX: { lw r4, r1; addi r1, r1, 4; addi r20, r1, 16 } /* r4 = WORD_12 */
  254. EX: { lw r8, r1; addi r1, r1, 4; slt_u r13, r20, r15 }/* r8 = WORD_13 */
  255. EX: { lw r11, r1; addi r1, r1, -52; mvz r20, r13, r1 } /* r11 = WORD_14 */
  256. /* Load the three remaining words from the first L1D line, first
  257. * stalling until it has filled by "looking at" r16.
  258. */
  259. EX: { lw r13, r1; addi r1, r1, 4; move zero, r16 } /* r13 = WORD_1 */
  260. EX: { lw r14, r1; addi r1, r1, 4 } /* r14 = WORD_2 */
  261. EX: { lw r15, r1; addi r1, r1, 8; addi r10, r0, 60 } /* r15 = WORD_3 */
  262. /* Load second word from the second L1D line, first
  263. * stalling until it has filled by "looking at" r17.
  264. */
  265. EX: { lw r19, r1; addi r1, r1, 4; move zero, r17 } /* r19 = WORD_5 */
  266. /* Store last word to the destination line, potentially dirtying it
  267. * for the first time, which keeps the L2 busy for two cycles.
  268. */
  269. EX: { sw r10, r12 } /* store(WORD_15) */
  270. /* Use two L1D hits to cover the sw L2 access above. */
  271. EX: { lw r10, r1; addi r1, r1, 4 } /* r10 = WORD_6 */
  272. EX: { lw r12, r1; addi r1, r1, 4 } /* r12 = WORD_7 */
  273. /* Fill third L1D line. */
  274. EX: { lw r18, r1; addi r1, r1, 4 } /* r18 = WORD_8 */
  275. /* Store first L1D line. */
  276. EX: { sw r0, r16; addi r0, r0, 4; add r16, r0, r2 } /* store(WORD_0) */
  277. EX: { sw r0, r13; addi r0, r0, 4; andi r16, r16, -64 } /* store(WORD_1) */
  278. EX: { sw r0, r14; addi r0, r0, 4; slt_u r16, r9, r16 } /* store(WORD_2) */
  279. #if CHIP_HAS_WH64()
  280. EX: { sw r0, r15; addi r0, r0, 4; addi r13, sp, -64 } /* store(WORD_3) */
  281. #else
  282. /* Back up the r9 to a cache line we are already storing to
  283. * if it gets past the end of the dest vector. Strictly speaking,
  284. * we don't need to back up to the start of a cache line, but it's free
  285. * and tidy, so why not?
  286. */
  287. EX: { sw r0, r15; addi r0, r0, 4; andi r13, r0, -64 } /* store(WORD_3) */
  288. #endif
  289. /* Store second L1D line. */
  290. EX: { sw r0, r17; addi r0, r0, 4; mvz r9, r16, r13 }/* store(WORD_4) */
  291. EX: { sw r0, r19; addi r0, r0, 4 } /* store(WORD_5) */
  292. EX: { sw r0, r10; addi r0, r0, 4 } /* store(WORD_6) */
  293. EX: { sw r0, r12; addi r0, r0, 4 } /* store(WORD_7) */
  294. EX: { lw r13, r1; addi r1, r1, 4; move zero, r18 } /* r13 = WORD_9 */
  295. EX: { lw r14, r1; addi r1, r1, 4 } /* r14 = WORD_10 */
  296. EX: { lw r15, r1; move r1, r20 } /* r15 = WORD_11 */
  297. /* Store third L1D line. */
  298. EX: { sw r0, r18; addi r0, r0, 4 } /* store(WORD_8) */
  299. EX: { sw r0, r13; addi r0, r0, 4 } /* store(WORD_9) */
  300. EX: { sw r0, r14; addi r0, r0, 4 } /* store(WORD_10) */
  301. EX: { sw r0, r15; addi r0, r0, 4 } /* store(WORD_11) */
  302. /* Store rest of fourth L1D line. */
  303. EX: { sw r0, r4; addi r0, r0, 4 } /* store(WORD_12) */
  304. {
  305. EX: sw r0, r8 /* store(WORD_13) */
  306. addi r0, r0, 4
  307. /* Will r2 be > 64 after we subtract 64 below? */
  308. shri r4, r2, 7
  309. }
  310. {
  311. EX: sw r0, r11 /* store(WORD_14) */
  312. addi r0, r0, 8
  313. /* Record 64 bytes successfully copied. */
  314. addi r2, r2, -64
  315. }
  316. { jrp lr; move lr, r27 }
  317. /* Convey to the backtrace library that the stack frame is size
  318. * zero, and the real return address is on the stack rather than
  319. * in 'lr'.
  320. */
  321. { info 8 }
  322. .align 64
  323. .Lcopy_unaligned_maybe_many:
  324. /* Skip the setup overhead if we aren't copying many bytes. */
  325. { slti_u r8, r2, 20; sub r4, zero, r0 }
  326. { bnzt r8, .Lcopy_unaligned_few; andi r4, r4, 3 }
  327. { bz r4, .Ldest_is_word_aligned; add r18, r1, r2 }
  328. /*
  329. *
  330. * unaligned 4 byte at a time copy handler.
  331. *
  332. */
  333. /* Copy single bytes until r0 == 0 mod 4, so we can store words. */
  334. .Lalign_dest_loop:
  335. EX: { lb_u r3, r1; addi r1, r1, 1; addi r4, r4, -1 }
  336. EX: { sb r0, r3; addi r0, r0, 1; addi r2, r2, -1 }
  337. { bnzt r4, .Lalign_dest_loop; andi r3, r1, 3 }
  338. /* If source and dest are now *both* aligned, do an aligned copy. */
  339. { bz r3, .Lcheck_aligned_copy_size; addli r4, r2, -256 }
  340. .Ldest_is_word_aligned:
  341. #if CHIP_HAS_DWORD_ALIGN()
  342. EX: { andi r8, r0, 63; lwadd_na r6, r1, 4}
  343. { slti_u r9, r2, 64; bz r8, .Ldest_is_L2_line_aligned }
  344. /* This copies unaligned words until either there are fewer
  345. * than 4 bytes left to copy, or until the destination pointer
  346. * is cache-aligned, whichever comes first.
  347. *
  348. * On entry:
  349. * - r0 is the next store address.
  350. * - r1 points 4 bytes past the load address corresponding to r0.
  351. * - r2 >= 4
  352. * - r6 is the next aligned word loaded.
  353. */
  354. .Lcopy_unaligned_src_words:
  355. EX: { lwadd_na r7, r1, 4; slti_u r8, r2, 4 + 4 }
  356. /* stall */
  357. { dword_align r6, r7, r1; slti_u r9, r2, 64 + 4 }
  358. EX: { swadd r0, r6, 4; addi r2, r2, -4 }
  359. { bnz r8, .Lcleanup_unaligned_words; andi r8, r0, 63 }
  360. { bnzt r8, .Lcopy_unaligned_src_words; move r6, r7 }
  361. /* On entry:
  362. * - r0 is the next store address.
  363. * - r1 points 4 bytes past the load address corresponding to r0.
  364. * - r2 >= 4 (# of bytes left to store).
  365. * - r6 is the next aligned src word value.
  366. * - r9 = (r2 < 64U).
  367. * - r18 points one byte past the end of source memory.
  368. */
  369. .Ldest_is_L2_line_aligned:
  370. {
  371. /* Not a full cache line remains. */
  372. bnz r9, .Lcleanup_unaligned_words
  373. move r7, r6
  374. }
  375. /* r2 >= 64 */
  376. /* Kick off two prefetches, but don't go past the end. */
  377. { addi r3, r1, 63 - 4; addi r8, r1, 64 + 63 - 4 }
  378. { prefetch r3; move r3, r8; slt_u r8, r8, r18 }
  379. { mvz r3, r8, r1; addi r8, r3, 64 }
  380. { prefetch r3; move r3, r8; slt_u r8, r8, r18 }
  381. { mvz r3, r8, r1; movei r17, 0 }
  382. .Lcopy_unaligned_line:
  383. /* Prefetch another line. */
  384. { prefetch r3; addi r15, r1, 60; addi r3, r3, 64 }
  385. /* Fire off a load of the last word we are about to copy. */
  386. EX: { lw_na r15, r15; slt_u r8, r3, r18 }
  387. EX: { mvz r3, r8, r1; wh64 r0 }
  388. /* This loop runs twice.
  389. *
  390. * On entry:
  391. * - r17 is even before the first iteration, and odd before
  392. * the second. It is incremented inside the loop. Encountering
  393. * an even value at the end of the loop makes it stop.
  394. */
  395. .Lcopy_half_an_unaligned_line:
  396. EX: {
  397. /* Stall until the last byte is ready. In the steady state this
  398. * guarantees all words to load below will be in the L2 cache, which
  399. * avoids shunting the loads to the RTF.
  400. */
  401. move zero, r15
  402. lwadd_na r7, r1, 16
  403. }
  404. EX: { lwadd_na r11, r1, 12 }
  405. EX: { lwadd_na r14, r1, -24 }
  406. EX: { lwadd_na r8, r1, 4 }
  407. EX: { lwadd_na r9, r1, 4 }
  408. EX: {
  409. lwadd_na r10, r1, 8
  410. /* r16 = (r2 < 64), after we subtract 32 from r2 below. */
  411. slti_u r16, r2, 64 + 32
  412. }
  413. EX: { lwadd_na r12, r1, 4; addi r17, r17, 1 }
  414. EX: { lwadd_na r13, r1, 8; dword_align r6, r7, r1 }
  415. EX: { swadd r0, r6, 4; dword_align r7, r8, r1 }
  416. EX: { swadd r0, r7, 4; dword_align r8, r9, r1 }
  417. EX: { swadd r0, r8, 4; dword_align r9, r10, r1 }
  418. EX: { swadd r0, r9, 4; dword_align r10, r11, r1 }
  419. EX: { swadd r0, r10, 4; dword_align r11, r12, r1 }
  420. EX: { swadd r0, r11, 4; dword_align r12, r13, r1 }
  421. EX: { swadd r0, r12, 4; dword_align r13, r14, r1 }
  422. EX: { swadd r0, r13, 4; addi r2, r2, -32 }
  423. { move r6, r14; bbst r17, .Lcopy_half_an_unaligned_line }
  424. { bzt r16, .Lcopy_unaligned_line; move r7, r6 }
  425. /* On entry:
  426. * - r0 is the next store address.
  427. * - r1 points 4 bytes past the load address corresponding to r0.
  428. * - r2 >= 0 (# of bytes left to store).
  429. * - r7 is the next aligned src word value.
  430. */
  431. .Lcleanup_unaligned_words:
  432. /* Handle any trailing bytes. */
  433. { bz r2, .Lcopy_unaligned_done; slti_u r8, r2, 4 }
  434. { bzt r8, .Lcopy_unaligned_src_words; move r6, r7 }
  435. /* Move r1 back to the point where it corresponds to r0. */
  436. { addi r1, r1, -4 }
  437. #else /* !CHIP_HAS_DWORD_ALIGN() */
  438. /* Compute right/left shift counts and load initial source words. */
  439. { andi r5, r1, -4; andi r3, r1, 3 }
  440. EX: { lw r6, r5; addi r5, r5, 4; shli r3, r3, 3 }
  441. EX: { lw r7, r5; addi r5, r5, 4; sub r4, zero, r3 }
  442. /* Load and store one word at a time, using shifts and ORs
  443. * to correct for the misaligned src.
  444. */
  445. .Lcopy_unaligned_src_loop:
  446. { shr r6, r6, r3; shl r8, r7, r4 }
  447. EX: { lw r7, r5; or r8, r8, r6; move r6, r7 }
  448. EX: { sw r0, r8; addi r0, r0, 4; addi r2, r2, -4 }
  449. { addi r5, r5, 4; slti_u r8, r2, 8 }
  450. { bzt r8, .Lcopy_unaligned_src_loop; addi r1, r1, 4 }
  451. { bz r2, .Lcopy_unaligned_done }
  452. #endif /* !CHIP_HAS_DWORD_ALIGN() */
  453. /* Fall through */
  454. /*
  455. *
  456. * 1 byte at a time copy handler.
  457. *
  458. */
  459. .Lcopy_unaligned_few:
  460. EX: { lb_u r3, r1; addi r1, r1, 1 }
  461. EX: { sb r0, r3; addi r0, r0, 1; addi r2, r2, -1 }
  462. { bnzt r2, .Lcopy_unaligned_few }
  463. .Lcopy_unaligned_done:
  464. /* For memcpy return original dest address, else zero. */
  465. { mz r0, r29, r23; jrp lr }
  466. .Lend_memcpy_common:
  467. .size memcpy_common, .Lend_memcpy_common - memcpy_common
  468. .section .fixup,"ax"
  469. memcpy_common_fixup:
  470. .type memcpy_common_fixup, @function
  471. /* Skip any bytes we already successfully copied.
  472. * r2 (num remaining) is correct, but r0 (dst) and r1 (src)
  473. * may not be quite right because of unrolling and prefetching.
  474. * So we need to recompute their values as the address just
  475. * after the last byte we are sure was successfully loaded and
  476. * then stored.
  477. */
  478. /* Determine how many bytes we successfully copied. */
  479. { sub r3, r25, r2 }
  480. /* Add this to the original r0 and r1 to get their new values. */
  481. { add r0, r23, r3; add r1, r24, r3 }
  482. { bzt r29, memcpy_fixup_loop }
  483. { blzt r29, copy_to_user_fixup_loop }
  484. copy_from_user_fixup_loop:
  485. /* Try copying the rest one byte at a time, expecting a load fault. */
  486. .Lcfu: { lb_u r3, r1; addi r1, r1, 1 }
  487. { sb r0, r3; addi r0, r0, 1; addi r2, r2, -1 }
  488. { bnzt r2, copy_from_user_fixup_loop }
  489. .Lcopy_from_user_fixup_zero_remainder:
  490. { bbs r29, 2f } /* low bit set means IS_COPY_FROM_USER */
  491. /* byte-at-a-time loop faulted, so zero the rest. */
  492. { move r3, r2; bz r2, 2f /* should be impossible, but handle it. */ }
  493. 1: { sb r0, zero; addi r0, r0, 1; addi r3, r3, -1 }
  494. { bnzt r3, 1b }
  495. 2: move lr, r27
  496. { move r0, r2; jrp lr }
  497. copy_to_user_fixup_loop:
  498. /* Try copying the rest one byte at a time, expecting a store fault. */
  499. { lb_u r3, r1; addi r1, r1, 1 }
  500. .Lctu: { sb r0, r3; addi r0, r0, 1; addi r2, r2, -1 }
  501. { bnzt r2, copy_to_user_fixup_loop }
  502. .Lcopy_to_user_fixup_done:
  503. move lr, r27
  504. { move r0, r2; jrp lr }
  505. memcpy_fixup_loop:
  506. /* Try copying the rest one byte at a time. We expect a disastrous
  507. * fault to happen since we are in fixup code, but let it happen.
  508. */
  509. { lb_u r3, r1; addi r1, r1, 1 }
  510. { sb r0, r3; addi r0, r0, 1; addi r2, r2, -1 }
  511. { bnzt r2, memcpy_fixup_loop }
  512. /* This should be unreachable, we should have faulted again.
  513. * But be paranoid and handle it in case some interrupt changed
  514. * the TLB or something.
  515. */
  516. move lr, r27
  517. { move r0, r23; jrp lr }
  518. .size memcpy_common_fixup, . - memcpy_common_fixup
  519. .section __ex_table,"a"
  520. .word .Lcfu, .Lcopy_from_user_fixup_zero_remainder
  521. .word .Lctu, .Lcopy_to_user_fixup_done