udp: refactor udp code