- first_action += "\t\tif (unlikely(diff > 1000000000))\n"
- first_action += "\t\t\trate->sent_bytes = 0;\n"
- first_action += "\t\telse if (likely(diff > 0))\n"
- first_action += f"\t\t\textra_bytes = ((uint64_t)diff) * {math.floor(value)} / 1000000000;\n"
- first_action += "\t}\n"
- first_action += "\tif (rate->sent_bytes - ((int64_t)extra_bytes) <= 0) {\n"
- first_action += "\t\trate->sent_bytes = data_end - pktdata;\n"
- first_action += "\t\trate->sent_time = time;\n"
- first_action += "\t\tbpf_spin_unlock(&rate->lock);\n"
- first_action += "\t} else {\n"
- first_action += "\t\tbpf_spin_unlock(&rate->lock);\n"
- first_action += "\t\treturn XDP_DROP;\n"
- first_action += "\t}\n"
- first_action += "}\n"
+ first_action += "\t\tif (unlikely(diff > 1000000000))\n"
+ first_action += f"\t\t\t{accessor} = 0;\n"
+ first_action += "\t\telse if (likely(diff > 0))\n"
+ first_action += f"\t\t\tallowed_since_last = ((uint64_t)diff) * {math.floor(value)} / 1000000000;\n"
+ first_action += "\t}\n"
+ first_action += f"\tif ({accessor} - ((int64_t)allowed_since_last) <= 0)" + " {\n"
+ if ty == "0x8006":
+ first_action += f"\t\t{accessor} = data_end - pktdata;\n"
+ else:
+ first_action += f"\t\t{accessor} = 1;\n"
+ first_action += "\t\trate->sent_time = time;\n"
+ first_action += "\t\tbpf_spin_unlock(&rate->lock);\n"
+ first_action += "\t} else {\n"
+ first_action += "\t\tbpf_spin_unlock(&rate->lock);\n"
+ first_action += "\t\treturn XDP_DROP;\n"
+ first_action += "\t}\n"
+ first_action += "}\n"