X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ian/git?p=secnet.git;a=blobdiff_plain;f=udp.c;h=bbf8c64163dbd8815001493c90a12d8f8cac30c9;hp=8ec8581f6c28aa1eaa89f002f055feb5f7167427;hb=4a1a591973e0be6f33a55b8f1fc5abc827f6969d;hpb=a15faeb2b12e9cc0e3d1352cf4c233fc61c51c1c diff --git a/udp.c b/udp.c index 8ec8581..bbf8c64 100644 --- a/udp.c +++ b/udp.c @@ -16,6 +16,8 @@ #include #include #include +#include +#include #include "util.h" #include "unaligned.h" #include "ipaddr.h" @@ -46,6 +48,30 @@ struct udp { struct sockaddr_in proxy; }; +static const char *saddr_to_string(const struct sockaddr_in *sin) { + static char bufs[2][100]; + static int b; + + b ^= 1; + snprintf(bufs[b], sizeof(bufs[b]), "[%s]:%d", + inet_ntoa(sin->sin_addr), + ntohs(sin->sin_port)); + return bufs[b]; +} + +static const char *addr_to_string(void *commst, const struct comm_addr *ca) { + struct udp *st=commst; + static char sbuf[100]; + + struct sockaddr_in la; + la.sin_addr.s_addr=htonl(st->addr); + la.sin_port=htons(st->port); + + snprintf(sbuf, sizeof(sbuf), "udp:%s-%s", + saddr_to_string(&la), saddr_to_string(&ca->sin)); + return sbuf; +} + static int udp_beforepoll(void *state, struct pollfd *fds, int *nfds_io, int *timeout_io) { @@ -215,8 +241,8 @@ static void udp_phase_hook(void *sst, uint32_t new_phase) } if (c==0) { char *argv[4], addrstr[9], portstr[5]; - sprintf(addrstr,"%08lX",(long)st->addr); - sprintf(portstr,"%04X",st->port); + sprintf(addrstr,"%08lX",(long)addr.sin_addr.s_addr); + sprintf(portstr,"%04X",addr.sin_port); argv[0]=st->authbind; argv[1]=addrstr; argv[2]=portstr; @@ -267,6 +293,7 @@ static list_t *udp_apply(closure_t *self, struct cloc loc, dict_t *context, st->ops.request_notify=request_notify; st->ops.release_notify=release_notify; st->ops.sendmsg=udp_sendmsg; + st->ops.addr_to_string=addr_to_string; st->port=0; st->use_proxy=False;