From: Ben Pfaff <blp@cs.stanford.edu>
Date: Tue, 14 Sep 2004 06:53:02 +0000 (+0000)
Subject: first try
X-Git-Url: https://pintos-os.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=8fc448c07982e2f706487356555688e2ed91f046;p=pintos-anon

first try
---

diff --git a/src/utils/pintos b/src/utils/pintos
new file mode 100755
index 0000000..3cd9fef
--- /dev/null
+++ b/src/utils/pintos
@@ -0,0 +1,146 @@
+#! /usr/bin/perl
+
+$sim = "bochs";
+while (@ARGV) {
+    my ($arg) = shift (@ARGV);
+    if ($arg eq '--bochs' || $arg eq '--bochs-dbg' || $arg eq '--bochs-gdb'
+	|| $arg eq '--qemu' || $arg eq '--vmware') {
+	$sim = substr ($arg, 2);
+    } elsif ($arg eq 'run') {
+	run_vm (@ARGV);
+	exit 0;
+    } elsif ($arg eq 'make-disk') {
+	usage () if @ARGV != 2;
+	my ($file, $kb) = @ARGV;
+	usage () if $kb =~ /^\d+$/;
+	die "$file: already exists\n" if -e $file;
+
+	create_disk ($file, $kb);
+	exit 0;
+    } elsif ($arg eq 'put') {
+	usage () if @ARGV != 1 && @ARGV != 2;
+	my ($hostfn, $guestfn) = @ARGV;
+	$guestfn = $hostfn if !defined $guestfn;
+
+	# Create scratch disk from file.
+	die "$hostfn: $!\n" if ! -e $hostfn;
+	my ($size) = -s _;
+	copy_pad ($hostfn, "scratch.dsk", 512);
+
+	# Do copy.
+	run_vm ("-ci", $hostfn, $size, "-q");
+	exit 0;
+    } elsif ($arg eq 'get') {
+	usage () if @ARGV != 1 && @ARGV != 2;
+	my ($guestfn, $hostfn) = @ARGV;
+	$hostfn = $guestfn if !defined $hostfn;
+	die "$hostfn: already exists\n" if -e $file;
+
+	# Create scratch disk big enough for any file in the filesystem
+	# (modulo sparse files).
+	die "fs.dsk: $!\n" if ! -e "fs.dsk";
+	my ($fs_size) = -s _;
+	my ($scratch_size) = -s "scratch.dsk";
+	$scratch_size = 0 if !defined $scratch_size;
+	create_disk ("scratch.dsk", $fs_size / 1024 + 16)
+	    if $scratch_size < $fs_size + 16384;
+
+	# Do copy.
+	run_vm ("-co", $guestfn, $hostfn, "-q");
+
+	# Read out scratch disk.
+	open (SRC, "<scratch.dsk") or die "scratch.dsk: open: $!\n";
+	open (DST, ">$hostfn") or die "$hostfn: create: $!\n";
+	my ($input);
+	read (SRC, $input, 512) == 512 or die "scratch.dsk: read error\n";
+	my ($size) = unpack ("%V", $input);
+	$size != 0xffffffff or die "$guestfn: too big for scratch.dsk?";
+	read (SRC, $src, $size) == $src or die "scratch.dsk: read error\n";
+	print DST $src or die "$hostfn: write error\n";
+	close (DST);
+	close (SRC);
+
+	exit 0;
+    }
+}
+usage ();
+
+sub usage {
+    my ($exitcode) = @_;
+    $exitcode = 1 unless defined $exitcode;
+    print "pintos, a utility for invoking Pintos in a simulator\n";
+    print "Usage: pintos [OPTION...] COMMAND [ARG...]\n";
+    print "where COMMAND is one of the following:\n";
+    print "  run [CMDLINE...]        run a VM in the simulator\n";
+    print "  make-disk FILE.DSK KB   create FILE.DSK as empty KB-kB disk\n";
+    print "  put HOSTFN [GUESTFN]    copy HOSTFN into VM (as GUESTFN)\n";
+    print "  get GUESTFN [HOSTFN]    copy GUESTFN out of VM (to HOSTFN)\n";
+    print "  help                    print this help message and exit\n";
+    print "Available options:\n";
+    print "  --bochs       Use Bochs as simulator (default)\n";
+    print "  --bochs-dbg   Use Bochs with debugger as simulator\n";
+    print "  --bochs-gdb   Use Bochs with gdb as simulator\n";
+    print "  --qemu        Use qemu as simulator\n";
+    print "  --vmware      Use VMware Workstation as simulator\n";
+    exit $exitcode;
+}
+
+sub copy_pad {
+    my ($src, $dst, $blocksize) = @_;
+    open (SRC, "<$src") or die "$src: open: $!\n";
+    open (DST, ">$dst") or die "$dst: create: $!\n";
+    my ($size) = 0;
+    my ($input);
+    while (read (SRC, $input, 4096)) {
+	$size += length ($input);
+	print DST $input;
+    }
+    my ($rem) = $size % $blocksize;
+    print DST "\0" x ($blocksize - $rem) if $rem != 0;
+    close (DST);
+    close (SRC);
+}
+
+sub create_disk {
+    my ($disk, $kb) = @_;
+    open (DISK, ">$disk") or die "$disk: create: $!\n";
+    for (my ($i) = 0; $i < $kb; $i++) {
+	print DISK "\0" x 1024;
+    }
+    close (DISK);
+}
+
+sub run_vm {
+    my (@disks) = (undef, undef, undef, undef);
+
+    $disks[0] = "os.dsk";
+    $disks[1] = "fs.dsk" if -e "fs.dsk";
+    $disks[2] = "scratch.dsk" if -e "scratch.dsk";
+    $disks[3] = "swap.dsk" if -e "swap.dsk";
+
+    die "$disks[0]: can't find OS disk\n" if ! -e $disks[0];
+    write_cmd_line ($disks[0], @_);
+
+    if ($sim eq 'qemu') {
+	my (@cmd) = ('qemu');
+	push (@cmd, '-hda', $disks[0]) if defined $disks[0];
+	push (@cmd, '-hdb', $disks[1]) if defined $disks[1];
+	push (@cmd, '-hdc', $disks[2]) if defined $disks[2];
+	push (@cmd, '-hdd', $disks[3]) if defined $disks[3];
+	system (@cmd);
+    }
+}
+
+sub write_cmd_line {
+    my ($disk, @args) = @_;
+
+    die "command line includes empty string" if grep (/^$/, @args);
+    $args = join ("\0", @args) . "\0\0";
+    die "command line exceeds 128 bytes" if length ($args) > 128;
+    $args .= "\0" x (128 - length ($args));
+
+    open (DISK, "+<$disk") or die "$disk: open: $!\n";
+    seek (DISK, 0x17e, 0) or die "$disk: seek: $!\n";
+    syswrite (DISK, $args) or die "$disk: write: $!\n";
+    close (DISK) or die "$disk: close: $!\n";
+}