快捷導(dǎo)航

Shell腳本創(chuàng)建指定大小文件的測(cè)試數(shù)據(jù)

更新時(shí)間：2015年03月17日 09:24:42 投稿：junjie

這篇文章主要介紹了Shell腳本創(chuàng)建指定大小文件的測(cè)試數(shù)據(jù),本文同時(shí)給出使用DD命令和Shell腳本解決這個(gè)需求,給出的腳本功能更加強(qiáng)大,需要的朋友可以參考下

我們?cè)跍y(cè)試或調(diào)試的時(shí)候，有時(shí)候會(huì)需要生成某個(gè)size的文件，比如在測(cè)試存儲(chǔ)系統(tǒng)時(shí)，需要將磁盤(pán)剩余空間減少5G，最簡(jiǎn)單的辦法就是拷貝一個(gè)5G的文件過(guò)來(lái)，但是從哪兒去弄這樣大小的文件呢，或許你想到隨便找一個(gè)文件，不停的拷貝，最后合并，這也不失為一種辦法，但是有了dd，你會(huì)更容易且更靈活的實(shí)現(xiàn)。
我們來(lái)case by case的介紹dd的用法。先看第一個(gè)

生成一個(gè)大小為5G的文件，內(nèi)容不做要求

命令如下

復(fù)制代碼代碼如下:

$ dd if=/dev/zero of=tmp.5G bs=1G count=5

解釋一下這里用到的參數(shù)

復(fù)制代碼代碼如下:

if=FILE      : 指定輸入文件，若不指定則從標(biāo)注輸入讀取。這里指定為/dev/zero是Linux的一個(gè)偽文件，它可以產(chǎn)生連續(xù)不斷的null流（二進(jìn)制的0）  

of=FILE      : 指定輸出文件，若不指定則輸出到標(biāo)準(zhǔn)輸出  

bs=BYTES     : 每次讀寫(xiě)的字節(jié)數(shù)，可以使用單位K、M、G等等。另外輸入輸出可以分別用ibs、obs指定，若使用bs，則表示是ibs和obs都是用該參數(shù)  

count=BLOCKS : 讀取的block數(shù)，block的大小由ibs指定（只針對(duì)輸入?yún)?shù)）

這樣上面生成5G文件的命令就很好理解了，即從/dev/null每次讀取1G數(shù)據(jù)，讀5次，寫(xiě)入tmp.5G這個(gè)文件

再看下面一個(gè)問(wèn)題

復(fù)制代碼代碼如下:

將file.in的前1M追加到file.out的末尾

命令如下

復(fù)制代碼代碼如下:

$ 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設(shè)置為了不同的值，和前面的命令相比，只多了一個(gè)seek參數(shù)

復(fù)制代碼代碼如下:

seek=BLOCKS : 在拷貝數(shù)據(jù)之前，從輸出文件開(kāi)頭跳過(guò)BLOCKS個(gè)block，block的大小由obs指定  

命令的意思就是從file.in讀取1個(gè)1M的數(shù)據(jù)塊寫(xiě)入file.out，不過(guò)寫(xiě)入位置并不在file.out的開(kāi)頭，而是在1*$file_out_size字節(jié)偏移處（也就是文件末尾）

在此基礎(chǔ)上再增加一個(gè)要求

將file.in的第3M追加到file.out的末尾

復(fù)制代碼代碼如下:

$ 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  

這里多了一個(gè)參數(shù)skip

復(fù)制代碼代碼如下:

skip=BLOCKS : 拷貝數(shù)據(jù)前，從輸入文件跳過(guò)BLOCKS個(gè)block，block的大小由ibs指定。這個(gè)參數(shù)和seek是對(duì)應(yīng)的

上面命令的意思就是，從文件file.in開(kāi)始跳過(guò)2*1M，拷貝1*1M數(shù)據(jù)，寫(xiě)入文件file.out的1*$file_out_size偏移處

這樣基本的參數(shù)都介紹全了，無(wú)非就是設(shè)置輸入輸出文件以及各自的偏移，設(shè)置讀寫(xiě)數(shù)據(jù)塊大小和讀取數(shù)據(jù)塊個(gè)數(shù)，下面總結(jié)一下

復(fù)制代碼代碼如下:

輸入?yún)?shù)：  

    if  

    skip  

    ibs  

    count  

輸出參數(shù)：  

    of  

    seek  

    obs

最后來(lái)一道終極題。前面創(chuàng)建的都是null流，這次換一個(gè)

復(fù)制代碼代碼如下:

指定某個(gè)字符，創(chuàng)建一個(gè)全是這個(gè)字符的指定大小的文件。比如創(chuàng)建一個(gè)文件，大小為123456字節(jié)，每個(gè)字節(jié)都是字符A

這問(wèn)題看似沒(méi)什么意義，但有時(shí)候確實(shí)需要用到。比如我通過(guò)/dev/null創(chuàng)建了一個(gè)1G的文件，但是出于測(cè)試需求我想修改中間100M數(shù)據(jù)，這時(shí)我需要?jiǎng)?chuàng)建一個(gè)100M的文件，將該文件寫(xiě)入到那個(gè)1G文件的指定位置，而這個(gè)100M的文件是不能從/dev/null創(chuàng)建的，否則達(dá)不到修改的目的，這時(shí)候就需要這樣的功能了

話不多說(shuō)，直接上腳本，有了前面的基礎(chǔ)，相信都能看得懂

復(fù)制代碼代碼如下:

#!/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輸出默認(rèn)是帶'\n'字符的，所以需要通過(guò)dd指定輸入字節(jié)數(shù)  

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