1 // SPDX-License-Identifier: GPL-2.0-only 2 /* 3 * BPF JIT compiler for LoongArch 4 * 5 * Copyright (C) 2022 Loongson Technology Corporation Limited 6 */ 7 #include "bpf_jit.h" 8 9 #define REG_TCC LOONGARCH_GPR_A6 10 #define TCC_SAVED LOONGARCH_GPR_S5 11 12 #define SAVE_RA BIT(0) 13 #define SAVE_TCC BIT(1) 14 15 static const int regmap[] = { 16 /* return value from in-kernel function, and exit value for eBPF program */ 17 [BPF_REG_0] = LOONGARCH_GPR_A5, 18 /* arguments from eBPF program to in-kernel function */ 19 [BPF_REG_1] = LOONGARCH_GPR_A0, 20 [BPF_REG_2] = LOONGARCH_GPR_A1, 21 [BPF_REG_3] = LOONGARCH_GPR_A2, 22 [BPF_REG_4] = LOONGARCH_GPR_A3, 23 [BPF_REG_5] = LOONGARCH_GPR_A4, 24 /* callee saved registers that in-kernel function will preserve */ 25 [BPF_REG_6] = LOONGARCH_GPR_S0, 26 [BPF_REG_7] = LOONGARCH_GPR_S1, 27 [BPF_REG_8] = LOONGARCH_GPR_S2, 28 [BPF_REG_9] = LOONGARCH_GPR_S3, 29 /* read-only frame pointer to access stack */ 30 [BPF_REG_FP] = LOONGARCH_GPR_S4, 31 /* temporary register for blinding constants */ 32 [BPF_REG_AX] = LOONGARCH_GPR_T0, 33 }; 34 35 static void mark_call(struct jit_ctx *ctx) 36 { 37 ctx->flags |= SAVE_RA; 38 } 39 40 static void mark_tail_call(struct jit_ctx *ctx) 41 { 42 ctx->flags |= SAVE_TCC; 43 } 44 45 static bool seen_call(struct jit_ctx *ctx) 46 { 47 return (ctx->flags & SAVE_RA); 48 } 49 50 static bool seen_tail_call(struct jit_ctx *ctx) 51 { 52 return (ctx->flags & SAVE_TCC); 53 } 54 55 static u8 tail_call_reg(struct jit_ctx *ctx) 56 { 57 if (seen_call(ctx)) 58 return TCC_SAVED; 59 60 return REG_TCC; 61 } 62 63 /* 64 * eBPF prog stack layout: 65 * 66 * high 67 * original $sp ------------> +-------------------------+ <--LOONGARCH_GPR_FP 68 * | $ra | 69 * +-------------------------+ 70 * | $fp | 71 * +-------------------------+ 72 * | $s0 | 73 * +-------------------------+ 74 * | $s1 | 75 * +-------------------------+ 76 * | $s2 | 77 * +-------------------------+ 78 * | $s3 | 79 * +-------------------------+ 80 * | $s4 | 81 * +-------------------------+ 82 * | $s5 | 83 * +-------------------------+ <--BPF_REG_FP 84 * | prog->aux->stack_depth | 85 * | (optional) | 86 * current $sp -------------> +-------------------------+ 87 * low 88 */ 89 static void build_prologue(struct jit_ctx *ctx) 90 { 91 int stack_adjust = 0, store_offset, bpf_stack_adjust; 92 93 bpf_stack_adjust = round_up(ctx->prog->aux->stack_depth, 16); 94 95 /* To store ra, fp, s0, s1, s2, s3, s4 and s5. */ 96 stack_adjust += sizeof(long) * 8; 97 98 stack_adjust = round_up(stack_adjust, 16); 99 stack_adjust += bpf_stack_adjust; 100 101 /* 102 * First instruction initializes the tail call count (TCC). 103 * On tail call we skip this instruction, and the TCC is 104 * passed in REG_TCC from the caller. 105 */ 106 emit_insn(ctx, addid, REG_TCC, LOONGARCH_GPR_ZERO, MAX_TAIL_CALL_CNT); 107 108 emit_insn(ctx, addid, LOONGARCH_GPR_SP, LOONGARCH_GPR_SP, -stack_adjust); 109 110 store_offset = stack_adjust - sizeof(long); 111 emit_insn(ctx, std, LOONGARCH_GPR_RA, LOONGARCH_GPR_SP, store_offset); 112 113 store_offset -= sizeof(long); 114 emit_insn(ctx, std, LOONGARCH_GPR_FP, LOONGARCH_GPR_SP, store_offset); 115 116 store_offset -= sizeof(long); 117 emit_insn(ctx, std, LOONGARCH_GPR_S0, LOONGARCH_GPR_SP, store_offset); 118 119 store_offset -= sizeof(long); 120 emit_insn(ctx, std, LOONGARCH_GPR_S1, LOONGARCH_GPR_SP, store_offset); 121 122 store_offset -= sizeof(long); 123 emit_insn(ctx, std, LOONGARCH_GPR_S2, LOONGARCH_GPR_SP, store_offset); 124 125 store_offset -= sizeof(long); 126 emit_insn(ctx, std, LOONGARCH_GPR_S3, LOONGARCH_GPR_SP, store_offset); 127 128 store_offset -= sizeof(long); 129 emit_insn(ctx, std, LOONGARCH_GPR_S4, LOONGARCH_GPR_SP, store_offset); 130 131 store_offset -= sizeof(long); 132 emit_insn(ctx, std, LOONGARCH_GPR_S5, LOONGARCH_GPR_SP, store_offset); 133 134 emit_insn(ctx, addid, LOONGARCH_GPR_FP, LOONGARCH_GPR_SP, stack_adjust); 135 136 if (bpf_stack_adjust) 137 emit_insn(ctx, addid, regmap[BPF_REG_FP], LOONGARCH_GPR_SP, bpf_stack_adjust); 138 139 /* 140 * Program contains calls and tail calls, so REG_TCC need 141 * to be saved across calls. 142 */ 143 if (seen_tail_call(ctx) && seen_call(ctx)) 144 move_reg(ctx, TCC_SAVED, REG_TCC); 145 146 ctx->stack_size = stack_adjust; 147 } 148 149 static void __build_epilogue(struct jit_ctx *ctx, bool is_tail_call) 150 { 151 int stack_adjust = ctx->stack_size; 152 int load_offset; 153 154 load_offset = stack_adjust - sizeof(long); 155 emit_insn(ctx, ldd, LOONGARCH_GPR_RA, LOONGARCH_GPR_SP, load_offset); 156 157 load_offset -= sizeof(long); 158 emit_insn(ctx, ldd, LOONGARCH_GPR_FP, LOONGARCH_GPR_SP, load_offset); 159 160 load_offset -= sizeof(long); 161 emit_insn(ctx, ldd, LOONGARCH_GPR_S0, LOONGARCH_GPR_SP, load_offset); 162 163 load_offset -= sizeof(long); 164 emit_insn(ctx, ldd, LOONGARCH_GPR_S1, LOONGARCH_GPR_SP, load_offset); 165 166 load_offset -= sizeof(long); 167 emit_insn(ctx, ldd, LOONGARCH_GPR_S2, LOONGARCH_GPR_SP, load_offset); 168 169 load_offset -= sizeof(long); 170 emit_insn(ctx, ldd, LOONGARCH_GPR_S3, LOONGARCH_GPR_SP, load_offset); 171 172 load_offset -= sizeof(long); 173 emit_insn(ctx, ldd, LOONGARCH_GPR_S4, LOONGARCH_GPR_SP, load_offset); 174 175 load_offset -= sizeof(long); 176 emit_insn(ctx, ldd, LOONGARCH_GPR_S5, LOONGARCH_GPR_SP, load_offset); 177 178 emit_insn(ctx, addid, LOONGARCH_GPR_SP, LOONGARCH_GPR_SP, stack_adjust); 179 180 if (!is_tail_call) { 181 /* Set return value */ 182 move_reg(ctx, LOONGARCH_GPR_A0, regmap[BPF_REG_0]); 183 /* Return to the caller */ 184 emit_insn(ctx, jirl, LOONGARCH_GPR_RA, LOONGARCH_GPR_ZERO, 0); 185 } else { 186 /* 187 * Call the next bpf prog and skip the first instruction 188 * of TCC initialization. 189 */ 190 emit_insn(ctx, jirl, LOONGARCH_GPR_T3, LOONGARCH_GPR_ZERO, 1); 191 } 192 } 193 194 static void build_epilogue(struct jit_ctx *ctx) 195 { 196 __build_epilogue(ctx, false); 197 } 198 199 bool bpf_jit_supports_kfunc_call(void) 200 { 201 return true; 202 } 203 204 /* initialized on the first pass of build_body() */ 205 static int out_offset = -1; 206 static int emit_bpf_tail_call(struct jit_ctx *ctx) 207 { 208 int off; 209 u8 tcc = tail_call_reg(ctx); 210 u8 a1 = LOONGARCH_GPR_A1; 211 u8 a2 = LOONGARCH_GPR_A2; 212 u8 t1 = LOONGARCH_GPR_T1; 213 u8 t2 = LOONGARCH_GPR_T2; 214 u8 t3 = LOONGARCH_GPR_T3; 215 const int idx0 = ctx->idx; 216 217 #define cur_offset (ctx->idx - idx0) 218 #define jmp_offset (out_offset - (cur_offset)) 219 220 /* 221 * a0: &ctx 222 * a1: &array 223 * a2: index 224 * 225 * if (index >= array->map.max_entries) 226 * goto out; 227 */ 228 off = offsetof(struct bpf_array, map.max_entries); 229 emit_insn(ctx, ldwu, t1, a1, off); 230 /* bgeu $a2, $t1, jmp_offset */ 231 if (emit_tailcall_jmp(ctx, BPF_JGE, a2, t1, jmp_offset) < 0) 232 goto toofar; 233 234 /* 235 * if (--TCC < 0) 236 * goto out; 237 */ 238 emit_insn(ctx, addid, REG_TCC, tcc, -1); 239 if (emit_tailcall_jmp(ctx, BPF_JSLT, REG_TCC, LOONGARCH_GPR_ZERO, jmp_offset) < 0) 240 goto toofar; 241 242 /* 243 * prog = array->ptrs[index]; 244 * if (!prog) 245 * goto out; 246 */ 247 emit_insn(ctx, alsld, t2, a2, a1, 2); 248 off = offsetof(struct bpf_array, ptrs); 249 emit_insn(ctx, ldd, t2, t2, off); 250 /* beq $t2, $zero, jmp_offset */ 251 if (emit_tailcall_jmp(ctx, BPF_JEQ, t2, LOONGARCH_GPR_ZERO, jmp_offset) < 0) 252 goto toofar; 253 254 /* goto *(prog->bpf_func + 4); */ 255 off = offsetof(struct bpf_prog, bpf_func); 256 emit_insn(ctx, ldd, t3, t2, off); 257 __build_epilogue(ctx, true); 258 259 /* out: */ 260 if (out_offset == -1) 261 out_offset = cur_offset; 262 if (cur_offset != out_offset) { 263 pr_err_once("tail_call out_offset = %d, expected %d!\n", 264 cur_offset, out_offset); 265 return -1; 266 } 267 268 return 0; 269 270 toofar: 271 pr_info_once("tail_call: jump too far\n"); 272 return -1; 273 #undef cur_offset 274 #undef jmp_offset 275 } 276 277 static void emit_atomic(const struct bpf_insn *insn, struct jit_ctx *ctx) 278 { 279 const u8 t1 = LOONGARCH_GPR_T1; 280 const u8 t2 = LOONGARCH_GPR_T2; 281 const u8 t3 = LOONGARCH_GPR_T3; 282 const u8 r0 = regmap[BPF_REG_0]; 283 const u8 src = regmap[insn->src_reg]; 284 const u8 dst = regmap[insn->dst_reg]; 285 const s16 off = insn->off; 286 const s32 imm = insn->imm; 287 const bool isdw = BPF_SIZE(insn->code) == BPF_DW; 288 289 move_imm(ctx, t1, off, false); 290 emit_insn(ctx, addd, t1, dst, t1); 291 move_reg(ctx, t3, src); 292 293 switch (imm) { 294 /* lock *(size *)(dst + off) <op>= src */ 295 case BPF_ADD: 296 if (isdw) 297 emit_insn(ctx, amaddd, t2, t1, src); 298 else 299 emit_insn(ctx, amaddw, t2, t1, src); 300 break; 301 case BPF_AND: 302 if (isdw) 303 emit_insn(ctx, amandd, t2, t1, src); 304 else 305 emit_insn(ctx, amandw, t2, t1, src); 306 break; 307 case BPF_OR: 308 if (isdw) 309 emit_insn(ctx, amord, t2, t1, src); 310 else 311 emit_insn(ctx, amorw, t2, t1, src); 312 break; 313 case BPF_XOR: 314 if (isdw) 315 emit_insn(ctx, amxord, t2, t1, src); 316 else 317 emit_insn(ctx, amxorw, t2, t1, src); 318 break; 319 /* src = atomic_fetch_<op>(dst + off, src) */ 320 case BPF_ADD | BPF_FETCH: 321 if (isdw) { 322 emit_insn(ctx, amaddd, src, t1, t3); 323 } else { 324 emit_insn(ctx, amaddw, src, t1, t3); 325 emit_zext_32(ctx, src, true); 326 } 327 break; 328 case BPF_AND | BPF_FETCH: 329 if (isdw) { 330 emit_insn(ctx, amandd, src, t1, t3); 331 } else { 332 emit_insn(ctx, amandw, src, t1, t3); 333 emit_zext_32(ctx, src, true); 334 } 335 break; 336 case BPF_OR | BPF_FETCH: 337 if (isdw) { 338 emit_insn(ctx, amord, src, t1, t3); 339 } else { 340 emit_insn(ctx, amorw, src, t1, t3); 341 emit_zext_32(ctx, src, true); 342 } 343 break; 344 case BPF_XOR | BPF_FETCH: 345 if (isdw) { 346 emit_insn(ctx, amxord, src, t1, t3); 347 } else { 348 emit_insn(ctx, amxorw, src, t1, t3); 349 emit_zext_32(ctx, src, true); 350 } 351 break; 352 /* src = atomic_xchg(dst + off, src); */ 353 case BPF_XCHG: 354 if (isdw) { 355 emit_insn(ctx, amswapd, src, t1, t3); 356 } else { 357 emit_insn(ctx, amswapw, src, t1, t3); 358 emit_zext_32(ctx, src, true); 359 } 360 break; 361 /* r0 = atomic_cmpxchg(dst + off, r0, src); */ 362 case BPF_CMPXCHG: 363 move_reg(ctx, t2, r0); 364 if (isdw) { 365 emit_insn(ctx, lld, r0, t1, 0); 366 emit_insn(ctx, bne, t2, r0, 4); 367 move_reg(ctx, t3, src); 368 emit_insn(ctx, scd, t3, t1, 0); 369 emit_insn(ctx, beq, t3, LOONGARCH_GPR_ZERO, -4); 370 } else { 371 emit_insn(ctx, llw, r0, t1, 0); 372 emit_zext_32(ctx, t2, true); 373 emit_zext_32(ctx, r0, true); 374 emit_insn(ctx, bne, t2, r0, 4); 375 move_reg(ctx, t3, src); 376 emit_insn(ctx, scw, t3, t1, 0); 377 emit_insn(ctx, beq, t3, LOONGARCH_GPR_ZERO, -6); 378 emit_zext_32(ctx, r0, true); 379 } 380 break; 381 } 382 } 383 384 static bool is_signed_bpf_cond(u8 cond) 385 { 386 return cond == BPF_JSGT || cond == BPF_JSLT || 387 cond == BPF_JSGE || cond == BPF_JSLE; 388 } 389 390 #define BPF_FIXUP_REG_MASK GENMASK(31, 27) 391 #define BPF_FIXUP_OFFSET_MASK GENMASK(26, 0) 392 393 bool ex_handler_bpf(const struct exception_table_entry *ex, 394 struct pt_regs *regs) 395 { 396 int dst_reg = FIELD_GET(BPF_FIXUP_REG_MASK, ex->fixup); 397 off_t offset = FIELD_GET(BPF_FIXUP_OFFSET_MASK, ex->fixup); 398 399 regs->regs[dst_reg] = 0; 400 regs->csr_era = (unsigned long)&ex->fixup - offset; 401 402 return true; 403 } 404 405 /* For accesses to BTF pointers, add an entry to the exception table */ 406 static int add_exception_handler(const struct bpf_insn *insn, 407 struct jit_ctx *ctx, 408 int dst_reg) 409 { 410 unsigned long pc; 411 off_t offset; 412 struct exception_table_entry *ex; 413 414 if (!ctx->image || !ctx->prog->aux->extable || BPF_MODE(insn->code) != BPF_PROBE_MEM) 415 return 0; 416 417 if (WARN_ON_ONCE(ctx->num_exentries >= ctx->prog->aux->num_exentries)) 418 return -EINVAL; 419 420 ex = &ctx->prog->aux->extable[ctx->num_exentries]; 421 pc = (unsigned long)&ctx->image[ctx->idx - 1]; 422 423 offset = pc - (long)&ex->insn; 424 if (WARN_ON_ONCE(offset >= 0 || offset < INT_MIN)) 425 return -ERANGE; 426 427 ex->insn = offset; 428 429 /* 430 * Since the extable follows the program, the fixup offset is always 431 * negative and limited to BPF_JIT_REGION_SIZE. Store a positive value 432 * to keep things simple, and put the destination register in the upper 433 * bits. We don't need to worry about buildtime or runtime sort 434 * modifying the upper bits because the table is already sorted, and 435 * isn't part of the main exception table. 436 */ 437 offset = (long)&ex->fixup - (pc + LOONGARCH_INSN_SIZE); 438 if (!FIELD_FIT(BPF_FIXUP_OFFSET_MASK, offset)) 439 return -ERANGE; 440 441 ex->type = EX_TYPE_BPF; 442 ex->fixup = FIELD_PREP(BPF_FIXUP_OFFSET_MASK, offset) | FIELD_PREP(BPF_FIXUP_REG_MASK, dst_reg); 443 444 ctx->num_exentries++; 445 446 return 0; 447 } 448 449 static int build_insn(const struct bpf_insn *insn, struct jit_ctx *ctx, bool extra_pass) 450 { 451 u8 tm = -1; 452 u64 func_addr; 453 bool func_addr_fixed; 454 int i = insn - ctx->prog->insnsi; 455 int ret, jmp_offset; 456 const u8 code = insn->code; 457 const u8 cond = BPF_OP(code); 458 const u8 t1 = LOONGARCH_GPR_T1; 459 const u8 t2 = LOONGARCH_GPR_T2; 460 const u8 src = regmap[insn->src_reg]; 461 const u8 dst = regmap[insn->dst_reg]; 462 const s16 off = insn->off; 463 const s32 imm = insn->imm; 464 const bool is32 = BPF_CLASS(insn->code) == BPF_ALU || BPF_CLASS(insn->code) == BPF_JMP32; 465 466 switch (code) { 467 /* dst = src */ 468 case BPF_ALU | BPF_MOV | BPF_X: 469 case BPF_ALU64 | BPF_MOV | BPF_X: 470 move_reg(ctx, dst, src); 471 emit_zext_32(ctx, dst, is32); 472 break; 473 474 /* dst = imm */ 475 case BPF_ALU | BPF_MOV | BPF_K: 476 case BPF_ALU64 | BPF_MOV | BPF_K: 477 move_imm(ctx, dst, imm, is32); 478 break; 479 480 /* dst = dst + src */ 481 case BPF_ALU | BPF_ADD | BPF_X: 482 case BPF_ALU64 | BPF_ADD | BPF_X: 483 emit_insn(ctx, addd, dst, dst, src); 484 emit_zext_32(ctx, dst, is32); 485 break; 486 487 /* dst = dst + imm */ 488 case BPF_ALU | BPF_ADD | BPF_K: 489 case BPF_ALU64 | BPF_ADD | BPF_K: 490 if (is_signed_imm12(imm)) { 491 emit_insn(ctx, addid, dst, dst, imm); 492 } else { 493 move_imm(ctx, t1, imm, is32); 494 emit_insn(ctx, addd, dst, dst, t1); 495 } 496 emit_zext_32(ctx, dst, is32); 497 break; 498 499 /* dst = dst - src */ 500 case BPF_ALU | BPF_SUB | BPF_X: 501 case BPF_ALU64 | BPF_SUB | BPF_X: 502 emit_insn(ctx, subd, dst, dst, src); 503 emit_zext_32(ctx, dst, is32); 504 break; 505 506 /* dst = dst - imm */ 507 case BPF_ALU | BPF_SUB | BPF_K: 508 case BPF_ALU64 | BPF_SUB | BPF_K: 509 if (is_signed_imm12(-imm)) { 510 emit_insn(ctx, addid, dst, dst, -imm); 511 } else { 512 move_imm(ctx, t1, imm, is32); 513 emit_insn(ctx, subd, dst, dst, t1); 514 } 515 emit_zext_32(ctx, dst, is32); 516 break; 517 518 /* dst = dst * src */ 519 case BPF_ALU | BPF_MUL | BPF_X: 520 case BPF_ALU64 | BPF_MUL | BPF_X: 521 emit_insn(ctx, muld, dst, dst, src); 522 emit_zext_32(ctx, dst, is32); 523 break; 524 525 /* dst = dst * imm */ 526 case BPF_ALU | BPF_MUL | BPF_K: 527 case BPF_ALU64 | BPF_MUL | BPF_K: 528 move_imm(ctx, t1, imm, is32); 529 emit_insn(ctx, muld, dst, dst, t1); 530 emit_zext_32(ctx, dst, is32); 531 break; 532 533 /* dst = dst / src */ 534 case BPF_ALU | BPF_DIV | BPF_X: 535 case BPF_ALU64 | BPF_DIV | BPF_X: 536 emit_zext_32(ctx, dst, is32); 537 move_reg(ctx, t1, src); 538 emit_zext_32(ctx, t1, is32); 539 emit_insn(ctx, divdu, dst, dst, t1); 540 emit_zext_32(ctx, dst, is32); 541 break; 542 543 /* dst = dst / imm */ 544 case BPF_ALU | BPF_DIV | BPF_K: 545 case BPF_ALU64 | BPF_DIV | BPF_K: 546 move_imm(ctx, t1, imm, is32); 547 emit_zext_32(ctx, dst, is32); 548 emit_insn(ctx, divdu, dst, dst, t1); 549 emit_zext_32(ctx, dst, is32); 550 break; 551 552 /* dst = dst % src */ 553 case BPF_ALU | BPF_MOD | BPF_X: 554 case BPF_ALU64 | BPF_MOD | BPF_X: 555 emit_zext_32(ctx, dst, is32); 556 move_reg(ctx, t1, src); 557 emit_zext_32(ctx, t1, is32); 558 emit_insn(ctx, moddu, dst, dst, t1); 559 emit_zext_32(ctx, dst, is32); 560 break; 561 562 /* dst = dst % imm */ 563 case BPF_ALU | BPF_MOD | BPF_K: 564 case BPF_ALU64 | BPF_MOD | BPF_K: 565 move_imm(ctx, t1, imm, is32); 566 emit_zext_32(ctx, dst, is32); 567 emit_insn(ctx, moddu, dst, dst, t1); 568 emit_zext_32(ctx, dst, is32); 569 break; 570 571 /* dst = -dst */ 572 case BPF_ALU | BPF_NEG: 573 case BPF_ALU64 | BPF_NEG: 574 move_imm(ctx, t1, imm, is32); 575 emit_insn(ctx, subd, dst, LOONGARCH_GPR_ZERO, dst); 576 emit_zext_32(ctx, dst, is32); 577 break; 578 579 /* dst = dst & src */ 580 case BPF_ALU | BPF_AND | BPF_X: 581 case BPF_ALU64 | BPF_AND | BPF_X: 582 emit_insn(ctx, and, dst, dst, src); 583 emit_zext_32(ctx, dst, is32); 584 break; 585 586 /* dst = dst & imm */ 587 case BPF_ALU | BPF_AND | BPF_K: 588 case BPF_ALU64 | BPF_AND | BPF_K: 589 if (is_unsigned_imm12(imm)) { 590 emit_insn(ctx, andi, dst, dst, imm); 591 } else { 592 move_imm(ctx, t1, imm, is32); 593 emit_insn(ctx, and, dst, dst, t1); 594 } 595 emit_zext_32(ctx, dst, is32); 596 break; 597 598 /* dst = dst | src */ 599 case BPF_ALU | BPF_OR | BPF_X: 600 case BPF_ALU64 | BPF_OR | BPF_X: 601 emit_insn(ctx, or, dst, dst, src); 602 emit_zext_32(ctx, dst, is32); 603 break; 604 605 /* dst = dst | imm */ 606 case BPF_ALU | BPF_OR | BPF_K: 607 case BPF_ALU64 | BPF_OR | BPF_K: 608 if (is_unsigned_imm12(imm)) { 609 emit_insn(ctx, ori, dst, dst, imm); 610 } else { 611 move_imm(ctx, t1, imm, is32); 612 emit_insn(ctx, or, dst, dst, t1); 613 } 614 emit_zext_32(ctx, dst, is32); 615 break; 616 617 /* dst = dst ^ src */ 618 case BPF_ALU | BPF_XOR | BPF_X: 619 case BPF_ALU64 | BPF_XOR | BPF_X: 620 emit_insn(ctx, xor, dst, dst, src); 621 emit_zext_32(ctx, dst, is32); 622 break; 623 624 /* dst = dst ^ imm */ 625 case BPF_ALU | BPF_XOR | BPF_K: 626 case BPF_ALU64 | BPF_XOR | BPF_K: 627 if (is_unsigned_imm12(imm)) { 628 emit_insn(ctx, xori, dst, dst, imm); 629 } else { 630 move_imm(ctx, t1, imm, is32); 631 emit_insn(ctx, xor, dst, dst, t1); 632 } 633 emit_zext_32(ctx, dst, is32); 634 break; 635 636 /* dst = dst << src (logical) */ 637 case BPF_ALU | BPF_LSH | BPF_X: 638 emit_insn(ctx, sllw, dst, dst, src); 639 emit_zext_32(ctx, dst, is32); 640 break; 641 642 case BPF_ALU64 | BPF_LSH | BPF_X: 643 emit_insn(ctx, slld, dst, dst, src); 644 break; 645 646 /* dst = dst << imm (logical) */ 647 case BPF_ALU | BPF_LSH | BPF_K: 648 emit_insn(ctx, slliw, dst, dst, imm); 649 emit_zext_32(ctx, dst, is32); 650 break; 651 652 case BPF_ALU64 | BPF_LSH | BPF_K: 653 emit_insn(ctx, sllid, dst, dst, imm); 654 break; 655 656 /* dst = dst >> src (logical) */ 657 case BPF_ALU | BPF_RSH | BPF_X: 658 emit_insn(ctx, srlw, dst, dst, src); 659 emit_zext_32(ctx, dst, is32); 660 break; 661 662 case BPF_ALU64 | BPF_RSH | BPF_X: 663 emit_insn(ctx, srld, dst, dst, src); 664 break; 665 666 /* dst = dst >> imm (logical) */ 667 case BPF_ALU | BPF_RSH | BPF_K: 668 emit_insn(ctx, srliw, dst, dst, imm); 669 emit_zext_32(ctx, dst, is32); 670 break; 671 672 case BPF_ALU64 | BPF_RSH | BPF_K: 673 emit_insn(ctx, srlid, dst, dst, imm); 674 break; 675 676 /* dst = dst >> src (arithmetic) */ 677 case BPF_ALU | BPF_ARSH | BPF_X: 678 emit_insn(ctx, sraw, dst, dst, src); 679 emit_zext_32(ctx, dst, is32); 680 break; 681 682 case BPF_ALU64 | BPF_ARSH | BPF_X: 683 emit_insn(ctx, srad, dst, dst, src); 684 break; 685 686 /* dst = dst >> imm (arithmetic) */ 687 case BPF_ALU | BPF_ARSH | BPF_K: 688 emit_insn(ctx, sraiw, dst, dst, imm); 689 emit_zext_32(ctx, dst, is32); 690 break; 691 692 case BPF_ALU64 | BPF_ARSH | BPF_K: 693 emit_insn(ctx, sraid, dst, dst, imm); 694 break; 695 696 /* dst = BSWAP##imm(dst) */ 697 case BPF_ALU | BPF_END | BPF_FROM_LE: 698 switch (imm) { 699 case 16: 700 /* zero-extend 16 bits into 64 bits */ 701 emit_insn(ctx, bstrpickd, dst, dst, 15, 0); 702 break; 703 case 32: 704 /* zero-extend 32 bits into 64 bits */ 705 emit_zext_32(ctx, dst, is32); 706 break; 707 case 64: 708 /* do nothing */ 709 break; 710 } 711 break; 712 713 case BPF_ALU | BPF_END | BPF_FROM_BE: 714 switch (imm) { 715 case 16: 716 emit_insn(ctx, revb2h, dst, dst); 717 /* zero-extend 16 bits into 64 bits */ 718 emit_insn(ctx, bstrpickd, dst, dst, 15, 0); 719 break; 720 case 32: 721 emit_insn(ctx, revb2w, dst, dst); 722 /* zero-extend 32 bits into 64 bits */ 723 emit_zext_32(ctx, dst, is32); 724 break; 725 case 64: 726 emit_insn(ctx, revbd, dst, dst); 727 break; 728 } 729 break; 730 731 /* PC += off if dst cond src */ 732 case BPF_JMP | BPF_JEQ | BPF_X: 733 case BPF_JMP | BPF_JNE | BPF_X: 734 case BPF_JMP | BPF_JGT | BPF_X: 735 case BPF_JMP | BPF_JGE | BPF_X: 736 case BPF_JMP | BPF_JLT | BPF_X: 737 case BPF_JMP | BPF_JLE | BPF_X: 738 case BPF_JMP | BPF_JSGT | BPF_X: 739 case BPF_JMP | BPF_JSGE | BPF_X: 740 case BPF_JMP | BPF_JSLT | BPF_X: 741 case BPF_JMP | BPF_JSLE | BPF_X: 742 case BPF_JMP32 | BPF_JEQ | BPF_X: 743 case BPF_JMP32 | BPF_JNE | BPF_X: 744 case BPF_JMP32 | BPF_JGT | BPF_X: 745 case BPF_JMP32 | BPF_JGE | BPF_X: 746 case BPF_JMP32 | BPF_JLT | BPF_X: 747 case BPF_JMP32 | BPF_JLE | BPF_X: 748 case BPF_JMP32 | BPF_JSGT | BPF_X: 749 case BPF_JMP32 | BPF_JSGE | BPF_X: 750 case BPF_JMP32 | BPF_JSLT | BPF_X: 751 case BPF_JMP32 | BPF_JSLE | BPF_X: 752 jmp_offset = bpf2la_offset(i, off, ctx); 753 move_reg(ctx, t1, dst); 754 move_reg(ctx, t2, src); 755 if (is_signed_bpf_cond(BPF_OP(code))) { 756 emit_sext_32(ctx, t1, is32); 757 emit_sext_32(ctx, t2, is32); 758 } else { 759 emit_zext_32(ctx, t1, is32); 760 emit_zext_32(ctx, t2, is32); 761 } 762 if (emit_cond_jmp(ctx, cond, t1, t2, jmp_offset) < 0) 763 goto toofar; 764 break; 765 766 /* PC += off if dst cond imm */ 767 case BPF_JMP | BPF_JEQ | BPF_K: 768 case BPF_JMP | BPF_JNE | BPF_K: 769 case BPF_JMP | BPF_JGT | BPF_K: 770 case BPF_JMP | BPF_JGE | BPF_K: 771 case BPF_JMP | BPF_JLT | BPF_K: 772 case BPF_JMP | BPF_JLE | BPF_K: 773 case BPF_JMP | BPF_JSGT | BPF_K: 774 case BPF_JMP | BPF_JSGE | BPF_K: 775 case BPF_JMP | BPF_JSLT | BPF_K: 776 case BPF_JMP | BPF_JSLE | BPF_K: 777 case BPF_JMP32 | BPF_JEQ | BPF_K: 778 case BPF_JMP32 | BPF_JNE | BPF_K: 779 case BPF_JMP32 | BPF_JGT | BPF_K: 780 case BPF_JMP32 | BPF_JGE | BPF_K: 781 case BPF_JMP32 | BPF_JLT | BPF_K: 782 case BPF_JMP32 | BPF_JLE | BPF_K: 783 case BPF_JMP32 | BPF_JSGT | BPF_K: 784 case BPF_JMP32 | BPF_JSGE | BPF_K: 785 case BPF_JMP32 | BPF_JSLT | BPF_K: 786 case BPF_JMP32 | BPF_JSLE | BPF_K: 787 jmp_offset = bpf2la_offset(i, off, ctx); 788 if (imm) { 789 move_imm(ctx, t1, imm, false); 790 tm = t1; 791 } else { 792 /* If imm is 0, simply use zero register. */ 793 tm = LOONGARCH_GPR_ZERO; 794 } 795 move_reg(ctx, t2, dst); 796 if (is_signed_bpf_cond(BPF_OP(code))) { 797 emit_sext_32(ctx, tm, is32); 798 emit_sext_32(ctx, t2, is32); 799 } else { 800 emit_zext_32(ctx, tm, is32); 801 emit_zext_32(ctx, t2, is32); 802 } 803 if (emit_cond_jmp(ctx, cond, t2, tm, jmp_offset) < 0) 804 goto toofar; 805 break; 806 807 /* PC += off if dst & src */ 808 case BPF_JMP | BPF_JSET | BPF_X: 809 case BPF_JMP32 | BPF_JSET | BPF_X: 810 jmp_offset = bpf2la_offset(i, off, ctx); 811 emit_insn(ctx, and, t1, dst, src); 812 emit_zext_32(ctx, t1, is32); 813 if (emit_cond_jmp(ctx, cond, t1, LOONGARCH_GPR_ZERO, jmp_offset) < 0) 814 goto toofar; 815 break; 816 817 /* PC += off if dst & imm */ 818 case BPF_JMP | BPF_JSET | BPF_K: 819 case BPF_JMP32 | BPF_JSET | BPF_K: 820 jmp_offset = bpf2la_offset(i, off, ctx); 821 move_imm(ctx, t1, imm, is32); 822 emit_insn(ctx, and, t1, dst, t1); 823 emit_zext_32(ctx, t1, is32); 824 if (emit_cond_jmp(ctx, cond, t1, LOONGARCH_GPR_ZERO, jmp_offset) < 0) 825 goto toofar; 826 break; 827 828 /* PC += off */ 829 case BPF_JMP | BPF_JA: 830 jmp_offset = bpf2la_offset(i, off, ctx); 831 if (emit_uncond_jmp(ctx, jmp_offset) < 0) 832 goto toofar; 833 break; 834 835 /* function call */ 836 case BPF_JMP | BPF_CALL: 837 mark_call(ctx); 838 ret = bpf_jit_get_func_addr(ctx->prog, insn, extra_pass, 839 &func_addr, &func_addr_fixed); 840 if (ret < 0) 841 return ret; 842 843 move_addr(ctx, t1, func_addr); 844 emit_insn(ctx, jirl, t1, LOONGARCH_GPR_RA, 0); 845 move_reg(ctx, regmap[BPF_REG_0], LOONGARCH_GPR_A0); 846 break; 847 848 /* tail call */ 849 case BPF_JMP | BPF_TAIL_CALL: 850 mark_tail_call(ctx); 851 if (emit_bpf_tail_call(ctx) < 0) 852 return -EINVAL; 853 break; 854 855 /* function return */ 856 case BPF_JMP | BPF_EXIT: 857 if (i == ctx->prog->len - 1) 858 break; 859 860 jmp_offset = epilogue_offset(ctx); 861 if (emit_uncond_jmp(ctx, jmp_offset) < 0) 862 goto toofar; 863 break; 864 865 /* dst = imm64 */ 866 case BPF_LD | BPF_IMM | BPF_DW: 867 { 868 const u64 imm64 = (u64)(insn + 1)->imm << 32 | (u32)insn->imm; 869 870 move_imm(ctx, dst, imm64, is32); 871 return 1; 872 } 873 874 /* dst = *(size *)(src + off) */ 875 case BPF_LDX | BPF_MEM | BPF_B: 876 case BPF_LDX | BPF_MEM | BPF_H: 877 case BPF_LDX | BPF_MEM | BPF_W: 878 case BPF_LDX | BPF_MEM | BPF_DW: 879 case BPF_LDX | BPF_PROBE_MEM | BPF_DW: 880 case BPF_LDX | BPF_PROBE_MEM | BPF_W: 881 case BPF_LDX | BPF_PROBE_MEM | BPF_H: 882 case BPF_LDX | BPF_PROBE_MEM | BPF_B: 883 switch (BPF_SIZE(code)) { 884 case BPF_B: 885 if (is_signed_imm12(off)) { 886 emit_insn(ctx, ldbu, dst, src, off); 887 } else { 888 move_imm(ctx, t1, off, is32); 889 emit_insn(ctx, ldxbu, dst, src, t1); 890 } 891 break; 892 case BPF_H: 893 if (is_signed_imm12(off)) { 894 emit_insn(ctx, ldhu, dst, src, off); 895 } else { 896 move_imm(ctx, t1, off, is32); 897 emit_insn(ctx, ldxhu, dst, src, t1); 898 } 899 break; 900 case BPF_W: 901 if (is_signed_imm12(off)) { 902 emit_insn(ctx, ldwu, dst, src, off); 903 } else if (is_signed_imm14(off)) { 904 emit_insn(ctx, ldptrw, dst, src, off); 905 } else { 906 move_imm(ctx, t1, off, is32); 907 emit_insn(ctx, ldxwu, dst, src, t1); 908 } 909 break; 910 case BPF_DW: 911 move_imm(ctx, t1, off, is32); 912 emit_insn(ctx, ldxd, dst, src, t1); 913 break; 914 } 915 916 ret = add_exception_handler(insn, ctx, dst); 917 if (ret) 918 return ret; 919 break; 920 921 /* *(size *)(dst + off) = imm */ 922 case BPF_ST | BPF_MEM | BPF_B: 923 case BPF_ST | BPF_MEM | BPF_H: 924 case BPF_ST | BPF_MEM | BPF_W: 925 case BPF_ST | BPF_MEM | BPF_DW: 926 switch (BPF_SIZE(code)) { 927 case BPF_B: 928 move_imm(ctx, t1, imm, is32); 929 if (is_signed_imm12(off)) { 930 emit_insn(ctx, stb, t1, dst, off); 931 } else { 932 move_imm(ctx, t2, off, is32); 933 emit_insn(ctx, stxb, t1, dst, t2); 934 } 935 break; 936 case BPF_H: 937 move_imm(ctx, t1, imm, is32); 938 if (is_signed_imm12(off)) { 939 emit_insn(ctx, sth, t1, dst, off); 940 } else { 941 move_imm(ctx, t2, off, is32); 942 emit_insn(ctx, stxh, t1, dst, t2); 943 } 944 break; 945 case BPF_W: 946 move_imm(ctx, t1, imm, is32); 947 if (is_signed_imm12(off)) { 948 emit_insn(ctx, stw, t1, dst, off); 949 } else if (is_signed_imm14(off)) { 950 emit_insn(ctx, stptrw, t1, dst, off); 951 } else { 952 move_imm(ctx, t2, off, is32); 953 emit_insn(ctx, stxw, t1, dst, t2); 954 } 955 break; 956 case BPF_DW: 957 move_imm(ctx, t1, imm, is32); 958 if (is_signed_imm12(off)) { 959 emit_insn(ctx, std, t1, dst, off); 960 } else if (is_signed_imm14(off)) { 961 emit_insn(ctx, stptrd, t1, dst, off); 962 } else { 963 move_imm(ctx, t2, off, is32); 964 emit_insn(ctx, stxd, t1, dst, t2); 965 } 966 break; 967 } 968 break; 969 970 /* *(size *)(dst + off) = src */ 971 case BPF_STX | BPF_MEM | BPF_B: 972 case BPF_STX | BPF_MEM | BPF_H: 973 case BPF_STX | BPF_MEM | BPF_W: 974 case BPF_STX | BPF_MEM | BPF_DW: 975 switch (BPF_SIZE(code)) { 976 case BPF_B: 977 if (is_signed_imm12(off)) { 978 emit_insn(ctx, stb, src, dst, off); 979 } else { 980 move_imm(ctx, t1, off, is32); 981 emit_insn(ctx, stxb, src, dst, t1); 982 } 983 break; 984 case BPF_H: 985 if (is_signed_imm12(off)) { 986 emit_insn(ctx, sth, src, dst, off); 987 } else { 988 move_imm(ctx, t1, off, is32); 989 emit_insn(ctx, stxh, src, dst, t1); 990 } 991 break; 992 case BPF_W: 993 if (is_signed_imm12(off)) { 994 emit_insn(ctx, stw, src, dst, off); 995 } else if (is_signed_imm14(off)) { 996 emit_insn(ctx, stptrw, src, dst, off); 997 } else { 998 move_imm(ctx, t1, off, is32); 999 emit_insn(ctx, stxw, src, dst, t1); 1000 } 1001 break; 1002 case BPF_DW: 1003 if (is_signed_imm12(off)) { 1004 emit_insn(ctx, std, src, dst, off); 1005 } else if (is_signed_imm14(off)) { 1006 emit_insn(ctx, stptrd, src, dst, off); 1007 } else { 1008 move_imm(ctx, t1, off, is32); 1009 emit_insn(ctx, stxd, src, dst, t1); 1010 } 1011 break; 1012 } 1013 break; 1014 1015 case BPF_STX | BPF_ATOMIC | BPF_W: 1016 case BPF_STX | BPF_ATOMIC | BPF_DW: 1017 emit_atomic(insn, ctx); 1018 break; 1019 1020 /* Speculation barrier */ 1021 case BPF_ST | BPF_NOSPEC: 1022 break; 1023 1024 default: 1025 pr_err("bpf_jit: unknown opcode %02x\n", code); 1026 return -EINVAL; 1027 } 1028 1029 return 0; 1030 1031 toofar: 1032 pr_info_once("bpf_jit: opcode %02x, jump too far\n", code); 1033 return -E2BIG; 1034 } 1035 1036 static int build_body(struct jit_ctx *ctx, bool extra_pass) 1037 { 1038 int i; 1039 const struct bpf_prog *prog = ctx->prog; 1040 1041 for (i = 0; i < prog->len; i++) { 1042 const struct bpf_insn *insn = &prog->insnsi[i]; 1043 int ret; 1044 1045 if (ctx->image == NULL) 1046 ctx->offset[i] = ctx->idx; 1047 1048 ret = build_insn(insn, ctx, extra_pass); 1049 if (ret > 0) { 1050 i++; 1051 if (ctx->image == NULL) 1052 ctx->offset[i] = ctx->idx; 1053 continue; 1054 } 1055 if (ret) 1056 return ret; 1057 } 1058 1059 if (ctx->image == NULL) 1060 ctx->offset[i] = ctx->idx; 1061 1062 return 0; 1063 } 1064 1065 /* Fill space with break instructions */ 1066 static void jit_fill_hole(void *area, unsigned int size) 1067 { 1068 u32 *ptr; 1069 1070 /* We are guaranteed to have aligned memory */ 1071 for (ptr = area; size >= sizeof(u32); size -= sizeof(u32)) 1072 *ptr++ = INSN_BREAK; 1073 } 1074 1075 static int validate_code(struct jit_ctx *ctx) 1076 { 1077 int i; 1078 union loongarch_instruction insn; 1079 1080 for (i = 0; i < ctx->idx; i++) { 1081 insn = ctx->image[i]; 1082 /* Check INSN_BREAK */ 1083 if (insn.word == INSN_BREAK) 1084 return -1; 1085 } 1086 1087 if (WARN_ON_ONCE(ctx->num_exentries != ctx->prog->aux->num_exentries)) 1088 return -1; 1089 1090 return 0; 1091 } 1092 1093 struct bpf_prog *bpf_int_jit_compile(struct bpf_prog *prog) 1094 { 1095 bool tmp_blinded = false, extra_pass = false; 1096 u8 *image_ptr; 1097 int image_size, prog_size, extable_size; 1098 struct jit_ctx ctx; 1099 struct jit_data *jit_data; 1100 struct bpf_binary_header *header; 1101 struct bpf_prog *tmp, *orig_prog = prog; 1102 1103 /* 1104 * If BPF JIT was not enabled then we must fall back to 1105 * the interpreter. 1106 */ 1107 if (!prog->jit_requested) 1108 return orig_prog; 1109 1110 tmp = bpf_jit_blind_constants(prog); 1111 /* 1112 * If blinding was requested and we failed during blinding, 1113 * we must fall back to the interpreter. Otherwise, we save 1114 * the new JITed code. 1115 */ 1116 if (IS_ERR(tmp)) 1117 return orig_prog; 1118 1119 if (tmp != prog) { 1120 tmp_blinded = true; 1121 prog = tmp; 1122 } 1123 1124 jit_data = prog->aux->jit_data; 1125 if (!jit_data) { 1126 jit_data = kzalloc(sizeof(*jit_data), GFP_KERNEL); 1127 if (!jit_data) { 1128 prog = orig_prog; 1129 goto out; 1130 } 1131 prog->aux->jit_data = jit_data; 1132 } 1133 if (jit_data->ctx.offset) { 1134 ctx = jit_data->ctx; 1135 image_ptr = jit_data->image; 1136 header = jit_data->header; 1137 extra_pass = true; 1138 prog_size = sizeof(u32) * ctx.idx; 1139 goto skip_init_ctx; 1140 } 1141 1142 memset(&ctx, 0, sizeof(ctx)); 1143 ctx.prog = prog; 1144 1145 ctx.offset = kvcalloc(prog->len + 1, sizeof(u32), GFP_KERNEL); 1146 if (ctx.offset == NULL) { 1147 prog = orig_prog; 1148 goto out_offset; 1149 } 1150 1151 /* 1. Initial fake pass to compute ctx->idx and set ctx->flags */ 1152 build_prologue(&ctx); 1153 if (build_body(&ctx, extra_pass)) { 1154 prog = orig_prog; 1155 goto out_offset; 1156 } 1157 ctx.epilogue_offset = ctx.idx; 1158 build_epilogue(&ctx); 1159 1160 extable_size = prog->aux->num_exentries * sizeof(struct exception_table_entry); 1161 1162 /* Now we know the actual image size. 1163 * As each LoongArch instruction is of length 32bit, 1164 * we are translating number of JITed intructions into 1165 * the size required to store these JITed code. 1166 */ 1167 prog_size = sizeof(u32) * ctx.idx; 1168 image_size = prog_size + extable_size; 1169 /* Now we know the size of the structure to make */ 1170 header = bpf_jit_binary_alloc(image_size, &image_ptr, 1171 sizeof(u32), jit_fill_hole); 1172 if (header == NULL) { 1173 prog = orig_prog; 1174 goto out_offset; 1175 } 1176 1177 /* 2. Now, the actual pass to generate final JIT code */ 1178 ctx.image = (union loongarch_instruction *)image_ptr; 1179 if (extable_size) 1180 prog->aux->extable = (void *)image_ptr + prog_size; 1181 1182 skip_init_ctx: 1183 ctx.idx = 0; 1184 ctx.num_exentries = 0; 1185 1186 build_prologue(&ctx); 1187 if (build_body(&ctx, extra_pass)) { 1188 bpf_jit_binary_free(header); 1189 prog = orig_prog; 1190 goto out_offset; 1191 } 1192 build_epilogue(&ctx); 1193 1194 /* 3. Extra pass to validate JITed code */ 1195 if (validate_code(&ctx)) { 1196 bpf_jit_binary_free(header); 1197 prog = orig_prog; 1198 goto out_offset; 1199 } 1200 1201 /* And we're done */ 1202 if (bpf_jit_enable > 1) 1203 bpf_jit_dump(prog->len, prog_size, 2, ctx.image); 1204 1205 /* Update the icache */ 1206 flush_icache_range((unsigned long)header, (unsigned long)(ctx.image + ctx.idx)); 1207 1208 if (!prog->is_func || extra_pass) { 1209 int err; 1210 1211 if (extra_pass && ctx.idx != jit_data->ctx.idx) { 1212 pr_err_once("multi-func JIT bug %d != %d\n", 1213 ctx.idx, jit_data->ctx.idx); 1214 goto out_free; 1215 } 1216 err = bpf_jit_binary_lock_ro(header); 1217 if (err) { 1218 pr_err_once("bpf_jit_binary_lock_ro() returned %d\n", 1219 err); 1220 goto out_free; 1221 } 1222 } else { 1223 jit_data->ctx = ctx; 1224 jit_data->image = image_ptr; 1225 jit_data->header = header; 1226 } 1227 prog->jited = 1; 1228 prog->jited_len = prog_size; 1229 prog->bpf_func = (void *)ctx.image; 1230 1231 if (!prog->is_func || extra_pass) { 1232 int i; 1233 1234 /* offset[prog->len] is the size of program */ 1235 for (i = 0; i <= prog->len; i++) 1236 ctx.offset[i] *= LOONGARCH_INSN_SIZE; 1237 bpf_prog_fill_jited_linfo(prog, ctx.offset + 1); 1238 1239 out_offset: 1240 kvfree(ctx.offset); 1241 kfree(jit_data); 1242 prog->aux->jit_data = NULL; 1243 } 1244 1245 out: 1246 if (tmp_blinded) 1247 bpf_jit_prog_release_other(prog, prog == orig_prog ? tmp : orig_prog); 1248 1249 out_offset = -1; 1250 1251 return prog; 1252 1253 out_free: 1254 bpf_jit_binary_free(header); 1255 prog->bpf_func = NULL; 1256 prog->jited = 0; 1257 prog->jited_len = 0; 1258 goto out_offset; 1259 } 1260 1261 /* Indicate the JIT backend supports mixing bpf2bpf and tailcalls. */ 1262 bool bpf_jit_supports_subprog_tailcalls(void) 1263 { 1264 return true; 1265 } 1266