xnu-2050.18.24.tar.gz

[apple/xnu.git] / pexpert / gen / device_tree.c

diff --git a/pexpert/gen/device_tree.c b/pexpert/gen/device_tree.c
index dc3ea9ddc05f7d0a6a3830b13cbc4c1bb7a38143..d78bed0bf68ff0f6acef7319f99d43c175a4f4bd 100644 (file)
--- a/pexpert/gen/device_tree.c
+++ b/pexpert/gen/device_tree.c
@@ -127,11 +127,18 @@ GetNextChild(RealDTEntry sibling)
 static const char *
 GetNextComponent(const char *cp, char *bp)
 {
+       size_t length = 0;
+       char *origbp = bp;
+
        while (*cp != 0) {
                if (*cp == kDTPathNameSeparator) {
                        cp++;
                        break;
                }
+               if (++length > kDTMaxEntryNameLength) {
+                       *origbp = '\0';
+                       return cp;
+               }
                *bp++ = *cp++;
        }
        *bp = 0;