check payload of the NLMSG_DONE nlmsg

check the payload of the NLMSG_DONE type netlink message for possible error Signed-off-by: Jianlin Lv <Jianlin.Lv@arm.com>
vishvananda · Feb 7, 2020 · 723d1e1 · 723d1e1
1 parent 261f000
commit 723d1e1
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/nl/nl_linux.go b/nl/nl_linux.go
@@ -439,10 +439,7 @@ done:
 			if m.Header.Pid != pid {
 				continue
 			}
-			if m.Header.Type == unix.NLMSG_DONE {
-				break done
-			}
-			if m.Header.Type == unix.NLMSG_ERROR {
+			if m.Header.Type == unix.NLMSG_DONE || m.Header.Type == unix.NLMSG_ERROR {
 				native := NativeEndian()
 				error := int32(native.Uint32(m.Data[0:4]))
 				if error == 0 {