/* SPDX-License-Identifier: GPL-2.0-only */
/*
 * Copyright (c) 2010-2011, The Linux Foundation. All rights reserved.
 */

/*
 * Description
 *
 *   library function for memcpy where length bytes are copied from
 *   ptr_in to ptr_out. ptr_out is returned unchanged.
 *   Allows any combination of alignment on input and output pointers
 *   and length from 0 to 2^32-1
 *
 * Restrictions
 *   The arrays should not overlap, the program will produce undefined output
 *   if they do.
 *   For blocks less than 16 bytes a byte by byte copy is performed. For
 *   8byte alignments, and length multiples, a dword copy is performed up to
 *   96bytes
 * History
 *
 *   DJH  5/15/09 Initial version 1.0
 *   DJH  6/ 1/09 Version 1.1 modified ABI to inlcude R16-R19
 *   DJH  7/12/09 Version 1.2 optimized codesize down to 760 was 840
 *   DJH 10/14/09 Version 1.3 added special loop for aligned case, was
 *                            overreading bloated codesize back up to 892
 *   DJH  4/20/10 Version 1.4 fixed Ldword_loop_epilog loop to prevent loads
 *                            occurring if only 1 left outstanding, fixes bug
 *                            # 3888, corrected for all alignments. Peeled off
 *                            1 32byte chunk from kernel loop and extended 8byte
 *                            loop at end to solve all combinations and prevent
 *                            over read.  Fixed Ldword_loop_prolog to prevent
 *                            overread for blocks less than 48bytes. Reduced
 *                            codesize to 752 bytes
 *   DJH  4/21/10 version 1.5 1.4 fix broke code for input block ends not
 *                            aligned to dword boundaries,underwriting by 1
 *                            byte, added detection for this and fixed. A
 *                            little bloat.
 *   DJH  4/23/10 version 1.6 corrected stack error, R20 was not being restored
 *                            always, fixed the error of R20 being modified
 *                            before it was being saved
 * Natural c model
 * ===============
 * void * memcpy(char * ptr_out, char * ptr_in, int length) {
 *   int i;
 *   if(length) for(i=0; i < length; i++) { ptr_out[i] = ptr_in[i]; }
 *   return(ptr_out);
 * }
 *
 * Optimized memcpy function
 * =========================
 * void * memcpy(char * ptr_out, char * ptr_in, int len) {
 *   int i, prolog, kernel, epilog, mask;
 *   u8 offset;
 *   s64 data0, dataF8, data70;
 *
 *   s64 * ptr8_in;
 *   s64 * ptr8_out;
 *   s32 * ptr4;
 *   s16 * ptr2;
 *
 *   offset = ((int) ptr_in) & 7;
 *   ptr8_in = (s64 *) &ptr_in[-offset];   //read in the aligned pointers
 *
 *   data70 = *ptr8_in++;
 *   dataF8 = *ptr8_in++;
 *
 *   data0 = HEXAGON_P_valignb_PPp(dataF8, data70, offset);
 *
 *   prolog = 32 - ((int) ptr_out);
 *   mask  = 0x7fffffff >> HEXAGON_R_cl0_R(len);
 *   prolog = prolog & mask;
 *   kernel = len - prolog;
 *   epilog = kernel & 0x1F;
 *   kernel = kernel>>5;
 *
 *   if (prolog & 1) { ptr_out[0] = (u8) data0; data0 >>= 8; ptr_out += 1;}
 *   ptr2 = (s16 *) &ptr_out[0];
 *   if (prolog & 2) { ptr2[0] = (u16) data0;  data0 >>= 16; ptr_out += 2;}
 *   ptr4 = (s32 *) &ptr_out[0];
 *   if (prolog & 4) { ptr4[0] = (u32) data0;  data0 >>= 32; ptr_out += 4;}
 *
 *   offset = offset + (prolog & 7);
 *   if (offset >= 8) {
 *     data70 = dataF8;
 *     dataF8 = *ptr8_in++;
 *   }
 *   offset = offset & 0x7;
 *
 *   prolog = prolog >> 3;
 *   if (prolog) for (i=0; i < prolog; i++) {
 *       data0 = HEXAGON_P_valignb_PPp(dataF8, data70, offset);
 *       ptr8_out = (s64 *) &ptr_out[0]; *ptr8_out = data0; ptr_out += 8;
 *       data70 = dataF8;
 *       dataF8 = *ptr8_in++;
 *   }
 *   if(kernel) { kernel -= 1; epilog += 32; }
 *   if(kernel) for(i=0; i < kernel; i++) {
 *       data0 = HEXAGON_P_valignb_PPp(dataF8, data70, offset);
 *       ptr8_out = (s64 *) &ptr_out[0]; *ptr8_out = data0; ptr_out += 8;
 *       data70 = *ptr8_in++;
 *
 *       data0 = HEXAGON_P_valignb_PPp(data70, dataF8, offset);
 *       ptr8_out = (s64 *) &ptr_out[0]; *ptr8_out = data0; ptr_out += 8;
 *       dataF8 = *ptr8_in++;
 *
 *       data0 = HEXAGON_P_valignb_PPp(dataF8, data70, offset);
 *       ptr8_out = (s64 *) &ptr_out[0]; *ptr8_out = data0; ptr_out += 8;
 *       data70 = *ptr8_in++;
 *
 *       data0 = HEXAGON_P_valignb_PPp(data70, dataF8, offset);
 *       ptr8_out = (s64 *) &ptr_out[0]; *ptr8_out = data0; ptr_out += 8;
 *       dataF8 = *ptr8_in++;
 *   }
 *   epilogdws = epilog >> 3;
 *   if (epilogdws) for (i=0; i < epilogdws; i++) {
 *       data0 = HEXAGON_P_valignb_PPp(dataF8, data70, offset);
 *       ptr8_out = (s64 *) &ptr_out[0]; *ptr8_out = data0; ptr_out += 8;
 *       data70 = dataF8;
 *       dataF8 = *ptr8_in++;
 *   }
 *   data0 = HEXAGON_P_valignb_PPp(dataF8, data70, offset);
 *
 *   ptr4 = (s32 *) &ptr_out[0];
 *   if (epilog & 4) { ptr4[0] = (u32) data0; data0 >>= 32; ptr_out += 4;}
 *   ptr2 = (s16 *) &ptr_out[0];
 *   if (epilog & 2) { ptr2[0] = (u16) data0; data0 >>= 16; ptr_out += 2;}
 *   if (epilog & 1) { *ptr_out++ = (u8) data0; }
 *
 *   return(ptr_out - length);
 * }
 *
 * Codesize : 784 bytes
 */


