Neat script, looks pretty easy to use. The wg repo has a kprobes script too for extracting ephemeral keys from the kernel: https://git.zx2c4.com/WireGuard/tree/contrib/examples/extract-handshakes