Linux/Unix编程

开发平台：
Unix_Linux

addRamDisk.c：源码内容
							#include <stdio.h>
#include <stdlib.h>
#include <netinet/in.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <string.h>
#define ElfHeaderSize  (64 * 1024)
#define ElfPages  (ElfHeaderSize / 4096)
#define KERNELBASE (0xc000000000000000)
void get4k(FILE *file, char *buf )
{
	unsigned j;
	unsigned num = fread(buf, 1, 4096, file);
	for ( j=num; j<4096; ++j )
		buf[j] = 0;
}
void put4k(FILE *file, char *buf )
{
	fwrite(buf, 1, 4096, file);
}
void death(const char *msg, FILE *fdesc, const char *fname) 
{
	fprintf(stderr, msg);
	fclose(fdesc);
	unlink(fname);
	exit(1);
}
int main(int argc, char **argv)
{
	char inbuf[4096];
	FILE *ramDisk = NULL;
	FILE *sysmap = NULL;
	FILE *inputVmlinux = NULL;
	FILE *outputVmlinux = NULL;
  
	unsigned i = 0;
	unsigned long ramFileLen = 0;
	unsigned long ramLen = 0;
	unsigned long roundR = 0;
  
	unsigned long sysmapFileLen = 0;
	unsigned long sysmapLen = 0;
	unsigned long sysmapPages = 0;
	char* ptr_end = NULL; 
	unsigned long offset_end = 0;
	unsigned long kernelLen = 0;
	unsigned long actualKernelLen = 0;
	unsigned long round = 0;
	unsigned long roundedKernelLen = 0;
	unsigned long ramStartOffs = 0;
	unsigned long ramPages = 0;
	unsigned long roundedKernelPages = 0;
	unsigned long hvReleaseData = 0;
	u_int32_t eyeCatcher = 0xc8a5d9c4;
	unsigned long naca = 0;
	unsigned long xRamDisk = 0;
	unsigned long xRamDiskSize = 0;
	long padPages = 0;
  
  
	if (argc < 2) {
		fprintf(stderr, "Name of RAM disk file missing.n");
		exit(1);
	}
	if (argc < 3) {
		fprintf(stderr, "Name of System Map input file is missing.n");
		exit(1);
	}
  
	if (argc < 4) {
		fprintf(stderr, "Name of vmlinux file missing.n");
		exit(1);
	}
	if (argc < 5) {
		fprintf(stderr, "Name of vmlinux output file missing.n");
		exit(1);
	}
	ramDisk = fopen(argv[1], "r");
	if ( ! ramDisk ) {
		fprintf(stderr, "RAM disk file "%s" failed to open.n", argv[1]);
		exit(1);
	}
	sysmap = fopen(argv[2], "r");
	if ( ! sysmap ) {
		fprintf(stderr, "System Map file "%s" failed to open.n", argv[2]);
		exit(1);
	}
  
	inputVmlinux = fopen(argv[3], "r");
	if ( ! inputVmlinux ) {
		fprintf(stderr, "vmlinux file "%s" failed to open.n", argv[3]);
		exit(1);
	}
  
	outputVmlinux = fopen(argv[4], "w+");
	if ( ! outputVmlinux ) {
		fprintf(stderr, "output vmlinux file "%s" failed to open.n", argv[4]);
		exit(1);
	}
  
  
  
	/* Input Vmlinux file */
	fseek(inputVmlinux, 0, SEEK_END);
	kernelLen = ftell(inputVmlinux);
	fseek(inputVmlinux, 0, SEEK_SET);
	printf("kernel file size = %dn", kernelLen);
	if ( kernelLen == 0 ) {
		fprintf(stderr, "You must have a linux kernel specified as argv[3]n");
		exit(1);
	}
	actualKernelLen = kernelLen - ElfHeaderSize;
	printf("actual kernel length (minus ELF header) = %dn", actualKernelLen);
	round = actualKernelLen % 4096;
	roundedKernelLen = actualKernelLen;
	if ( round )
		roundedKernelLen += (4096 - round);
	printf("Vmlinux length rounded up to a 4k multiple = %ld/0x%lx n", roundedKernelLen, roundedKernelLen);
	roundedKernelPages = roundedKernelLen / 4096;
	printf("Vmlinux pages to copy = %ld/0x%lx n", roundedKernelPages, roundedKernelPages);
	/* Input System Map file */
	/* (needs to be processed simply to determine if we need to add pad pages due to the static variables not being included in the vmlinux) */
	fseek(sysmap, 0, SEEK_END);
	sysmapFileLen = ftell(sysmap);
	fseek(sysmap, 0, SEEK_SET);
	printf("%s file size = %ld/0x%lx n", argv[2], sysmapFileLen, sysmapFileLen);
	sysmapLen = sysmapFileLen;
	roundR = 4096 - (sysmapLen % 4096);
	if (roundR) {
		printf("Rounding System Map file up to a multiple of 4096, adding %ld/0x%lx n", roundR, roundR);
		sysmapLen += roundR;
	}
	printf("Rounded System Map size is %ld/0x%lx n", sysmapLen, sysmapLen);
  
	/* Process the Sysmap file to determine where _end is */
	sysmapPages = sysmapLen / 4096;
	for (i=0; i<sysmapPages; ++i) {
		get4k(sysmap, inbuf);
	}
	/* search for _end in the last page of the system map */
	ptr_end = strstr(inbuf, " _end");
	if (!ptr_end) {
		fprintf(stderr, "Unable to find _end in the sysmap file n");
		fprintf(stderr, "inbuf: n");
		fprintf(stderr, "%s n", inbuf);
		exit(1);
	}
	printf("Found _end in the last page of the sysmap - backing up 10 characters it looks like %s", ptr_end-10);
	/* convert address of _end in system map to hex offset. */
	offset_end = (unsigned int)strtol(ptr_end-10, NULL, 16);
	/* calc how many pages we need to insert between the vmlinux and the start of the ram disk */
	padPages = offset_end/4096 - roundedKernelPages;
	/* Check and see if the vmlinux is already larger than _end in System.map */
	if (padPages < 0) {
		/* vmlinux is larger than _end - adjust the offset to the start of the embedded ram disk */ 
		offset_end = roundedKernelLen;
		printf("vmlinux is larger than _end indicates it needs to be - offset_end = %lx n", offset_end);
		padPages = 0;
		printf("will insert %lx pages between the vmlinux and the start of the ram disk n", padPages);
	}
	else {
		/* _end is larger than vmlinux - use the offset to _end that we calculated from the system map */
		printf("vmlinux is smaller than _end indicates is needed - offset_end = %lx n", offset_end);
		printf("will insert %lx pages between the vmlinux and the start of the ram disk n", padPages);
	}
	/* Input Ram Disk file */
	// Set the offset that the ram disk will be started at.
	ramStartOffs = offset_end;  /* determined from the input vmlinux file and the system map */
	printf("Ram Disk will start at offset = 0x%lx n", ramStartOffs);
  
	fseek(ramDisk, 0, SEEK_END);
	ramFileLen = ftell(ramDisk);
	fseek(ramDisk, 0, SEEK_SET);
	printf("%s file size = %ld/0x%lx n", argv[1], ramFileLen, ramFileLen);
	ramLen = ramFileLen;
	roundR = 4096 - (ramLen % 4096);
	if ( roundR ) {
		printf("Rounding RAM disk file up to a multiple of 4096, adding %ld/0x%lx n", roundR, roundR);
		ramLen += roundR;
	}
	printf("Rounded RAM disk size is %ld/0x%lx n", ramLen, ramLen);
	ramPages = ramLen / 4096;
	printf("RAM disk pages to copy = %ld/0x%lxn", ramPages, ramPages);
  // Copy 64K ELF header
	for (i=0; i<(ElfPages); ++i) {
		get4k( inputVmlinux, inbuf );
		put4k( outputVmlinux, inbuf );
	}
	/* Copy the vmlinux (as full pages). */
	fseek(inputVmlinux, ElfHeaderSize, SEEK_SET);
	for ( i=0; i<roundedKernelPages; ++i ) {
		get4k( inputVmlinux, inbuf );
		put4k( outputVmlinux, inbuf );
	}
  
	/* Insert pad pages (if appropriate) that are needed between */
	/* | the end of the vmlinux and the ram disk. */
	for (i=0; i<padPages; ++i) {
		memset(inbuf, 0, 4096);
		put4k(outputVmlinux, inbuf);
	}
	/* Copy the ram disk (as full pages). */
	for ( i=0; i<ramPages; ++i ) {
		get4k( ramDisk, inbuf );
		put4k( outputVmlinux, inbuf );
	}
	/* Close the input files */
	fclose(ramDisk);
	fclose(inputVmlinux);
	/* And flush the written output file */
	fflush(outputVmlinux);
	/* Fixup the new vmlinux to contain the ram disk starting offset (xRamDisk) and the ram disk size (xRamDiskSize) */
	/* fseek to the hvReleaseData pointer */
	fseek(outputVmlinux, ElfHeaderSize + 0x24, SEEK_SET);
	if (fread(&hvReleaseData, 4, 1, outputVmlinux) != 1) {
		death("Could not read hvReleaseData pointern", outputVmlinux, argv[4]);
	}
	hvReleaseData = ntohl(hvReleaseData); /* Convert to native int */
	printf("hvReleaseData is at %08xn", hvReleaseData);
	/* fseek to the hvReleaseData */
	fseek(outputVmlinux, ElfHeaderSize + hvReleaseData, SEEK_SET);
	if (fread(inbuf, 0x40, 1, outputVmlinux) != 1) {
		death("Could not read hvReleaseDatan", outputVmlinux, argv[4]);
	}
	/* Check hvReleaseData sanity */
	if (memcmp(inbuf, &eyeCatcher, 4) != 0) {
		death("hvReleaseData is invalidn", outputVmlinux, argv[4]);
	}
	/* Get the naca pointer */
	naca = ntohl(*((u_int32_t*) &inbuf[0x0C])) - KERNELBASE;
	printf("Naca is at offset 0x%lx n", naca);
	/* fseek to the naca */
	fseek(outputVmlinux, ElfHeaderSize + naca, SEEK_SET);
	if (fread(inbuf, 0x18, 1, outputVmlinux) != 1) {
		death("Could not read nacan", outputVmlinux, argv[4]);
	}
	xRamDisk = ntohl(*((u_int32_t *) &inbuf[0x0c]));
	xRamDiskSize = ntohl(*((u_int32_t *) &inbuf[0x14]));
	/* Make sure a RAM disk isn't already present */
	if ((xRamDisk != 0) || (xRamDiskSize != 0)) {
		death("RAM disk is already attached to this kerneln", outputVmlinux, argv[4]);
	}
	/* Fill in the values */
	*((u_int32_t *) &inbuf[0x0c]) = htonl(ramStartOffs);
	*((u_int32_t *) &inbuf[0x14]) = htonl(ramPages);
	/* Write out the new naca */
	fflush(outputVmlinux);
	fseek(outputVmlinux, ElfHeaderSize + naca, SEEK_SET);
	if (fwrite(inbuf, 0x18, 1, outputVmlinux) != 1) {
		death("Could not write nacan", outputVmlinux, argv[4]);
	}
	printf("Ram Disk of 0x%lx pages is attached to the kernel at offset 0x%08xn",
	       ramPages, ramStartOffs);
	/* Done */
	fclose(outputVmlinux);
	/* Set permission to executable */
	chmod(argv[4], S_IRUSR|S_IWUSR|S_IXUSR|S_IRGRP|S_IXGRP|S_IROTH|S_IXOTH);
	return 0;
}

						