[PATCH bpf-next v2 2/9] selftests/bpf: Add socket_kind_to_str() to socket_helpers

[Michal Luczaj]

Add function that returns string representation of socket's domain/type.

Suggested-by: Jakub Sitnicki <ja...@cloudflare.com>
Signed-off-by: Michal Luczaj <m...@rbox.co>
---
 .../selftests/bpf/prog_tests/socket_helpers.h      | 55 ++++++++++++++++++++++
 1 file changed, 55 insertions(+)

diff --git a/tools/testing/selftests/bpf/prog_tests/socket_helpers.h 
b/tools/testing/selftests/bpf/prog_tests/socket_helpers.h
index 
e505c2f2595a24be9dd5b534deef4793f9a02f77..e02cabcc814e9a6a11d61ad02cf38696fb0f27ff
 100644
--- a/tools/testing/selftests/bpf/prog_tests/socket_helpers.h
+++ b/tools/testing/selftests/bpf/prog_tests/socket_helpers.h
@@ -410,4 +410,59 @@ static inline int create_socket_pairs(int family, int 
sotype, int *c0, int *c1,
        return err;
 }
 
+static inline const char *socket_kind_to_str(int sock_fd)
+{
+       socklen_t opt_len;
+       int domain, type;
+
+       opt_len = sizeof(domain);
+       if (getsockopt(sock_fd, SOL_SOCKET, SO_DOMAIN, &domain, &opt_len))
+               FAIL_ERRNO("getsockopt(SO_DOMAIN)");
+
+       opt_len = sizeof(type);
+       if (getsockopt(sock_fd, SOL_SOCKET, SO_TYPE, &type, &opt_len))
+               FAIL_ERRNO("getsockopt(SO_TYPE)");
+
+       switch (domain) {
+       case AF_INET:
+               switch (type) {
+               case SOCK_STREAM:
+                       return "tcp4";
+               case SOCK_DGRAM:
+                       return "udp4";
+               }
+               break;
+       case AF_INET6:
+               switch (type) {
+               case SOCK_STREAM:
+                       return "tcp6";
+               case SOCK_DGRAM:
+                       return "udp6";
+               }
+               break;
+       case AF_UNIX:
+               switch (type) {
+               case SOCK_STREAM:
+                       return "u_str";
+               case SOCK_DGRAM:
+                       return "u_dgr";
+               case SOCK_SEQPACKET:
+                       return "u_seq";
+               }
+               break;
+       case AF_VSOCK:
+               switch (type) {
+               case SOCK_STREAM:
+                       return "v_str";
+               case SOCK_DGRAM:
+                       return "v_dgr";
+               case SOCK_SEQPACKET:
+                       return "v_seq";
+               }
+               break;
+       }
+
+       return "???";
+}
+
 #endif // __SOCKET_HELPERS__

-- 
2.49.0