gen-randomfile.py

gen-randomfile.py

#!/usr/bin/env python

import sys
import os
import random

def parse_size(s):
    s = s.lower()

    base = 1
    if s.endswith('k'):
        base = 2**10
    elif s.endswith('m'):
        base = 2**20
    elif s.endswith('g'):
        base = 2**30
    else:
        s = s + ' '
    return base * int(s[:-1])

def gen_randomfile(ofile, size):
    ifile = file('/dev/urandom')
    for i in range(size/2):
        t = ifile.read(1)
        t = ord(t)
        ofile.write('%02x' % t)

    if size % 2 == 1:
        ofile.write('=')

def main():
    if len(sys.argv) != 2:
        sys.stderr.write('Usage: %s SIZE\n' % os.path.basename(sys.argv[0]))   
        sys.exit(2)

    size = parse_size(sys.argv[1])
    gen_randomfile(sys.stdout, size)

if __name__ == '__main__':
    main()