#define ptr_out  R0 /*  destination  pounter  */
#define ptr_in  R1 /*  source pointer  */
#define len  R2 /*  length of copy in bytes  */

#define data70  R13:12 /*  lo 8 bytes of non-aligned transfer  */
#define dataF8  R11:10 /*  hi 8 bytes of non-aligned transfer  */
#define ldata0  R7:6 /*  even 8 bytes chunks  */
#define ldata1  R25:24 /*  odd 8 bytes chunks  */
#define data1  R7 /*  lower 8 bytes of ldata1  */
#define data0  R6 /*  lower 8 bytes of ldata0  */

#define ifbyte  p0 /*  if transfer has bytes in epilog/prolog  */
#define ifhword  p0 /*  if transfer has shorts in epilog/prolog  */
#define ifword  p0 /*  if transfer has words in epilog/prolog  */
#define noprolog p0 /*  no prolog, xfer starts at 32byte  */
#define nokernel p1 /*  no 32byte multiple block in the transfer  */
#define noepilog p0 /*  no epilog, xfer ends on 32byte boundary  */
#define align  p2 /*  alignment of input rel to 8byte boundary  */
#define kernel1  p0 /*  kernel count == 1  */

#define dalign  R25 /*  rel alignment of input to output data  */
#define star3  R16 /*  number bytes in prolog - dwords  */
#define rest  R8 /*  length - prolog bytes  */
#define back  R7 /*  nr bytes > dword boundary in src block  */
#define epilog  R3 /*  bytes in epilog  */
#define inc  R15:14 /*  inc kernel by -1 and defetch ptr by 32  */
#define kernel  R4 /*  number of 32byte chunks in kernel  */
#define ptr_in_p_128 R5 /*  pointer for prefetch of input data  */
#define mask  R8 /*  mask used to determine prolog size  */
#define shift  R8 /*  used to work a shifter to extract bytes  */
#define shift2  R5 /*  in epilog to workshifter to extract bytes */
#define prolog  R15 /*  bytes in  prolog  */
#define epilogdws R15 /*  number dwords in epilog  */
#define shiftb  R14 /*  used to extract bytes  */
#define offset  R9 /*  same as align in reg  */
#define ptr_out_p_32 R17 /*  pointer to output dczero  */
#define align888 R14 /*  if simple dword loop can be used  */
#define len8  R9 /*  number of dwords in length  */
#define over  R20 /*  nr of bytes > last inp buf dword boundary */

