lib/gcd.c: prevent possible div by 0

Account for all properties when a and/or b are 0:
gcd(0, 0) = 0
gcd(a, 0) = a
gcd(0, b) = b

Signed-off-by: Davidlohr Bueso <dave@gnu.org>
Cc: Eric Dumazet <eric.dumazet@gmail.com>
Signed-off-by: Andrew Morton <akpm@linux-foundation.org>

diff --git a/lib/gcd.c b/lib/gcd.c
index cce4f3cd14b36f1511b7372a5c8a8106b9495437..3657f129d7b8c5db0bee11a94a97d5d4bf251b52 100644 (file)
--- a/lib/gcd.c
+++ b/lib/gcd.c
@@ -9,6 +9,9 @@ unsigned long gcd(unsigned long a, unsigned long b)
 
        if (a < b)
                swap(a, b);
+
+       if (!b)
+               return a;
        while ((r = a % b) != 0) {
                a = b;
                b = r;