sortix-mirror/mkinitrd/lsinitrd.cpp

156 lines
3.5 KiB
C++

/******************************************************************************
COPYRIGHT(C) JONAS 'SORTIE' TERMANSEN 2011.
This file is part of Sortix.
Sortix is free software: you can redistribute it and/or modify it under the
terms of the GNU General Public License as published by the Free Software
Foundation, either version 3 of the License, or (at your option) any later
version.
Sortix is distributed in the hope that it will be useful, but WITHOUT ANY
WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
details.
You should have received a copy of the GNU General Public License along
with Sortix. If not, see <http://www.gnu.org/licenses/>.
lsinitrd.cpp
Lists the files in a Sortix ramdisk.
******************************************************************************/
#include <stdio.h>
#include <string.h>
#include <stdint.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <errno.h>
#include <error.h>
#include <unistd.h>
#include <sortix/initrd.h>
bool readall(int fd, void* p, size_t size)
{
uint8_t* buffer = (uint8_t*) p;
size_t bytesread = 0;
while ( bytesread < size )
{
ssize_t justread = read(fd, buffer + bytesread, size - bytesread);
if ( justread <= 0 ) { return false; }
bytesread += justread;
}
return true;
}
void usage(int argc, char* argv[])
{
printf("usage: %s [OPTIONS] <ramdisks>\n", argv[0]);
printf("Options:\n");
printf(" -q Surpress normal output\n");
printf(" -v Be verbose\n");
printf(" --usage Display this screen\n");
printf(" --help Display this screen\n");
printf(" --version Display version information\n");
}
void version()
{
printf("lsinitrd 0.1\n");
printf("Copyright (C) 2011 Jonas 'Sortie' Termansen\n");
printf("This is free software; see the source for copying conditions. There is NO\n");
printf("warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\n");
printf("website: http://www.maxsi.org/software/sortix/\n");
}
bool verbose = false;
int listfiles(const char* filepath)
{
int fd = open(filepath, O_RDONLY);
if ( fd < 0 ) { error(0, errno, "%s", filepath); return 1; }
Sortix::InitRD::Header header;
if ( !readall(fd, &header, sizeof(header)) )
{
error(0, errno, "read: %s", filepath);
close(fd);
return 1;
}
if ( strcmp(header.magic, "sortix-initrd-1") != 0 )
{
error(0, 0, "not a sortix ramdisk: %s", filepath);
close(fd);
return 1;
}
for ( uint32_t i = 0; i < header.numfiles; i++ )
{
Sortix::InitRD::FileHeader fileheader;
if ( !readall(fd, &fileheader, sizeof(fileheader)) )
{
error(0, errno, "read: %s", filepath);
close(fd);
return 1;
}
printf("%s\n", fileheader.name);
}
close(fd);
return 0;
}
int main(int argc, char* argv[])
{
if ( argc < 2 ) { usage(argc, argv); return 0; }
for ( int i = 1; i < argc; i++ )
{
if ( strcmp(argv[i], "-q") == 0 )
{
verbose = false;
argv[i] = NULL;
}
else if ( strcmp(argv[i], "-v") == 0 )
{
verbose = true;
argv[i] = NULL;
}
else if ( strcmp(argv[i], "--usage") == 0 )
{
usage(argc, argv);
return 0;
}
else if ( strcmp(argv[i], "--help") == 0 )
{
usage(argc, argv);
return 0;
}
else if ( strcmp(argv[i], "--version") == 0 )
{
version();
return 0;
}
}
int result = 0;
for ( int i = 1; i < argc; i++ )
{
if ( argv[i] == NULL ) { continue; }
result |= listfiles(argv[i]);
}
return result;
}