#define ptr_in_p_128kernel R5:4 /*  packed fetch pointer & kernel cnt */

 .section .text
 .p2align 4
        .global memcpy
        .type memcpy, @function
memcpy:
{
 p2 = cmp.eq(len, #0);  /*  =0 */
 align888 = or(ptr_in, ptr_out); /*  %8 < 97 */
 p0 = cmp.gtu(len, #23);  /*  %1, <24 */
 p1 = cmp.eq(ptr_in, ptr_out); /*  attempt to overwrite self */
}
{
 p1 = or(p2, p1);
 p3 = cmp.gtu(len, #95);  /*  %8 < 97 */
 align888 = or(align888, len); /*  %8 < 97 */
 len8 = lsr(len, #3);  /*  %8 < 97 */
}
{
 dcfetch(ptr_in);  /*  zero/ptrin=ptrout causes fetch */
 p2 = bitsclr(align888, #7); /*  %8 < 97  */
 if(p1) jumpr r31;  /*  =0  */
}
{
 p2 = and(p2,!p3); /*  %8 < 97  */
 if (p2.new) len = add(len, #-8); /*  %8 < 97  */
 if (p2.new) jump:NT .Ldwordaligned;  /*  %8 < 97  */
}
{
 if(!p0) jump .Lbytes23orless; /*  %1, <24  */
 mask.l = #LO(0x7fffffff);
 /*  all bytes before line multiples of data  */
 prolog = sub(#0, ptr_out);
}
{
 /*  save r31 on stack, decrement sp by 16  */
 allocframe(#24);
 mask.h = #HI(0x7fffffff);
 ptr_in_p_128 = add(ptr_in, #32);
 back = cl0(len);
}
{
 memd(sp+#0) = R17:16;  /*  save r16,r17 on stack6  */
 r31.l = #LO(.Lmemcpy_return); /*  set up final return pointer  */
 prolog &= lsr(mask, back);
 offset = and(ptr_in, #7);
}
{
 memd(sp+#8) = R25:24;  /*  save r25,r24 on stack  */
 dalign = sub(ptr_out, ptr_in);
 r31.h = #HI(.Lmemcpy_return); /*  set up final return pointer  */
}
{
 /*  see if there if input buffer end if aligned  */
 over = add(len, ptr_in);
 back = add(len, offset);
 memd(sp+#16) = R21:20;  /*  save r20,r21 on stack  */
}
{
 noprolog = bitsclr(prolog, #7);
 prolog = and(prolog, #31);
 dcfetch(ptr_in_p_128);
 ptr_in_p_128 = add(ptr_in_p_128, #32);
}
{
 kernel = sub(len, prolog);
 shift = asl(prolog, #3);
 star3 = and(prolog, #7);
 ptr_in = and(ptr_in, #-8);
}
{
 prolog = lsr(prolog, #3);
 epilog = and(kernel, #31);
 ptr_out_p_32 = add(ptr_out, prolog);
 over = and(over, #7);
}
{
 p3 = cmp.gtu(back, #8);
 kernel = lsr(kernel, #5);
 dcfetch(ptr_in_p_128);
 ptr_in_p_128 = add(ptr_in_p_128, #32);
}
{
 p1 = cmp.eq(prolog, #0);
 if(!p1.new) prolog = add(prolog, #1);
 dcfetch(ptr_in_p_128); /*  reserve the line 64bytes on  */
 ptr_in_p_128 = add(ptr_in_p_128, #32);
}
{
 nokernel = cmp.eq(kernel,#0);
 dcfetch(ptr_in_p_128); /* reserve the line 64bytes on  */
 ptr_in_p_128 = add(ptr_in_p_128, #32);
 shiftb = and(shift, #8);
}
{
 dcfetch(ptr_in_p_128);  /*  reserve the line 64bytes on  */
 ptr_in_p_128 = add(ptr_in_p_128, #32);
 if(nokernel) jump .Lskip64;
 p2 = cmp.eq(kernel, #1); /*  skip ovr if kernel == 0  */
}
{
 dczeroa(ptr_out_p_32);
 /*  don't advance pointer  */
 if(!p2) ptr_out_p_32 = add(ptr_out_p_32, #32);
}
{
 dalign = and(dalign, #31);
 dczeroa(ptr_out_p_32);
}
.Lskip64:
{
 data70 = memd(ptr_in++#16);
 if(p3) dataF8 = memd(ptr_in+#8);
 if(noprolog) jump .Lnoprolog32;
 align = offset;
}
/*  upto initial 7 bytes  */
{
 ldata0 = valignb(dataF8, data70, align);
 ifbyte = tstbit(shift,#3);
 offset = add(offset, star3);
}
{
 if(ifbyte) memb(ptr_out++#1) = data0;
 ldata0 = lsr(ldata0, shiftb);
 shiftb = and(shift, #16);
 ifhword = tstbit(shift,#4);
}
{
 if(ifhword) memh(ptr_out++#2) = data0;
 ldata0 = lsr(ldata0, shiftb);
 ifword = tstbit(shift,#5);
 p2 = cmp.gtu(offset, #7);
}
{
 if(ifword) memw(ptr_out++#4) = data0;
 if(p2) data70 = dataF8;
 if(p2) dataF8 = memd(ptr_in++#8); /*  another 8 bytes  */
 align = offset;
}
.Lnoprolog32:
{
 p3 = sp1loop0(.Ldword_loop_prolog, prolog)
 rest = sub(len, star3); /*  whats left after the loop  */
 p0 = cmp.gt(over, #0);
}
 if(p0) rest = add(rest, #16);
.Ldword_loop_prolog:
{
 if(p3) memd(ptr_out++#8) = ldata0;
 ldata0 = valignb(dataF8, data70, align);
 p0 = cmp.gt(rest, #16);
}
{
 data70 = dataF8;
 if(p0) dataF8 = memd(ptr_in++#8);
 rest = add(rest, #-8);
}:endloop0
.Lkernel:
{
 /*  kernel is at least 32bytes  */
 p3 = cmp.gtu(kernel, #0);
 /*  last itn. remove edge effects  */
 if(p3.new) kernel = add(kernel, #-1);
 /*  dealt with in last dword loop  */
 if(p3.new) epilog = add(epilog, #32);
}
{
 nokernel = cmp.eq(kernel, #0);  /*  after adjustment, recheck */
 if(nokernel.new) jump:NT .Lepilog; /*  likely not taken  */
 inc = combine(#32, #-1);
 p3 = cmp.gtu(dalign, #24);
}
{
 if(p3) jump .Lodd_alignment;
}
{
 loop0(.Loword_loop_25to31, kernel);
 kernel1 = cmp.gtu(kernel, #1);
 rest = kernel;
}
 .falign
.Loword_loop_25to31:
{
 dcfetch(ptr_in_p_128); /*  prefetch 4 lines ahead  */
 if(kernel1) ptr_out_p_32 = add(ptr_out_p_32, #32);
}
{
 dczeroa(ptr_out_p_32); /*  reserve the next 32bytes in cache  */
 p3 = cmp.eq(kernel, rest);
}
{
 /*  kernel -= 1  */
 ptr_in_p_128kernel = vaddw(ptr_in_p_128kernel, inc);
 /*  kill write on first iteration  */
 if(!p3) memd(ptr_out++#8) = ldata1;
 ldata1 = valignb(dataF8, data70, align);
 data70 = memd(ptr_in++#8);
}
{
 memd(ptr_out++#8) = ldata0;
 ldata0 = valignb(data70, dataF8, align);
 dataF8 = memd(ptr_in++#8);
}
{
 memd(ptr_out++#8) = ldata1;
 ldata1 = valignb(dataF8, data70, align);
 data70 = memd(ptr_in++#8);
}
{
 memd(ptr_out++#8) = ldata0;
 ldata0 = valignb(data70, dataF8, align);
 dataF8 = memd(ptr_in++#8);
 kernel1 = cmp.gtu(kernel, #1);
}:endloop0
{
 memd(ptr_out++#8) = ldata1;
 jump .Lepilog;
}
.Lodd_alignment:
{
 loop0(.Loword_loop_00to24, kernel);
 kernel1 = cmp.gtu(kernel, #1);
 rest = add(kernel, #-1);
}
 .falign
.Loword_loop_00to24:
{
 dcfetch(ptr_in_p_128); /*  prefetch 4 lines ahead  */
 ptr_in_p_128kernel = vaddw(ptr_in_p_128kernel, inc);
 if(kernel1) ptr_out_p_32 = add(ptr_out_p_32, #32);
}
{
 dczeroa(ptr_out_p_32); /*  reserve the next 32bytes in cache  */
}
{
 memd(ptr_out++#8) = ldata0;
 ldata0 = valignb(dataF8, data70, align);
 data70 = memd(ptr_in++#8);
}
{
 memd(ptr_out++#8) = ldata0;
 ldata0 = valignb(data70, dataF8, align);
 dataF8 = memd(ptr_in++#8);
}
{
 memd(ptr_out++#8) = ldata0;
 ldata0 = valignb(dataF8, data70, align);
 data70 = memd(ptr_in++#8);
}
{
 memd(ptr_out++#8) = ldata0;
 ldata0 = valignb(data70, dataF8, align);
 dataF8 = memd(ptr_in++#8);
 kernel1 = cmp.gtu(kernel, #1);
}:endloop0
.Lepilog:
{
 noepilog = cmp.eq(epilog,#0);
 epilogdws = lsr(epilog, #3);
 kernel = and(epilog, #7);
}
{
 if(noepilog) jumpr r31;
 if(noepilog) ptr_out = sub(ptr_out, len);
 p3 = cmp.eq(epilogdws, #0);
 shift2 = asl(epilog, #3);
}
{
 shiftb = and(shift2, #32);
 ifword = tstbit(epilog,#2);
 if(p3) jump .Lepilog60;
 if(!p3) epilog = add(epilog, #-16);
}
{
 loop0(.Ldword_loop_epilog, epilogdws);
 /*  stop criteria is lsbs unless = 0 then its 8  */
 p3 = cmp.eq(kernel, #0);
 if(p3.new) kernel= #8;
 p1 = cmp.gt(over, #0);
}
 /*  if not aligned to end of buffer execute 1 more iteration  */
 if(p1) kernel= #0;
.Ldword_loop_epilog:
{
 memd(ptr_out++#8) = ldata0;
 ldata0 = valignb(dataF8, data70, align);
 p3 = cmp.gt(epilog, kernel);
}
{
 data70 = dataF8;
 if(p3) dataF8 = memd(ptr_in++#8);
 epilog = add(epilog, #-8);
}:endloop0
/* copy last 7 bytes */
.Lepilog60:
{
 if(ifword) memw(ptr_out++#4) = data0;
 ldata0 = lsr(ldata0, shiftb);
 ifhword = tstbit(epilog,#1);
 shiftb = and(shift2, #16);
}
{
 if(ifhword) memh(ptr_out++#2) = data0;
 ldata0 = lsr(ldata0, shiftb);
 ifbyte = tstbit(epilog,#0);
 if(ifbyte.new) len = add(len, #-1);
}
{
 if(ifbyte) memb(ptr_out) = data0;
 ptr_out = sub(ptr_out, len); /*  return dest pointer  */
        jumpr r31;
}
/*  do byte copy for small n  */
.Lbytes23orless:
{
 p3 = sp1loop0(.Lbyte_copy, len);
 len = add(len, #-1);
}
.Lbyte_copy:
{
 data0 = memb(ptr_in++#1);
 if(p3) memb(ptr_out++#1) = data0;
}:endloop0
{
 memb(ptr_out) = data0;
 ptr_out = sub(ptr_out, len);
 jumpr r31;
}
/*  do dword copies for aligned in, out and length  */
.Ldwordaligned:
{
 p3 = sp1loop0(.Ldword_copy, len8);
}
.Ldword_copy:
{
 if(p3) memd(ptr_out++#8) = ldata0;
 ldata0 = memd(ptr_in++#8);
}:endloop0
{
 memd(ptr_out) = ldata0;
 ptr_out = sub(ptr_out, len);
 jumpr r31; /*  return to function caller  */
}
.Lmemcpy_return:
 r21:20 = memd(sp+#16); /*  restore r20+r21  */
{
 r25:24 = memd(sp+#8); /*  restore r24+r25  */
 r17:16 = memd(sp+#0); /*  restore r16+r17  */
}
 deallocframe; /*  restore r31 and incrment stack by 16  */
 jumpr r31