net: skb: change the definition SKB_DR_SET()

The SKB_DR_OR() is used to set the drop reason to a value when it is
not set or specified yet. SKB_NOT_DROPPED_YET should also be considered
as not set.

Reviewed-by: Jiang Biao <benbjiang@tencent.com>
Reviewed-by: Hao Peng <flyingpeng@tencent.com>
Signed-off-by: Menglong Dong <imagedong@tencent.com>
Signed-off-by: David S. Miller <davem@davemloft.net>

diff --git a/include/linux/skbuff.h b/include/linux/skbuff.h
index 9d82a8b6c8f16c072ceb8ca10157317634c3fe62..5a2b29df6cab925e5994159f92930d9861112d8c 100644 (file)
--- a/include/linux/skbuff.h
+++ b/include/linux/skbuff.h
@@ -510,7 +510,8 @@ enum skb_drop_reason {
        (name = SKB_DROP_REASON_##reason)
 #define SKB_DR_OR(name, reason)                                        \
        do {                                                    \
-               if (name == SKB_DROP_REASON_NOT_SPECIFIED)      \
+               if (name == SKB_DROP_REASON_NOT_SPECIFIED ||    \
+                   name == SKB_NOT_DROPPED_YET)                \
                        SKB_DR_SET(name, reason);               \
        } while (0)