cmd/damage-cmd.py

   1 #!/bin/sh
   2 """": # -*-python-*-
   3 bup_python="$(dirname "$0")/bup-python" || exit $?
   4 exec "$bup_python" "$0" ${1+"$@"}
   5 """
   6 # end of bup preamble
   7
   8 from __future__ import absolute_import
   9 import sys, os, random
  10
  11 from bup import options
  12 from bup.compat import range
  13 from bup.helpers import log
  14
  15
  16 def randblock(n):
  17     l = []
  18     for i in range(n):
  19         l.append(chr(random.randrange(0,256)))
  20     return ''.join(l)
  21
  22
  23 optspec = """
  24 bup damage [-n count] [-s maxsize] [-S seed] <filenames...>
  25 --
  26    WARNING: THIS COMMAND IS EXTREMELY DANGEROUS
  27 n,num=   number of blocks to damage
  28 s,size=  maximum size of each damaged block
  29 percent= maximum size of each damaged block (as a percent of entire file)
  30 equal    spread damage evenly throughout the file
  31 S,seed=  random number seed (for repeatable tests)
  32 """
  33 o = options.Options(optspec)
  34 (opt, flags, extra) = o.parse(sys.argv[1:])
  35
  36 if not extra:
  37     o.fatal('filenames expected')
  38
  39 if opt.seed != None:
  40     random.seed(opt.seed)
  41
  42 for name in extra:
  43     log('Damaging "%s"...\n' % name)
  44     f = open(name, 'r+b')
  45     st = os.fstat(f.fileno())
  46     size = st.st_size
  47     if opt.percent or opt.size:
  48         ms1 = int(float(opt.percent or 0)/100.0*size) or size
  49         ms2 = opt.size or size
  50         maxsize = min(ms1, ms2)
  51     else:
  52         maxsize = 1
  53     chunks = opt.num or 10
  54     chunksize = size/chunks
  55     for r in range(chunks):
  56         sz = random.randrange(1, maxsize+1)
  57         if sz > size:
  58             sz = size
  59         if opt.equal:
  60             ofs = r*chunksize
  61         else:
  62             ofs = random.randrange(0, size - sz + 1)
  63         log('  %6d bytes at %d\n' % (sz, ofs))
  64         f.seek(ofs)
  65         f.write(randblock(sz))
  66     f.close()