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

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

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

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

生成一個大小為5G的文件，內容不做要求

命令如下

復制代碼代碼如下:

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

解釋一下這里用到的參數

復制代碼代碼如下:

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

of=FILE      : 指定輸出文件，若不指定則輸出到標準輸出  

bs=BYTES     : 每次讀寫的字節(jié)數，可以使用單位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字節(jié)偏移處（也就是文件末尾）

在此基礎上再增加一個要求

將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

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

復制代碼代碼如下:

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

這問題看似沒什么意義，但有時候確實需要用到。比如我通過/dev/null創(chuàng)建了一個1G的文件，但是出于測試需求我想修改中間100M數據，這時我需要創(chuàng)建一個100M的文件，將該文件寫入到那個1G文件的指定位置，而這個100M的文件是不能從/dev/null創(chuàng)建的，否則達不到修改的目的，這時候就需要這樣的功能了

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

復制代碼代碼如下:

#!/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指定輸入字節(jié)數  

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