CTFd/populate.py

943 lines
17 KiB
Python

#!/usr/bin/python
# -*- coding: utf-8 -*-
from CTFd.models import Teams, Solves, Challenges, WrongKeys, Keys, Tags, Files, Tracking
from CTFd import create_app
from random import randint
import datetime
import random
import hashlib
import os
import sys
app = create_app(sys.argv[1], sys.argv[2], sys.argv[3])
USER_AMOUNT = 50
CHAL_AMOUNT = 20
categories = [
'Exploitation',
'Reversing',
'Web',
'Forensics',
'Scripting',
'Cryptography',
'Networking',
]
lorems = [
'Lorem',
'ipsum',
'dolor',
'sit',
'amet,',
'consectetur',
'adipiscing',
'elit.',
'Proin',
'fringilla',
'elit',
'velit,',
'sed',
'scelerisque',
'tellus',
'dapibus',
'vel.',
'Aenean',
'at',
'urna',
'porta,',
'fringilla',
'erat',
'eget,',
'lobortis',
'quam.',
'Praesent',
'luctus,',
'quam',
'at',
'consequat',
'luctus,',
'mauris',
'sem',
'pretium',
'metus,',
'eu',
'viverra',
'dui',
'leo',
'in',
'tortor.',
'Cras',
'iaculis',
'enim',
'erat,',
'sed',
'gravida',
'velit',
'consectetur',
'a.',
'Duis',
'eget',
'fermentum',
'elit.',
'Vivamus',
'laoreet',
'elementum',
'massa,',
'ut',
'sodales',
'mi',
'gravida',
'at.',
'Vivamus',
'dignissim',
'in',
'eros',
'non',
'iaculis.',
'Vivamus',
'nec',
'sem',
'fringilla,',
'semper',
'lectus',
'in,',
'malesuada',
'tellus.',
'Vestibulum',
'mattis',
'commodo',
'enim',
'sit',
'amet',
'scelerisque.',
'Proin',
'at',
'condimentum',
'nisi,',
'nec',
'fringilla',
'ante.',
'Vestibulum',
'sit',
'amet',
'neque',
'sit',
'amet',
'elit',
'placerat',
'interdum',
'egestas',
'ac',
'malesuada',
'quis',
'arcu',
'ac',
'blandit.',
'Vivamus',
'in',
'massa',
'a',
'purus',
'bibendum',
'sagittis.',
'Nunc',
'venenatis',
'lacus',
'sed',
'nulla',
'dapibus,',
'consequat',
'laoreet',
'nisi',
'faucibus.',
'Nam',
'consequat',
'viverra',
'nibh',
'a',
'cursus.',
'Phasellus',
'tristique',
'justo',
'vitae',
'rutrum',
'pharetra.',
'Sed',
'sed',
'porttitor',
'lacus.',
'Nam',
'ornare',
'sit',
'amet',
'nisi',
'imperdiet',
'vulputate.',
'Maecenas',
'hendrerit',
'ullamcorper',
'elit,',
'sed',
'pellentesque',
'lacus',
'bibendum',
'sit',
'amet.',
'Aliquam',
'consectetur',
'odio',
'quis',
'tellus',
'ornare,',
'id',
'malesuada',
'dui',
'rhoncus.',
'Quisque',
'fringilla',
'pellentesque',
'nulla',
'id',
'congue.',
'Nulla',
'ultricies',
'dolor',
'tristique',
'facilisis',
'at',
'accumsan',
'nisi.',
'Praesent',
'commodo,',
'mauris',
'sit',
'amet',
'placerat',
'condimentum,',
'nibh',
'leo',
'pulvinar',
'justo,',
'vel',
'dignissim',
'mi',
'dolor',
'et',
'est.',
'Nulla',
'facilisi.',
'Sed',
'nunc',
'est,',
'lobortis',
'id',
'diam',
'nec,',
'vulputate',
'varius',
'orci.',
'Maecenas',
'iaculis',
'vehicula',
'eros',
'eu',
'congue.',
'Nam',
'tempor',
'commodo',
'lobortis.',
'Donec',
'eget',
'posuere',
'dolor,',
'ut',
'rhoncus',
'tortor.',
'Donec',
'et',
'quam',
'quis',
'urna',
'rhoncus',
'fermentum',
'et',
'ut',
'tellus.',
'Aliquam',
'erat',
'volutpat.',
'Morbi',
'porttitor',
'ante',
'nec',
'porta',
'mollis.',
'Ut',
'sodales',
'pellentesque',
'rutrum.',
'Nullam',
'elit',
'eros,',
'sollicitudin',
'ac',
'rutrum',
'sit',
'amet,',
'eleifend',
'vel',
'nulla.',
'Morbi',
'quis',
'lacinia',
'nisi.',
'Integer',
'at',
'neque',
'vel',
'velit',
'tincidunt',
'elementum',
'lobortis',
'sit',
'amet',
'tellus.',
'Nunc',
'volutpat',
'diam',
'ac',
'diam',
'lacinia,',
'id',
'molestie',
'quam',
'eu',
'ultricies',
'ligula.',
'Duis',
'iaculis',
'massa',
'massa,',
'eget',
'venenatis',
'dolor',
'fermentum',
'laoreet.',
'Nam',
'posuere,',
'erat',
'quis',
'tempor',
'consequat,',
'purus',
'erat',
'hendrerit',
'arcu,',
'nec',
'aliquam',
'ligula',
'augue',
'vitae',
'felis.',
'Vestibulum',
'tincidunt',
'ipsum',
'vel',
'pharetra',
'lacinia.',
'Quisque',
'dignissim,',
'arcu',
'non',
'feugiat',
'semper,',
'felis',
'est',
'commodo',
'lorem,',
'malesuada',
'elementum',
'nibh',
'lectus',
'porttitor',
'nisi.',
'Duis',
'non',
'lacinia',
'nisl.',
'Etiam',
'ante',
'nisl,',
'mattis',
'eget',
'convallis',
'vel,',
'ullamcorper',
'ac',
'nisl.',
'Duis',
'eu',
'massa',
'at',
'urna',
'laoreet',
'convallis.',
'Donec',
'tincidunt',
'sapien',
'sit',
'amet',
'varius',
'eu',
'dignissim',
'tortor,',
'elementum',
'gravida',
'eros.',
'Cras',
'viverra',
'accumsan',
'erat,',
'et',
'euismod',
'dui',
'placerat',
'ac.',
'Ut',
'tortor',
'arcu,',
'euismod',
'vitae',
'aliquam',
'in,',
'interdum',
'vitae',
'magna.',
'Vestibulum',
'leo',
'ante,',
'posuere',
'eget',
'est',
'non,',
'adipiscing',
'ultrices',
'erat.',
'Donec',
'suscipit',
'felis',
'molestie,',
'ultricies',
'dui',
'a,',
'facilisis',
'magna.',
'Cum',
'sociis',
'natoque',
'penatibus',
'et',
'magnis',
'dis',
'parturient',
'montes,',
'nascetur',
'ridiculus',
'mus.',
'Nulla',
'quis',
'odio',
'sit',
'amet',
'ante',
'tristique',
'accumsan',
'ut',
'iaculis',
'neque.',
'Vivamus',
'in',
'venenatis',
'enim.',
'Nunc',
'dignissim',
'justo',
'neque,',
'sed',
'ultricies',
'justo',
'dictum',
'in.',
'Nulla',
'eget',
'nunc',
'ac',
'arcu',
'vestibulum',
'bibendum',
'vitae',
'quis',
'tellus.',
'Morbi',
'bibendum,',
'quam',
'ac',
'cursus',
'posuere,',
'purus',
'lectus',
'tempor',
'est,',
'eu',
'iaculis',
'quam',
'enim',
'a',
'nibh.',
'Etiam',
'consequat',
'libero',
'vitae',
'ullamcorper',
'tincidunt.',
]
hipsters = ['Ethnic', 'narwhal', 'pickled', 'Odd', 'Future', 'cliche', 'VHS', 'whatever', 'Etsy', 'American', 'Apparel', 'kitsch', 'wolf', 'mlkshk', 'fashion', 'axe', 'ethnic', 'banh', 'mi', 'cornhole', 'scenester', 'Echo', 'Park', 'Dreamcatcher', 'tofu', 'fap', 'selvage', 'authentic', 'cliche', 'High', 'Life', 'brunch', 'pork', 'belly', 'viral', 'XOXO', 'drinking', 'vinegar', 'bitters', 'Wayfarers', 'gastropub', 'dreamcatcher', 'chillwave', 'Shoreditch', 'kale', 'chips', 'swag', 'street', 'art', 'put', 'a', 'bird', 'on', 'it', 'Vice', 'synth', 'cliche', 'retro', 'Master', 'cleanse', 'ugh', 'Austin', 'slow-carb', 'small', 'batch', 'Hashtag', 'food', 'truck', 'deep', 'v', 'semiotics', 'chia', 'normcore', 'bicycle', 'rights', 'Austin', 'drinking', 'vinegar', 'hella', 'readymade', 'farm-to-table', 'Wes', 'Anderson', 'put', 'a', 'bird', 'on', 'it', 'freegan', 'Synth', 'lo-fi', 'food', 'truck', 'chambray', 'Shoreditch', 'cliche', 'kogiSynth', 'lo-fi', 'fap', 'single-origin', 'coffee', 'brunch', 'butcher', 'Pickled', 'Etsy', 'locavore', 'forage', 'pug', 'stumptown', 'occupy', 'PBR&B', 'actually', 'shabby', 'chic', 'church-key', 'disrupt', 'lomo', 'hoodie', 'Tumblr', 'biodiesel', 'Pinterest', 'butcher', 'Hella', 'Carles', 'pour-over', 'YOLO', 'VHS', 'literally', 'Selvage', 'narwhal', 'flexitarian', 'wayfarers', 'kitsch', 'bespoke', 'sriracha', 'Banh', 'mi', '8-bit', 'cornhole', 'viral', 'Tonx', 'keytar', 'gastropub', 'YOLO', 'hashtag', 'food', 'truck', '3', 'wolf', 'moonFingerstache', 'flexitarian', 'craft', 'beer', 'shabby', 'chic', '8-bit', 'try-hard', 'semiotics', 'Helvetica', 'keytar', 'PBR', 'four', 'loko', 'scenester', 'keytar', '3', 'wolf', 'moon', 'sriracha', 'gluten-free', 'literally', 'try-hard', 'put', 'a', 'bird', 'on', 'it', 'cornhole', 'blog', 'fanny', 'pack', 'Mumblecore', 'pickled', 'distillery', 'butcher', 'Ennui', 'tote', 'bag', 'letterpress', 'disrupt', 'keffiyeh', 'art', 'party', 'aesthetic', 'Helvetica', 'stumptown', 'Wes', 'Anderson', 'next', 'level', "McSweeney's", 'cornhole', 'Schlitz', 'skateboard', 'pop-up', 'Chillwave', 'biodiesel', 'semiotics', 'seitan', 'authentic', 'bicycle', 'rights', 'wolf', 'pork', 'belly', 'letterpress', 'locavore', 'whatever', 'fixie', 'viral', 'mustache', 'beard', 'Hashtag', 'sustainable', 'lomo', 'cardigan', 'lo-fiWilliamsburg', 'craft', 'beer', 'bitters', 'iPhone', 'gastropub', 'messenger', 'bag', 'Organic', 'post-ironic', 'fingerstache', 'ennui', 'banh', 'mi', 'Art', 'party', 'bitters', 'twee', 'bespoke', 'church-key', 'Intelligentsia', 'sriracha', 'Echo', 'Park', 'Tofu', 'locavore', 'street', 'art', 'freegan', 'farm-to-table', 'distillery', 'hoodie', 'swag', 'ugh', 'YOLO', 'VHS', 'Cred', 'hella', 'readymade', 'distillery', 'Banh', 'mi', 'Echo', 'Park', "McSweeney's,", 'mlkshk', 'photo', 'booth', 'swag', 'Odd', 'Future', 'squid', 'Tonx', 'craft', 'beer', 'High', 'Life', 'tousled', 'PBR', 'you', 'probably', "haven't", 'heard', 'of', 'them', 'locavore', 'PBR&B', 'street', 'art', 'pop-up', 'raw', 'denim']
names = [
'James',
'John',
'Robert',
'Michael',
'William',
'David',
'Richard',
'Joseph',
'Charles',
'Thomas',
'Christopher',
'Daniel',
'Matthew',
'Donald',
'Anthony',
'Paul',
'Mark',
'George',
'Steven',
'Kenneth',
'Andrew',
'Edward',
'Brian',
'Joshua',
'Kevin',
'Ronald',
'Timothy',
'Jason',
'Jeffrey',
'Gary',
'Ryan',
'Nicholas',
'Eric',
'Stephen',
'Jacob',
'Larry',
'Frank',
'Jonathan',
'Scott',
'Justin',
'Raymond',
'Brandon',
'Gregory',
'Samuel',
'Patrick',
'Benjamin',
'Jack',
'Dennis',
'Jerry',
'Alexander',
'Tyler',
'Douglas',
'Henry',
'Peter',
'Walter',
'Aaron',
'Jose',
'Adam',
'Harold',
'Zachary',
'Nathan',
'Carl',
'Kyle',
'Arthur',
'Gerald',
'Lawrence',
'Roger',
'Albert',
'Keith',
'Jeremy',
'Terry',
'Joe',
'Sean',
'Willie',
'Jesse',
'Ralph',
'Billy',
'Austin',
'Bruce',
'Christian',
'Roy',
'Bryan',
'Eugene',
'Louis',
'Harry',
'Wayne',
'Ethan',
'Jordan',
'Russell',
'Alan',
'Philip',
'Randy',
'Juan',
'Howard',
'Vincent',
'Bobby',
'Dylan',
'Johnny',
'Phillip',
'Craig',
'Mary',
'Patricia',
'Elizabeth',
'Jennifer',
'Linda',
'Barbara',
'Susan',
'Margaret',
'Jessica',
'Dorothy',
'Sarah',
'Karen',
'Nancy',
'Betty',
'Lisa',
'Sandra',
'Helen',
'Donna',
'Ashley',
'Kimberly',
'Carol',
'Michelle',
'Amanda',
'Emily',
'Melissa',
'Laura',
'Deborah',
'Stephanie',
'Rebecca',
'Sharon',
'Cynthia',
'Ruth',
'Kathleen',
'Anna',
'Shirley',
'Amy',
'Angela',
'Virginia',
'Brenda',
'Pamela',
'Catherine',
'Katherine',
'Nicole',
'Christine',
'Janet',
'Debra',
'Carolyn',
'Samantha',
'Rachel',
'Heather',
'Maria',
'Diane',
'Frances',
'Joyce',
'Julie',
'Martha',
'Joan',
'Evelyn',
'Kelly',
'Christina',
'Emma',
'Lauren',
'Alice',
'Judith',
'Marie',
'Doris',
'Ann',
'Jean',
'Victoria',
'Cheryl',
'Megan',
'Kathryn',
'Andrea',
'Jacqueline',
'Gloria',
'Teresa',
'Janice',
'Sara',
'Rose',
'Julia',
'Hannah',
'Theresa',
'Judy',
'Mildred',
'Grace',
'Beverly',
'Denise',
'Marilyn',
'Amber',
'Danielle',
'Brittany',
'Diana',
'Jane',
'Lori',
'Olivia',
'Tiffany',
'Kathy',
'Tammy',
'Crystal',
'Madison',
]
emails = [
'@gmail.com',
'@yahoo.com',
'@outlook.com',
'@hotmail.com',
'@mailinator.com',
'@poly.edu',
'@nyu.edu'
]
extensions = [
'.doc',
'.log',
'.msg',
'.rtf',
'.txt',
'.wpd',
'.wps',
'.123',
'.csv',
'.dat',
'.db ',
'.dll',
'.mdb',
'.pps',
'.ppt',
'.sql',
'.wks',
'.xls',
'.xml',
'.mng',
'.pct',
'.bmp',
'.gif',
'.jpe',
'.jpg',
'.png',
'.psd',
'.psp',
'.tif',
'.ai ',
'.drw',
'.dxf',
'.eps',
'.ps ',
'.svg',
'.3dm',
'.3dm',
'.ind',
'.pdf',
'.qxd',
'.qxp',
'.aac',
'.aif',
'.iff',
'.m3u',
'.mid',
'.mid',
'.mp3',
'.mpa',
'.ra ',
'.ram',
'.wav',
'.wma',
'.3gp',
'.asf',
'.asx',
'.avi',
'.mov',
'.mp4',
'.mpg',
'.qt ',
'.rm ',
'.swf',
'.wmv',
'.asp',
'.css',
'.htm',
'.htm',
'.js ',
'.jsp',
'.php',
'.xht',
'.fnt',
'.fon',
'.otf',
'.ttf',
'.8bi',
'.plu',
'.xll',
'.cab',
'.cpl',
'.cur',
'.dmp',
'.drv',
'.key',
'.lnk',
'.sys',
'.cfg',
'.ini',
'.reg',
'.app',
'.bat',
'.cgi',
'.com',
'.exe',
'.pif',
'.vb ',
'.ws ',
'.deb',
'.gz ',
'.pkg',
'.rar',
'.sea',
'.sit',
'.sit',
'.zip',
'.bin',
'.hqx',
'.0 E',
'.mim',
'.uue',
'.cpp',
'.jav',
'.pl ',
'.bak',
'.gho',
'.old',
'.ori',
'.tmp',
'.dmg',
'.iso',
'.toa',
'.vcd',
'.gam',
'.nes',
'.rom',
'.sav',
'.msi',
'.tor',
'.yps'
]
def gen_sentence():
return ' '.join(random.sample(lorems, 50))
def gen_name():
return random.choice(names)
def gen_email():
return random.choice(emails)
def gen_category():
return random.choice(categories)
def gen_value():
return random.choice(range(100, 500, 50))
def gen_word():
return random.choice(hipsters)
def gen_file():
return gen_word()+random.choice(extensions)
def random_date(start, end):
return start + datetime.timedelta(
seconds=randint(0, int((end - start).total_seconds())))
with app.app_context():
db = app.db
### Generating Challenges
print "GENERATING CHALLENGES"
for x in range(CHAL_AMOUNT):
word = gen_word()
db.session.add(Challenges(word, gen_sentence(), gen_value(), gen_category()))
db.session.commit()
db.session.add(Keys(x+1, word, 0))
db.session.commit()
### Generating Files
print "GENERATING FILES"
AMT_CHALS_WITH_FILES = int(CHAL_AMOUNT * (3.0/4.0))
for x in range(AMT_CHALS_WITH_FILES):
chal = random.randint(1, CHAL_AMOUNT)
filename = gen_file()
md5hash = hashlib.md5(filename).hexdigest()
db.session.add( Files(chal, os.path.join('static/uploads', md5hash, filename)) )
db.session.commit()
### Generating Users
print "GENERATING USERS"
used = []
while len(used) < USER_AMOUNT:
name = gen_name()
if name not in used:
used.append(name)
db.session.add( Teams(name , name.lower() + gen_email(), 'password') )
db.session.commit()
### Generating Solves
print "GENERATING SOLVES"
base_time = datetime.datetime.utcnow() + datetime.timedelta(minutes=-2880)
for x in range(USER_AMOUNT):
used = []
for y in range(random.randint(1,CHAL_AMOUNT)):
chalid = random.randint(1,CHAL_AMOUNT)
if chalid not in used:
used.append(chalid)
solve = Solves(chalid, x+1, '127.0.0.1')
new_base = random_date(base_time, base_time + datetime.timedelta(minutes=60))
solve.date = new_base
base_time = new_base
db.session.add(solve)
db.session.commit()
### Generating Wrong Keys
print "GENERATING WRONG KEYS"
base_time = datetime.datetime.utcnow() + datetime.timedelta(minutes=-3880)
for x in range(USER_AMOUNT):
used = []
for y in range(random.randint(1,CHAL_AMOUNT * 20)):
chalid = random.randint(1,CHAL_AMOUNT)
if chalid not in used:
used.append(chalid)
wrong = WrongKeys(x+1, chalid, gen_word())
new_base = random_date(base_time, base_time + datetime.timedelta(minutes=60))
wrong.date = new_base
base_time = new_base
db.session.add(wrong)
db.session.commit()
db.session.close()