Egress-Assess/datatypes/creditcards.py

100 lines
2.7 KiB
Python
Raw Normal View History

2014-12-24 13:37:31 +00:00
'''
This module generates credit card data
'''
import copy
import random
class Datatype:
def __init__(self, cli_object):
self.cli = "cc"
self.description = "Credit Card Numbers"
self.filetype = "text"
self.datasize = int(cli_object.data_size)
def completed_number(self, prefix, length, the_generator):
"""
'prefix' is the start of the CC number as a string, any number of digits.
'length' is the length of the CC number to generate. Typically 13 or 16
"""
ccnumber = prefix
# generate digits
while len(ccnumber) < (length - 1):
digit = str(the_generator.choice(range(0, 10)))
ccnumber.append(digit)
# Calculate sum
sum = 0
pos = 0
reversedCCnumber = []
reversedCCnumber.extend(ccnumber)
reversedCCnumber.reverse()
while pos < length - 1:
odd = int(reversedCCnumber[pos]) * 2
if odd > 9:
odd -= 9
sum += odd
if pos != (length - 2):
sum += int(reversedCCnumber[pos + 1])
pos += 2
# Calculate check digit
checkdigit = ((sum / 10 + 1) * 10 - sum) % 10
ccnumber.append(str(checkdigit))
return ''.join(ccnumber)
def credit_card_number(self, prefixList, length, howMany):
generator = random.Random()
generator.seed()
result = []
while len(result) < howMany:
ccnumber = copy.copy(generator.choice(prefixList))
result.append(self.completed_number(ccnumber, length, generator))
return result
def generate_data(self):
2015-01-15 14:42:00 +00:00
print "[*] Generating data..."
2014-12-24 13:37:31 +00:00
# credit card constants
visaPrefixList = [
['4', '5', '3', '9'],
['4', '5', '5', '6'],
['4', '9', '1', '6'],
['4', '5', '3', '2'],
['4', '9', '2', '9'],
['4', '0', '2', '4', '0', '0', '7', '1'],
['4', '4', '8', '6'],
['4', '7', '1', '6'],
['4']]
mastercardPrefixList = [
['5', '1'], ['5', '2'], ['5', '3'], ['5', '4'], ['5', '5']]
amexPrefixList = [['3', '4'], ['3', '7']]
mastercards = self.credit_card_number(
mastercardPrefixList, 16, 19800 * self.datasize)
visas = self.credit_card_number(
visaPrefixList, 16, 19800 * self.datasize)
amexes = self.credit_card_number(
amexPrefixList, 15, 19800 * self.datasize)
all_cards = mastercards + visas + amexes
final_cards = ''
for card in all_cards:
final_cards += card + ', '
return final_cards