d/freebsd-skq
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

dtc: don't crash if the argument is a directory.

Browse Source
This commit is contained in:
Rui Paulo 2014-06-10 06:04:25 +00:00
parent b60aab0101
commit 23afbfa96e
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
usr.bin/dtc/fdt.cc
View File

@ -42,6 +42,8 @@
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include "dtb.hh"
namespace dtc
@ -1078,6 +1080,13 @@ device_tree::buffer_for_file(const char *path)
fprintf(stderr, "Unable to open file %s\n", path);
return 0;
}
struct stat st;
if (fstat(source, &st) == 0 && S_ISDIR(st.st_mode))
{
fprintf(stderr, "File %s is a directory\n", path);
close(source);
return 0;
}
input_buffer *b = new mmap_input_buffer(source);
// Keep the buffer that owns the memory around for the lifetime
// of this FDT. Ones simply referring to it may have shorter