pad.c

来自「一个收集所有最基本功能的函数库；所有的函数都是尽量短小和简单 使用 doxyge」· C语言 代码 · 共 71 行

/***************************************************************************
 *            pad.c
 *
 *  Mon May 21 18:07:35 2007
 *  Copyright  2007  kf701
 *  Email <kf701.ye AT gmail.com>
 ****************************************************************************/

/*
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 */
 


#include <sys/types.h>
#include <sys/stat.h>

#include "kf701.h"

/**
 * @brief padding a file align ALIGN_SIZE * N size
 * @param filein will be aligned
 * @param fileout output file
 * @param align_size align size unit
 * @retval true success
 * @retval false faild
 */
bool pad_file(char *filein, char *fileout, uint32_t align_size)
{
	if( NULL == filein || NULL == fileout )
		return false;

	FILE* fp = fopen(filein, "r");
	if(!fp)
		return false;

	struct stat fst;
	int ext_size;
	stat(filein, &fst);

	if( 0 == fst.st_size % align_size )
		ext_size = fst.st_size;
	else
		ext_size = (fst.st_size / align_size + 1) * align_size;

	uint8_t buf[ext_size];
	fread( buf, 1, fst.st_size, fp );
	fclose(fp);

	if( ext_size != fst.st_size )
		memset( buf + fst.st_size, 0, ext_size - fst.st_size );

	fp = fopen(fileout, "w+");
	fwrite( buf, 1, ext_size, fp );
	fclose(fp);

	return true;
}