Shell脚本创建指定大小文件的测试数据
投稿：junjie
这篇文章主要介绍了Shell脚本创建指定大小文件的测试数据,本文同时给出使用DD命令和Shell脚本解决这个需求,给出的脚本功能更加强大,需要的朋友可以参考下
我们在测试或调试的时候，有时候会需要生成某个size的文件，比如在测试存储系统时，需要将磁盘剩余空间减少5G，最简单的办法就是拷贝一个5G的文件过来，但是从哪儿去弄这样大小的文件呢，或许你想到随便找一个文件，不停的拷贝，最后合并，这也不失为一种办法，但是有了dd，你会更容易且更灵活的实现。
我们来case by case的介绍dd的用法。先看第一个
生成一个大小为5G的文件，内容不做要求
命令如下
复制代码 代码如下:
$ dd if=/dev/zero of=tmp.5G bs=1G count=5
解释一下这里用到的参数
复制代码 代码如下:
if=FILE : 指定输入文件，若不指定则从标注输入读取。这里指定为/dev/zero是Linux的一个伪文件，它可以产生连续不断的null流（二进制的0）
of=FILE : 指定输出文件，若不指定则输出到标准输出
bs=BYTES : 每次读写的字节数，可以使用单位K、M、G等等。另外输入输出可以分别用ibs、obs指定，若使用bs，则表示是ibs和obs都是用该参数
count=BLOCKS : 读取的block数，block的大小由ibs指定（只针对输入参数）
这样上面生成5G文件的命令就很好理解了，即从/dev/null每次读取1G数据，读5次，写入tmp.5G这个文件
再看下面一个问题
复制代码 代码如下:
将file.in的前1M追加到file.out的末尾
命令如下
复制代码 代码如下:
$ file_out_size=`du -b file.out | awk '{print $1}'`
$ dd if=./file.in ibs=1M count=1 of=./file.out seek=1 obs=$file_out_size
这里ibs和obs设置为了不同的值，和前面的命令相比，只多了一个seek参数
复制代码 代码如下:
seek=BLOCKS : 在拷贝数据之前，从输出文件开头跳过BLOCKS个block，block的大小由obs指定
命令的意思就是从file.in读取1个1M的数据块写入file.out，不过写入位置并不在file.out的开头，而是在1*$file_out_size字节偏移处（也就是文件末尾）
在此基础上再增加一个要求
将file.in的第3M追加到file.out的末尾
复制代码 代码如下:
$ file_out_size=`du -b file.out | awk '{print $1}'`
$ dd if=./file.in skip=2 ibs=1M count=1 of=./file.out seek=1 obs=$file_out_size
这里多了一个参数skip
复制代码 代码如下:
skip=BLOCKS : 拷贝数据前，从输入文件跳过BLOCKS个block，block的大小由ibs指定。这个参数和seek是对应的
上面命令的意思就是，从文件file.in开始跳过2*1M，拷贝1*1M数据，写入文件file.out的1*$file_out_size偏移处
这样基本的参数都介绍全了，无非就是设置输入输出文件以及各自的偏移，设置读写数据块大小和读取数据块个数，下面总结一下
复制代码 代码如下:
输入参数：
if
skip
ibs
count
输出参数：
of
seek
obs
最后来一道终极题。前面创建的都是null流，这次换一个
复制代码 代码如下:
指定某个字符，创建一个全是这个字符的指定大小的文件。比如创建一个文件，大小为123456字节，每个字节都是字符A
这问题看似没什么意义，但有时候确实需要用到。比如我通过/dev/null创建了一个1G的文件，但是出于测试需求我想修改中间100M数据，这时我需要创建一个100M的文件，将该文件写入到那个1G文件的指定位置，而这个100M的文件是不能从/dev/null创建的，否则达不到修改的目的，这时候就需要这样的功能了
话不多说，直接上脚本，有了前面的基础，相信都能看得懂
复制代码 代码如下:
#!/bin/bash
if [ $# -ne 3 ];then
echo "usage : $0 character out_file file_size(Byte)"
exit 1
fi
echo "$1" | grep -q "^[a-zA-Z]$"
if [ $? -ne 0 ];then
echo "arg1 must be character"
exit 1
fi
character=$1
out_file=$2
target_size=$3
# echo输出默认是带'\n'字符的，所以需要通过dd指定输入字节数
echo "$character" | dd of=$out_file ibs=1 count=1
while true
do
cur_size=`du -b $out_file | awk '{print $1}'`
if [ $cur_size -ge $target_size ];then
break
fi
remain_size=$((target_size-$cur_size))
if [ $remain_size -ge $cur_size ];then
input_size=$cur_size
else
input_size=$remain_size
fi
dd if=$out_file ibs=$input_size count=1 of=$out_file seek=1 obs=$cur_size || exit 1
done
有了这些技巧，在对文件内容无要求的前提下，你就可以任意创建指定大小的文件，任意修改文件指定字节数，这会让某些测试场合变得非常方便