Fix: check for sizeof sa_family to save sa_family in accept and connect

[lttng-modules.git] / instrumentation / syscalls / headers / x86-64-syscalls-3.10.0-rc7_pointers_override.h

diff --git a/instrumentation/syscalls/headers/x86-64-syscalls-3.10.0-rc7_pointers_override.h b/instrumentation/syscalls/headers/x86-64-syscalls-3.10.0-rc7_pointers_override.h
index 22c328ac397e7205a1d09a73dc2f96ea7d1a8a9d..b262228f9a641835a287489ebd724ac27e696ee8 100644 (file)
--- a/instrumentation/syscalls/headers/x86-64-syscalls-3.10.0-rc7_pointers_override.h
+++ b/instrumentation/syscalls/headers/x86-64-syscalls-3.10.0-rc7_pointers_override.h
@@ -14,7 +14,7 @@ SC_LTTNG_TRACEPOINT_EVENT_CODE(connect,
        TP_code_pre(
                sc_in(
                        memset(tp_locvar, 0, sizeof(*tp_locvar));
-                       if (addrlen < sizeof(struct sockaddr))
+                       if (addrlen < sizeof(tp_locvar->sa_family))
                                goto skip_code;
                        (void) get_user(tp_locvar->sa_family, &uservaddr->sa_family);
                        switch (tp_locvar->sa_family) {
@@ -66,7 +66,7 @@ SC_LTTNG_TRACEPOINT_EVENT_CODE(connect,
                (void) get_user(tp_locvar->uaddr_len, upeer_addrlen);                                                   \
        )                                                                                                               \
        sc_out(                                                                                                         \
-               if (tp_locvar->uaddr_len < sizeof(struct sockaddr))                                                     \
+               if (tp_locvar->uaddr_len < sizeof(tp_locvar->sa_family))                                                \
                        goto skip_code;                                                                                 \
                (void) get_user(tp_locvar->sa_family, &upeer_sockaddr->sa_family);                                      \
                switch (tp_locvar->sa_family) {                                                                         \