mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
469 lines
12 KiB
469 lines
12 KiB
/*++
|
|
|
|
Copyright (c) 1994-1996 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
setnvram.c
|
|
|
|
Abstract:
|
|
|
|
This program is an example of how you could use a text file to create
|
|
input for nvram.exe.
|
|
|
|
Author:
|
|
|
|
Chuck Lenzmeier (chuckl)
|
|
|
|
Revision History:
|
|
|
|
--*/
|
|
|
|
//
|
|
// setnvram.c
|
|
//
|
|
// This program is an example of
|
|
|
|
#define _DLL 1
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#define SEPARATOR "|"
|
|
|
|
#define MAXLINESIZE 256
|
|
|
|
#define FALSE 0
|
|
#define TRUE 1
|
|
|
|
char Line[MAXLINESIZE];
|
|
|
|
char Countdown[MAXLINESIZE];
|
|
|
|
char LoadIdentifier[MAXLINESIZE];
|
|
char SystemPartition[MAXLINESIZE];
|
|
char OsLoader[MAXLINESIZE];
|
|
char OsLoadPartition[MAXLINESIZE];
|
|
char OsLoadFilename[MAXLINESIZE];
|
|
char OsLoadOptions[MAXLINESIZE];
|
|
|
|
char DefaultSystemPartition[MAXLINESIZE];
|
|
char DefaultOsLoadPartition[MAXLINESIZE];
|
|
char DefaultOsLoadOptions[MAXLINESIZE];
|
|
|
|
char *
|
|
Trim (
|
|
char *String
|
|
)
|
|
{
|
|
char *start;
|
|
char *end;
|
|
|
|
start = String;
|
|
while ( (*start == ' ') || (*start == '\t') ) {
|
|
start++;
|
|
}
|
|
|
|
end = strrchr( start, 0 ) - 1;
|
|
if ( (end > start) && ((*end == ' ') || (*end == '\t')) ) {
|
|
do {
|
|
end--;
|
|
} while ( (*end == ' ') || (*end == '\t') );
|
|
end++;
|
|
*end = 0;
|
|
}
|
|
|
|
return start;
|
|
}
|
|
|
|
int
|
|
ParsePartition (
|
|
char *String,
|
|
char *Partition
|
|
)
|
|
{
|
|
char buffer[MAXLINESIZE];
|
|
char *multi;
|
|
char *scsi;
|
|
char *disk;
|
|
char *part;
|
|
char *dot;
|
|
|
|
strcpy( buffer, String );
|
|
|
|
if ( _strnicmp(buffer, "scsi.", 5) != 0 ) {
|
|
return FALSE;
|
|
}
|
|
multi = "0";
|
|
scsi = "0";
|
|
disk = &buffer[5];
|
|
dot = strchr( disk, '.' );
|
|
if ( dot == NULL ) {
|
|
return FALSE;
|
|
}
|
|
*dot = 0;
|
|
part = dot + 1;
|
|
dot = strchr( part, '.' );
|
|
if ( dot != NULL ) {
|
|
scsi = disk;
|
|
disk = part;
|
|
*dot = 0;
|
|
part = dot + 1;
|
|
dot = strchr( part, '.' );
|
|
if ( dot != NULL ) {
|
|
multi = scsi;
|
|
scsi = disk;
|
|
disk = part;
|
|
*dot = 0;
|
|
part = dot + 1;
|
|
}
|
|
}
|
|
|
|
#if !defined(_PPC_)
|
|
strcpy( Partition, "scsi()disk(" );
|
|
#else
|
|
strcpy( Partition, "multi(" );
|
|
strcat( Partition, multi );
|
|
strcat( Partition, ")scsi(" );
|
|
strcat( Partition, scsi );
|
|
strcat( Partition, ")disk(" );
|
|
#endif
|
|
strcat( Partition, disk );
|
|
#if !defined(_PPC_)
|
|
strcat( Partition, ")rdisk()partition(" );
|
|
#else
|
|
strcat( Partition, ")rdisk(0)partition(" );
|
|
#endif
|
|
strcat( Partition, part );
|
|
strcat( Partition, ")" );
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
int
|
|
main (
|
|
int argc,
|
|
char *argv[]
|
|
)
|
|
{
|
|
FILE *file = stdin;
|
|
|
|
char *build;
|
|
int len;
|
|
int linenum;
|
|
|
|
char *ident;
|
|
char *token;
|
|
char *sysdir;
|
|
char *osdir;
|
|
char *options;
|
|
char *syspart;
|
|
char *ospart;
|
|
char *loader;
|
|
|
|
char options1[MAXLINESIZE];
|
|
char syspart1[MAXLINESIZE];
|
|
char ospart1[MAXLINESIZE];
|
|
|
|
if ( argc > 1 ) {
|
|
#if 1
|
|
if ( argc > 2 ) {
|
|
#endif
|
|
fprintf( stderr, "This program accepts no arguments\n" );
|
|
fprintf( stderr, "Redirect stdin to build data file\n" );
|
|
fprintf( stderr, "Redirect stdout to nvram.exe input file\n" );
|
|
return 1;
|
|
#if 1
|
|
} else {
|
|
file = fopen( argv[1], "r" );
|
|
if ( file == NULL ) {
|
|
fprintf( stderr, "Can't open input file %s\n", argv[1] );
|
|
return 1;
|
|
}
|
|
}
|
|
#endif
|
|
}
|
|
|
|
Countdown[0] = 0;
|
|
|
|
LoadIdentifier[0] = 0;
|
|
SystemPartition[0] = 0;
|
|
OsLoader[0] = 0;
|
|
OsLoadPartition[0] = 0;
|
|
OsLoadFilename[0] = 0;
|
|
OsLoadOptions[0] = 0;
|
|
|
|
DefaultOsLoadOptions[0] = 0;
|
|
DefaultOsLoadPartition[0] = 0;
|
|
DefaultSystemPartition[0] = 0;
|
|
|
|
linenum = 0;
|
|
|
|
while ( TRUE ) {
|
|
|
|
//
|
|
// Get the next line from the input stream.
|
|
//
|
|
|
|
linenum++;
|
|
|
|
build = fgets( Line, MAXLINESIZE, file );
|
|
if ( build == NULL ) {
|
|
if ( feof(file) ) break;
|
|
fprintf( stderr, "Error %d reading input at line %d\n", ferror(file), linenum );
|
|
return ferror(file);
|
|
}
|
|
|
|
build = Trim( build );
|
|
len = strlen( build );
|
|
|
|
//
|
|
// Ignore blank lines and lines that start with //.
|
|
//
|
|
|
|
if ( len == 0 ) continue;
|
|
if ( (build[0] == '/') && (build[1] == '/') ) continue;
|
|
if ( build[len-1] != '\n' ) {
|
|
fprintf( stderr, "Line %d is too long; %d characters max\n", linenum, MAXLINESIZE-2 );
|
|
return 1;
|
|
}
|
|
if ( len == 1 ) continue;
|
|
build[len-1] = 0;
|
|
|
|
//
|
|
// Check for the special "countdown" line. If found, save the countdown value.
|
|
//
|
|
|
|
if ( strstr(build,"countdown=") == build ) {
|
|
strcpy( Countdown, strchr(build,'=') + 1 );
|
|
continue;
|
|
}
|
|
|
|
//
|
|
// Check for the special "default systempartition" line. If found, save the
|
|
// default string.
|
|
//
|
|
|
|
if ( strstr(build,"default systempartition=") == build ) {
|
|
strcpy( DefaultSystemPartition, Trim( strchr(build,'=') + 1 ) );
|
|
continue;
|
|
}
|
|
|
|
//
|
|
// Check for the special "default osloadpartition" line. If found, save the
|
|
// default string.
|
|
//
|
|
|
|
if ( strstr(build,"default osloadpartition=") == build ) {
|
|
strcpy( DefaultOsLoadPartition, Trim( strchr(build,'=') + 1 ) );
|
|
continue;
|
|
}
|
|
|
|
//
|
|
// Check for the special "default options" line. If found, save the
|
|
// default string.
|
|
//
|
|
|
|
if ( strstr(build,"default options=") == build ) {
|
|
strcpy( DefaultOsLoadOptions, Trim( strchr(build,'=') + 1 ) );
|
|
strcat( DefaultOsLoadOptions, " " );
|
|
continue;
|
|
}
|
|
|
|
//
|
|
// OK, we should have an OS load line. Required format is:
|
|
//
|
|
// <ident>[|<sys-dir>][|<os-dir>][<dir>][|<options>][|<sys-part>][|<os-part>][|<loader>]
|
|
//
|
|
// Everything after <ident> is optional and may be specified in any order.
|
|
//
|
|
// <sys-dir> defines the directory path to the osloader/hal directory.
|
|
// <os-dir> defines the directory path to the OS directory.
|
|
// The default value for both of these fields is <ident>.
|
|
// <dir> sets both <sys-dir> and <os-dir>.
|
|
//
|
|
// <sys-part> and <os-part> are optional only if the corresponding defaults
|
|
// have been specified.
|
|
//
|
|
// <loader> is used to override the selection of osloader.exe as the OS loader.
|
|
//
|
|
// <sys-dir> format is sysdir=<directory path (no leading \)>
|
|
// <os-dir> format is osdir=<directory path (no leading \)>
|
|
// <dir> format is dir=<directory path (no leading \)>
|
|
// <options> format is options=<text of options>
|
|
// <sys-part> format is syspart=<partition specification>
|
|
// <os-part> format is ospart=<partition specification>
|
|
// <loader> format is loader=<filename>
|
|
//
|
|
|
|
//
|
|
// Get the load-identifier.
|
|
//
|
|
|
|
ident = Trim( strtok( build, SEPARATOR ) );
|
|
|
|
//
|
|
// Set defaults for optional fields.
|
|
//
|
|
|
|
osdir = ident;
|
|
sysdir = ident;
|
|
options = DefaultOsLoadOptions;
|
|
syspart = DefaultSystemPartition;
|
|
ospart = DefaultOsLoadPartition;
|
|
loader = "osloader.exe";
|
|
|
|
//
|
|
// Get optional fields.
|
|
//
|
|
|
|
while ( (token = strtok( NULL, SEPARATOR )) != NULL ) {
|
|
|
|
token = Trim( token );
|
|
|
|
if ( strstr(token,"sysdir=") == token ) {
|
|
|
|
sysdir = Trim( strchr(token,'=') + 1 );
|
|
|
|
} else if ( strstr(token,"osdir=") == token ) {
|
|
|
|
osdir = Trim( strchr(token,'=') + 1 );
|
|
|
|
} else if ( strstr(token,"dir=") == token ) {
|
|
|
|
sysdir = Trim( strchr(token,'=') + 1 );
|
|
osdir = sysdir;
|
|
|
|
} else if ( strstr(token,"options=") == token ) {
|
|
|
|
//
|
|
// If the options do not start with "nodef", preface the
|
|
// default options (if any) to the specified options.
|
|
//
|
|
|
|
options = Trim( strchr(token,'=') + 1 );
|
|
if ( _strnicmp(options,"nodef",5) == 0 ) {
|
|
options = options+5;
|
|
} else {
|
|
strcpy( options1, DefaultOsLoadOptions );
|
|
strcat( options1, options );
|
|
options = options1;
|
|
}
|
|
|
|
} else if ( strstr(token,"syspart=") == token ) {
|
|
|
|
syspart = Trim( strchr(token,'=') + 1 );
|
|
|
|
} else if ( strstr(token,"ospart=") == token ) {
|
|
|
|
ospart = Trim( strchr(token,'=') + 1 );
|
|
|
|
} else if ( strstr(token,"loader=") == token ) {
|
|
|
|
loader = Trim( strchr(token,'=') + 1 );
|
|
|
|
} else {
|
|
|
|
//
|
|
// Unrecognized optional field.
|
|
//
|
|
|
|
fprintf( stderr, "Unreconized optional field at line %d\n", linenum );
|
|
return 1;
|
|
|
|
}
|
|
|
|
} // while
|
|
|
|
//
|
|
// Verify the validity of the input fields.
|
|
//
|
|
|
|
if ( strlen(ident) == 0 ) {
|
|
fprintf( stderr, "Bad <load-identifier> at line %d\n", linenum );
|
|
return 1;
|
|
}
|
|
if ( strlen(sysdir) == 0 ) {
|
|
fprintf( stderr, "Bad <system-directory> at line %d\n", linenum );
|
|
return 1;
|
|
}
|
|
if ( strlen(osdir) == 0 ) {
|
|
fprintf( stderr, "Bad <os-directory> at line %d\n", linenum );
|
|
return 1;
|
|
}
|
|
if ( strlen(syspart) == 0 ) {
|
|
fprintf( stderr, "Missing <system-partition> (no default) at line %d\n", linenum );
|
|
return 1;
|
|
}
|
|
if ( strlen(ospart) == 0 ) {
|
|
fprintf( stderr, "Missing <os-partition> (no default) at line %d\n", linenum );
|
|
return 1;
|
|
}
|
|
if ( !ParsePartition(syspart, syspart1) ) {
|
|
fprintf( stderr, "Bad <system-partition> at line %d\n", linenum );
|
|
return 1;
|
|
}
|
|
if ( !ParsePartition(ospart, ospart1) ) {
|
|
fprintf( stderr, "Bad <os-partition> at line %d\n", linenum );
|
|
return 1;
|
|
}
|
|
if ( strlen(loader) == 0 ) {
|
|
fprintf( stderr, "Bad <loader> at line %d\n", linenum );
|
|
return 1;
|
|
}
|
|
|
|
//
|
|
// If this is not the first load line, append ';' to all of the NVRAM strings.
|
|
//
|
|
|
|
if ( strlen(LoadIdentifier) != 0 ) {
|
|
strcat( LoadIdentifier, ";" );
|
|
strcat( SystemPartition, ";" );
|
|
strcat( OsLoader, ";" );
|
|
strcat( OsLoadPartition, ";" );
|
|
strcat( OsLoadFilename, ";" );
|
|
strcat( OsLoadOptions, ";" );
|
|
}
|
|
|
|
//
|
|
// Append this load line to the NVRAM strings.
|
|
//
|
|
|
|
strcat( LoadIdentifier, ident );
|
|
|
|
strcat( SystemPartition, syspart1 );
|
|
|
|
strcat( OsLoader, syspart1 );
|
|
if ( loader[0] != '\\' ) {
|
|
strcat( OsLoader, "\\" );
|
|
strcat( OsLoader, sysdir );
|
|
strcat( OsLoader, "\\" );
|
|
}
|
|
strcat( OsLoader, loader );
|
|
|
|
strcat( OsLoadPartition, ospart1 );
|
|
|
|
strcat( OsLoadFilename, "\\" );
|
|
strcat( OsLoadFilename, osdir );
|
|
|
|
strcat( OsLoadOptions, options );
|
|
Trim( OsLoadOptions );
|
|
|
|
}
|
|
|
|
//
|
|
// Write the necessary nvram.exe commands to the output stream.
|
|
//
|
|
|
|
if ( Countdown[0] != 0 ) {
|
|
fprintf( stdout, "nvram /set COUNTDOWN = \"%s\"\n", Countdown );
|
|
}
|
|
fprintf( stdout, "nvram /set LOADIDENTIFIER = \"%s\"\n", LoadIdentifier );
|
|
fprintf( stdout, "nvram /set SYSTEMPARTITION = \"%s\"\n", SystemPartition );
|
|
fprintf( stdout, "nvram /set OSLOADER = \"%s\"\n", OsLoader );
|
|
fprintf( stdout, "nvram /set OSLOADPARTITION = \"%s\"\n", OsLoadPartition );
|
|
fprintf( stdout, "nvram /set OSLOADFILENAME = \"%s\"\n", OsLoadFilename );
|
|
fprintf( stdout, "nvram /set OSLOADOPTIONS = \"%s\"\n", OsLoadOptions );
|
|
|
|
return 0;
|
|
}
|