round.h 11.9 KB
Newer Older
Martin Schläffer committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229
#ifndef ROUND_H_
#define ROUND_H_

#include "ascon.h"
#include "printstate.h"

forceinline void ROUND_LOOP(state_t* s, const uint8_t* C, const uint8_t* E) {
  uint32_t tmp0, tmp1;
  __asm__ __volatile__(
      "rbegin_%=:;\n\t"
      "ldrb %[tmp1], [%[C]], #1\n\t"
      "eor %[x0_l], %[x0_l], %[x4_l]\n\t"
      "eor %[x4_l], %[x4_l], %[x3_l]\n\t"
      "eor %[x2_l], %[x2_l], %[x1_l]\n\t"
      "mvn %[tmp0], %[x0_l]\n\t"
      "orr %[tmp0], %[x4_l], %[tmp0]\n\t"
      "eor %[x2_l], %[x2_l], %[tmp1]\n\t"
      "bic %[tmp1], %[x2_l], %[x1_l]\n\t"
      "eor %[x0_l], %[x0_l], %[tmp1]\n\t"
      "mvn %[tmp1], %[x4_l]\n\t"
      "orr %[tmp1], %[x3_l], %[tmp1]\n\t"
      "eor %[x2_l], %[x2_l], %[tmp1]\n\t"
      "bic %[tmp1], %[x1_l], %[x0_l]\n\t"
      "eor %[x4_l], %[x4_l], %[tmp1]\n\t"
      "and %[tmp1], %[x3_l], %[x2_l]\n\t"
      "eor %[x1_l], %[x1_l], %[tmp1]\n\t"
      "eor %[x3_l], %[x3_l], %[tmp0]\n\t"
      "ldrb %[tmp1], [%[C]], #1\n\t"
      "eor %[x1_l], %[x1_l], %[x0_l]\n\t"
      "eor %[x3_l], %[x3_l], %[x2_l]\n\t"
      "eor %[x0_l], %[x0_l], %[x4_l]\n\t"
      "eor %[x0_h], %[x0_h], %[x4_h]\n\t"
      "eor %[x4_h], %[x4_h], %[x3_h]\n\t"
      "eor %[x2_h], %[x2_h], %[tmp1]\n\t"
      "eor %[x2_h], %[x2_h], %[x1_h]\n\t"
      "mvn %[tmp0], %[x0_h]\n\t"
      "orr %[tmp0], %[x4_h], %[tmp0]\n\t"
      "bic %[tmp1], %[x2_h], %[x1_h]\n\t"
      "eor %[x0_h], %[x0_h], %[tmp1]\n\t"
      "mvn %[tmp1], %[x4_h]\n\t"
      "orr %[tmp1], %[x3_h], %[tmp1]\n\t"
      "eor %[x2_h], %[x2_h], %[tmp1]\n\t"
      "bic %[tmp1], %[x1_h], %[x0_h]\n\t"
      "eor %[x4_h], %[x4_h], %[tmp1]\n\t"
      "and %[tmp1], %[x3_h], %[x2_h]\n\t"
      "eor %[x1_h], %[x1_h], %[tmp1]\n\t"
      "eor %[x3_h], %[x3_h], %[tmp0]\n\t"
      "eor %[x1_h], %[x1_h], %[x0_h]\n\t"
      "eor %[x3_h], %[x3_h], %[x2_h]\n\t"
      "eor %[x0_h], %[x0_h], %[x4_h]\n\t"
      "eor %[tmp0], %[x0_l], %[x0_h], ror #4\n\t"
      "eor %[tmp1], %[x0_h], %[x0_l], ror #5\n\t"
      "eor %[x0_h], %[x0_h], %[tmp0], ror #10\n\t"
      "eor %[x0_l], %[x0_l], %[tmp1], ror #9\n\t"
      "eor %[tmp0], %[x1_l], %[x1_l], ror #11\n\t"
      "eor %[tmp1], %[x1_h], %[x1_h], ror #11\n\t"
      "eor %[x1_h], %[x1_h], %[tmp0], ror #20\n\t"
      "eor %[x1_l], %[x1_l], %[tmp1], ror #19\n\t"
      "eor %[tmp0], %[x2_l], %[x2_h], ror #2\n\t"
      "eor %[tmp1], %[x2_h], %[x2_l], ror #3\n\t"
      "eor %[x2_h], %[x2_h], %[tmp0], ror #1\n\t"
      "eor %[x2_l], %[x2_l], %[tmp1]\n\t"
      "eor %[tmp0], %[x3_l], %[x3_h], ror #3\n\t"
      "eor %[tmp1], %[x3_h], %[x3_l], ror #4\n\t"
      "eor %[x3_l], %[x3_l], %[tmp0], ror #5\n\t"
      "eor %[x3_h], %[x3_h], %[tmp1], ror #5\n\t"
      "eor %[tmp0], %[x4_l], %[x4_l], ror #17\n\t"
      "eor %[tmp1], %[x4_h], %[x4_h], ror #17\n\t"
      "eor %[x4_h], %[x4_h], %[tmp0], ror #4\n\t"
      "eor %[x4_l], %[x4_l], %[tmp1], ror #3\n\t"
      "cmp %[C], %[E]\n\t"
      "bne rbegin_%=\n\t"
      : [ x0_l ] "+r"(s->w[0][0]), [ x0_h ] "+r"(s->w[0][1]),
        [ x1_l ] "+r"(s->w[1][0]), [ x1_h ] "+r"(s->w[1][1]),
        [ x2_l ] "+r"(s->w[2][0]), [ x2_h ] "+r"(s->w[2][1]),
        [ x3_l ] "+r"(s->w[3][0]), [ x3_h ] "+r"(s->w[3][1]),
        [ x4_l ] "+r"(s->w[4][0]), [ x4_h ] "+r"(s->w[4][1]), [ C ] "+r"(C),
        [ E ] "+r"(E), [ tmp0 ] "=r"(tmp0), [ tmp1 ] "=r"(tmp1)
      :
      : "cc");
}

