Because netlink parsing needs a way to get the netlink protocol and the socketutils.c file can only print the protocol, this commit adds an interface to get some socket information without printing it.
* defs.h (get_sockaddr_by_inode_cached, get_sockaddr_by_inode): Add. * socketutils.c (cache_print_inode_details): Remove function. (cache_inode_details, get_sockaddr_by_inode_cached, get_sockaddr_by_inode, netlink_get): New functions. (inet_print): Call print_sockaddr_by_inode_details. (unix_print): Likewise. (netlink_print): Likewise. (unix_parse_response): Call cache_inode_details. (netlink_parse_resonse): Likewise. (inet_parse_response): Likewise. --- defs.h | 2 ++ socketutils.c | 75 ++++++++++++++++++++++++++++++++++++++++++++++++++++------- 2 files changed, 69 insertions(+), 8 deletions(-) diff --git a/defs.h b/defs.h index 134e0fe..06dba19 100644 --- a/defs.h +++ b/defs.h @@ -513,6 +513,8 @@ extern int umoven_or_printaddr(struct tcb *, long, unsigned int, void *); umoven_or_printaddr((pid), (addr), sizeof(*(objp)), (void *) (objp)) extern int umovestr(struct tcb *, long, unsigned int, char *); extern int upeek(int pid, long, long *); +extern char *get_sockaddr_by_inode_cached(const unsigned long); +extern char *get_sockaddr_by_inode(const unsigned long inode, int); extern bool print_array(struct tcb *tcp, diff --git a/socketutils.c b/socketutils.c index 0ac8463..3c2ba75 100644 --- a/socketutils.c +++ b/socketutils.c @@ -57,14 +57,13 @@ static cache_entry cache[CACHE_SIZE]; #define CACHE_MASK (CACHE_SIZE - 1) static int -cache_and_print_inode_details(const unsigned long inode, char *const details) +cache_inode_details(const unsigned long inode, char *const details) { cache_entry *e = &cache[inode & CACHE_MASK]; free(e->details); e->inode = inode; e->details = details; - tprints(details); return 1; } @@ -79,6 +78,16 @@ print_sockaddr_by_inode_cached(const unsigned long inode) return false; } +char * +get_sockaddr_by_inode_cached(const unsigned long inode) +{ + const cache_entry *const e = &cache[inode & CACHE_MASK]; + if (e && inode == e->inode) { + return e->details; + } + return NULL; +} + static bool send_query(const int fd, void *req, size_t req_size) { @@ -178,7 +187,7 @@ inet_parse_response(const char *const proto_name, const void *const data, return false; } - return cache_and_print_inode_details(inode, details); + return cache_inode_details(inode, details); } static bool @@ -238,7 +247,8 @@ inet_print(const int fd, const int family, const int protocol, const unsigned long inode, const char *proto_name) { return inet_send_query(fd, family, protocol) - && receive_responses(fd, inode, proto_name, inet_parse_response); + && receive_responses(fd, inode, proto_name, inet_parse_response) + && print_sockaddr_by_inode_cached(inode); } static bool @@ -337,7 +347,7 @@ unix_parse_response(const char *proto_name, const void *data, peer_str, path_str) < 0) return -1; - return cache_and_print_inode_details(inode, details); + return cache_inode_details(inode, details); } static bool @@ -396,14 +406,15 @@ netlink_parse_response(const char *proto_name, const void *data, return -1; } - return cache_and_print_inode_details(inode, details); + return cache_inode_details(inode, details); } static bool unix_print(const int fd, const unsigned long inode) { return unix_send_query(fd, inode) - && receive_responses(fd, inode, "UNIX", unix_parse_response); + && receive_responses(fd, inode, "UNIX", unix_parse_response) + && print_sockaddr_by_inode_cached(inode); } static bool @@ -435,10 +446,58 @@ netlink_print(const int fd, const unsigned long inode) { return netlink_send_query(fd, inode) && receive_responses(fd, inode, "NETLINK", - netlink_parse_response); + netlink_parse_response) + && print_sockaddr_by_inode_cached(inode); +} + +static char * +netlink_get(const int fd, const unsigned long inode) +{ + if (netlink_send_query(fd, inode) + && receive_responses(fd, inode, "NETLINK", + netlink_parse_response)) { + cache_entry *e = &cache[inode & CACHE_MASK]; + if (e && inode == e->inode) + return e->details; + } + + return NULL; } #include "socketutils.h" +char * +get_sockaddr_by_inode(const unsigned long inode, const int proto) +{ + const int fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_SOCK_DIAG); + if (fd < 0) + return NULL; + char *details = NULL; + unsigned int i; + + if (proto != -1) { + switch (proto) { + case NETLINK: + details = netlink_get(fd, inode); + break; + } + + } else { + char *(*const protocols_get[])(int, unsigned long) = + { + netlink_get + }; + + for (i = 0; i < ARRAY_SIZE(protocols_get); ++i) { + if ((details = protocols_get[i](fd, inode))) + break; + } + } + + close(fd); + + return details; +} + /* Given an inode number of a socket, print out the details * of the ip address and port. */ bool -- 2.8.0 ------------------------------------------------------------------------------ What NetFlow Analyzer can do for you? Monitors network bandwidth and traffic patterns at an interface-level. Reveals which users, apps, and protocols are consuming the most bandwidth. Provides multi-vendor support for NetFlow, J-Flow, sFlow and other flows. Make informed decisions using capacity planning reports. https://ad.doubleclick.net/ddm/clk/305295220;132659582;e _______________________________________________ Strace-devel mailing list Strace-devel@lists.sourceforge.net https://lists.sourceforge.net/lists/listinfo/strace-devel