dnw.c

linux 下的dnw！源代码 速度很快

#include <stdio.h>
#include <stdlib.h>
#include <malloc.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <fcntl.h>

const char* dev = "/dev/secbulk0";


int main(int argc, char* argv[])
{
	unsigned char* file_buffer = NULL;
	if( 2 != argc )
	{
		printf("Usage: dwn <filename>\n");
		return 1;
	}

	int fd = open(argv[1], O_RDONLY);
	if(-1 == fd)
	{
		printf("Can not open file - %s\n", argv[1]);
		return 1;
	}

	struct stat file_stat;
	if( -1 == fstat(fd, &file_stat) )
	{
		printf("Get file size filed!\n");
		return 1;
	}	
	
	file_buffer = (char*)malloc(file_stat.st_size+10);
	if(NULL == file_buffer)
	{
		printf("malloc failed!\n");
		goto error;
	}
	if( file_stat.st_size !=  read(fd, file_buffer+8, file_stat.st_size))
	{
		printf("Read file failed!\n");
		goto error;
	}

	printf("file name : %s\n", argv[1]);
	printf("file size : %d bytes\n", file_stat.st_size);

	int fd_dev = open(dev, O_WRONLY);
	if( -1 == fd_dev)
	{
		printf("Can not open %s\n", dev);
		goto error;
	}
	
	*((unsigned long*)file_buffer) = 0x32000000; 	//load address
	*((unsigned long*)file_buffer+1) = file_stat.st_size+10;	//file size
	unsigned short sum = 0;
	int i;
	for(i=8; i<file_stat.st_size+8; i++)
	{
		sum += file_buffer[i];
	}
	
	printf("Writing data...\n");
	size_t remain_size = file_stat.st_size+10;
	size_t block_size = remain_size / 100;
	size_t writed = 0;
	while(remain_size>0)
	{
		size_t to_write = remain_size > block_size ? block_size:remain_size;
		if( to_write != write(fd_dev, file_buffer+writed, to_write))
		{
			printf("failed!\n");
			return 1;
		}
		remain_size -= to_write;
		writed += to_write;
		printf("\r%d%\t %d bytes     ", writed*100/(file_stat.st_size+10), writed);
		fflush(stdout);
		
	}	

	printf("OK\n");
	return 0;

error:
	if(-1!=fd_dev) close(fd_dev);
	if(fd != -1) close(fd);
	if( NULL!=file_buffer )
		free(file_buffer);
	return -1;
}