forceinline void ROUND(state_t* s, uint64_t C) {
  uint32_t tmp0, tmp1;
  __asm__ __volatile__(
      "eor %[x0_l], %[x0_l], %[x4_l]\n\t"
      "eor %[x4_l], %[x4_l], %[x3_l]\n\t"
      "eor %[x2_l], %[x2_l], %[x1_l]\n\t"
      "mvn %[tmp0], %[x0_l]\n\t"
      "orr %[tmp0], %[x4_l], %[tmp0]\n\t"
      "eor %[x2_l], %[x2_l], %[C_e]\n\t"
      "bic %[tmp1], %[x2_l], %[x1_l]\n\t"
      "eor %[x0_l], %[x0_l], %[tmp1]\n\t"
      "mvn %[tmp1], %[x4_l]\n\t"
      "orr %[tmp1], %[x3_l], %[tmp1]\n\t"
      "eor %[x2_l], %[x2_l], %[tmp1]\n\t"
      "bic %[tmp1], %[x1_l], %[x0_l]\n\t"
      "eor %[x4_l], %[x4_l], %[tmp1]\n\t"
      "and %[tmp1], %[x3_l], %[x2_l]\n\t"
      "eor %[x1_l], %[x1_l], %[tmp1]\n\t"
      "eor %[x3_l], %[x3_l], %[tmp0]\n\t"
      "eor %[x1_l], %[x1_l], %[x0_l]\n\t"
      "eor %[x3_l], %[x3_l], %[x2_l]\n\t"
      "eor %[x0_l], %[x0_l], %[x4_l]\n\t"
      "eor %[x0_h], %[x0_h], %[x4_h]\n\t"
      "eor %[x4_h], %[x4_h], %[x3_h]\n\t"
      "eor %[x2_h], %[x2_h], %[C_o]\n\t"
      "eor %[x2_h], %[x2_h], %[x1_h]\n\t"
      "mvn %[tmp0], %[x0_h]\n\t"
      "orr %[tmp0], %[x4_h], %[tmp0]\n\t"
      "bic %[tmp1], %[x2_h], %[x1_h]\n\t"
      "eor %[x0_h], %[x0_h], %[tmp1]\n\t"
      "mvn %[tmp1], %[x4_h]\n\t"
      "orr %[tmp1], %[x3_h], %[tmp1]\n\t"
      "eor %[x2_h], %[x2_h], %[tmp1]\n\t"
      "bic %[tmp1], %[x1_h], %[x0_h]\n\t"
      "eor %[x4_h], %[x4_h], %[tmp1]\n\t"
      "and %[tmp1], %[x3_h], %[x2_h]\n\t"
      "eor %[x1_h], %[x1_h], %[tmp1]\n\t"
      "eor %[x3_h], %[x3_h], %[tmp0]\n\t"
      "eor %[x1_h], %[x1_h], %[x0_h]\n\t"
      "eor %[x3_h], %[x3_h], %[x2_h]\n\t"
      "eor %[x0_h], %[x0_h], %[x4_h]\n\t"
      "eor %[tmp0], %[x0_l], %[x0_h], ror #4\n\t"
      "eor %[tmp1], %[x0_h], %[x0_l], ror #5\n\t"
      "eor %[x0_h], %[x0_h], %[tmp0], ror #10\n\t"
      "eor %[x0_l], %[x0_l], %[tmp1], ror #9\n\t"
      "eor %[tmp0], %[x1_l], %[x1_l], ror #11\n\t"
      "eor %[tmp1], %[x1_h], %[x1_h], ror #11\n\t"
      "eor %[x1_h], %[x1_h], %[tmp0], ror #20\n\t"
      "eor %[x1_l], %[x1_l], %[tmp1], ror #19\n\t"
      "eor %[tmp0], %[x2_l], %[x2_h], ror #2\n\t"
      "eor %[tmp1], %[x2_h], %[x2_l], ror #3\n\t"
      "eor %[x2_h], %[x2_h], %[tmp0], ror #1\n\t"
      "eor %[x2_l], %[x2_l], %[tmp1]\n\t"
      "eor %[tmp0], %[x3_l], %[x3_h], ror #3\n\t"
      "eor %[tmp1], %[x3_h], %[x3_l], ror #4\n\t"
      "eor %[x3_l], %[x3_l], %[tmp0], ror #5\n\t"
      "eor %[x3_h], %[x3_h], %[tmp1], ror #5\n\t"
      "eor %[tmp0], %[x4_l], %[x4_l], ror #17\n\t"
      "eor %[tmp1], %[x4_h], %[x4_h], ror #17\n\t"
      "eor %[x4_h], %[x4_h], %[tmp0], ror #4\n\t"
      "eor %[x4_l], %[x4_l], %[tmp1], ror #3\n\t"
      : [ x0_l ] "+r"(s->w[0][0]), [ x0_h ] "+r"(s->w[0][1]),
        [ x1_l ] "+r"(s->w[1][0]), [ x1_h ] "+r"(s->w[1][1]),
        [ x2_l ] "+r"(s->w[2][0]), [ x2_h ] "+r"(s->w[2][1]),
        [ x3_l ] "+r"(s->w[3][0]), [ x3_h ] "+r"(s->w[3][1]),
        [ x4_l ] "+r"(s->w[4][0]), [ x4_h ] "+r"(s->w[4][1]),
        [ tmp0 ] "=r"(tmp0), [ tmp1 ] "=r"(tmp1)
      : [ C_e ] "i"((uint32_t)C), [ C_o ] "i"((uint32_t)(C >> 32))
      :);
}

