Allow to read current command line

This commit is contained in:
Laurent Vivier 2004-05-21 10:42:54 +00:00
parent e5764a7b71
commit 21a346e3f0

View File

@ -9,6 +9,7 @@
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <stdlib.h>
#include "emile-first.h"
@ -17,9 +18,11 @@
static void usage(int argc, char** argv)
{
fprintf(stderr, "Usage: %s <image> <cmdline>\n", argv[0]);
fprintf(stderr, "Usage: %s -r <image>\n", argv[0]);
fprintf(stderr, "\n Allows to set the kernel command line <cmdline>\n");
fprintf(stderr, " into the floppy image <image>\n");
fprintf(stderr, " <image> can be a file or a device (/dev/fd0)\n");
fprintf(stderr, " with \"-r\" flag, display current command line\n");
fprintf(stderr, "\n Examples:\n");
fprintf(stderr, "\n To set root filesystem on disk 1 partition 4\n");
fprintf(stderr, "\n %s floppy.img \"root=/dev/sda4\"\n", argv[0]);
@ -32,17 +35,21 @@ static void usage(int argc, char** argv)
*/
}
int set_cmdline(char* image, char* cmdline)
int set_cmdline(int readonly, char* image, char* cmdline)
{
int fd;
short buffer_size;
int ret;
int len = strlen(cmdline) + 1;
int len;
if (readonly)
fd = open(image, O_RDONLY);
else
fd = open(image, O_RDWR);
if (fd == -1)
{
perror("Cannot open image file (rw mode)");
perror("Cannot open image file");
return 2;
}
@ -62,6 +69,32 @@ int set_cmdline(char* image, char* cmdline)
return 3;
}
if (readonly)
{
char* buffer;
buffer = (char*)malloc(buffer_size);
if (buffer == NULL)
{
perror("Cannot malloc()");
close(fd);
return 6;
}
ret = read(fd, buffer, buffer_size);
if (ret != buffer_size)
{
perror("Cannot read current command line");
free(buffer);
close(fd);
return 7;
}
printf("Current command line: \"%s\"\n", buffer);
free(buffer);
}
else
{
len = strlen(cmdline) + 1;
if (len > buffer_size)
{
fprintf(stderr, "Command line too long\n");
@ -77,6 +110,9 @@ int set_cmdline(char* image, char* cmdline)
return 5;
}
printf("Command line sucessfully modified\n");
}
close(fd);
return 0;
}
@ -90,9 +126,10 @@ int main(int argc, char** argv)
return 1;
}
ret = set_cmdline(argv[1], argv[2]);
if (ret == 0)
printf("Command line sucessfully modified\n");
if (strcmp(argv[1], "-r") == 0)
ret = set_cmdline(1, argv[2], NULL);
else
ret = set_cmdline(0, argv[1], argv[2]);
return ret;
}