#define ROUND5(x0, x1, x2, x3, x4, C)                                          \
  do {                                                                         \
    uint32_t tmp0, tmp1, tmp2;                                                 \
    /* Based on the round description of Ascon given in the Bachelor's */      \
    /* thesis: "Optimizing Ascon on RISC-V" of Lars Jellema */                 \
    /* see https://github.com/Lucus16/ascon-riscv/ */                          \
    __asm__ __volatile__(                                                      \
        "eor %[x2_l], %[x2_l], %[C_e]\n\t"                                     \
        "eor %[tmp0], %[x1_l], %[x2_l]\n\t"                                    \
        "eor %[tmp1], %[x0_l], %[x4_l]\n\t"                                    \
        "eor %[tmp2], %[x3_l], %[x4_l]\n\t"                                    \
        "mvn %[x4_l], %[x4_l]\n\t"                                             \
        "orr %[x4_l], %[x3_l], %[x4_l]\n\t"                                    \
        "eor %[x4_l], %[x4_l], %[tmp0]\n\t"                                    \
        "eor %[x3_l], %[x3_l], %[x1_l]\n\t"                                    \
        "orr %[x3_l], %[x3_l], %[tmp0]\n\t"                                    \
        "eor %[x3_l], %[x3_l], %[tmp1]\n\t"                                    \
        "eor %[x2_l], %[x2_l], %[tmp1]\n\t"                                    \
        "orr %[x2_l], %[x2_l], %[x1_l]\n\t"                                    \
        "eor %[x2_l], %[x2_l], %[tmp2]\n\t"                                    \
        "bic %[x1_l], %[x1_l], %[tmp1]\n\t"                                    \
        "eor %[x1_l], %[x1_l], %[tmp2]\n\t"                                    \
        "orr %[x0_l], %[x0_l], %[tmp2]\n\t"                                    \
        "eor %[x0_l], %[x0_l], %[tmp0]\n\t"                                    \
        "eor %[x2_h], %[x2_h], %[C_o]\n\t"                                     \
        "eor %[tmp0], %[x1_h], %[x2_h]\n\t"                                    \
        "eor %[tmp1], %[x0_h], %[x4_h]\n\t"                                    \
        "eor %[tmp2], %[x3_h], %[x4_h]\n\t"                                    \
        "mvn %[x4_h], %[x4_h]\n\t"                                             \
        "orr %[x4_h], %[x3_h], %[x4_h]\n\t"                                    \
        "eor %[x4_h], %[x4_h], %[tmp0]\n\t"                                    \
        "eor %[x3_h], %[x3_h], %[x1_h]\n\t"                                    \
        "orr %[x3_h], %[x3_h], %[tmp0]\n\t"                                    \
        "eor %[x3_h], %[x3_h], %[tmp1]\n\t"                                    \
        "eor %[x2_h], %[x2_h], %[tmp1]\n\t"                                    \
        "orr %[x2_h], %[x2_h], %[x1_h]\n\t"                                    \
        "eor %[x2_h], %[x2_h], %[tmp2]\n\t"                                    \
        "bic %[x1_h], %[x1_h], %[tmp1]\n\t"                                    \
        "eor %[x1_h], %[x1_h], %[tmp2]\n\t"                                    \
        "orr %[x0_h], %[x0_h], %[tmp2]\n\t"                                    \
        "eor %[x0_h], %[x0_h], %[tmp0]\n\t"                                    \
        "eor %[tmp0], %[x2_l], %[x2_h], ror #4\n\t"                            \
        "eor %[tmp1], %[x2_h], %[x2_l], ror #5\n\t"                            \
        "eor %[x2_h], %[x2_h], %[tmp0], ror #10\n\t"                           \
        "eor %[x2_l], %[x2_l], %[tmp1], ror #9\n\t"                            \
        "eor %[tmp0], %[x3_l], %[x3_l], ror #11\n\t"                           \
        "eor %[tmp1], %[x3_h], %[x3_h], ror #11\n\t"                           \
        "eor %[x3_h], %[x3_h], %[tmp0], ror #20\n\t"                           \
        "eor %[x3_l], %[x3_l], %[tmp1], ror #19\n\t"                           \
        "eor %[tmp0], %[x4_l], %[x4_h], ror #2\n\t"                            \
        "eor %[tmp1], %[x4_h], %[x4_l], ror #3\n\t"                            \
        "eor %[x4_h], %[x4_h], %[tmp0], ror #1\n\t"                            \
        "eor %[x4_l], %[x4_l], %[tmp1]\n\t"                                    \
        "eor %[tmp0], %[x0_l], %[x0_h], ror #3\n\t"                            \
        "eor %[tmp1], %[x0_h], %[x0_l], ror #4\n\t"                            \
        "eor %[x0_l], %[x0_l], %[tmp0], ror #5\n\t"                            \
        "eor %[x0_h], %[x0_h], %[tmp1], ror #5\n\t"                            \
        "eor %[tmp0], %[x1_l], %[x1_l], ror #17\n\t"                           \
        "eor %[tmp1], %[x1_h], %[x1_h], ror #17\n\t"                           \
        "eor %[x1_h], %[x1_h], %[tmp0], ror #4\n\t"                            \
        "eor %[x1_l], %[x1_l], %[tmp1], ror #3\n\t"                            \
        : [ x0_l ] "+r"(x0.w[0]), [ x0_h ] "+r"(x0.w[1]),                      \
          [ x1_l ] "+r"(x1.w[0]), [ x1_h ] "+r"(x1.w[1]),                      \
          [ x2_l ] "+r"(x2.w[0]), [ x2_h ] "+r"(x2.w[1]),                      \
          [ x3_l ] "+r"(x3.w[0]), [ x3_h ] "+r"(x3.w[1]),                      \
          [ x4_l ] "+r"(x4.w[0]), [ x4_h ] "+r"(x4.w[1]), [ tmp0 ] "=r"(tmp0), \
          [ tmp1 ] "=r"(tmp1), [ tmp2 ] "=r"(tmp2)                             \
        : [ C_e ] "i"((uint32_t)C), [ C_o ] "i"((uint32_t)(C >> 32))           \
        :);                                                                    \
  } while (0)

forceinline void PROUNDS(state_t* s, int nr) {
  ROUND_LOOP(s, constants + START(nr), constants + 24);
}

#endif /* ROUND_H_ */