crack.py 5.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165
  1. #!/usr/bin/python
  2. # -*- coding: iso-8859-15 -*-
  3. """
  4. This file is part of the cintruder project, http://cintruder.03c8.net
  5. Copyright (c) 2012/2019 psy <epsylon@riseup.net>
  6. cintruder is free software; you can redistribute it and/or modify it under
  7. the terms of the GNU General Public License as published by the Free
  8. Software Foundation version 3 of the License.
  9. cintruder is distributed in the hope that it will be useful, but WITHOUT ANY
  10. WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  11. FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
  12. details.
  13. You should have received a copy of the GNU General Public License along
  14. with cintruder; if not, write to the Free Software Foundation, Inc., 51
  15. Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
  16. """
  17. from PIL import Image
  18. import hashlib, os, math, time
  19. import shutil
  20. class VectorCompare:
  21. def magnitude(self, concordance):
  22. total = 0
  23. for word, count in concordance.iteritems():
  24. # print concordance
  25. total += count ** 2
  26. return math.sqrt(total)
  27. def relation(self, concordance1, concordance2):
  28. topvalue = 0
  29. for word, count in concordance1.iteritems():
  30. if concordance2.has_key(word):
  31. topvalue += count * concordance2[word]
  32. return topvalue / (self.magnitude(concordance1) * self.magnitude(concordance2))
  33. class CIntruderCrack(object):
  34. """
  35. Class to bruteforce captchas
  36. """
  37. def __init__(self, captcha=""):
  38. """
  39. Initialize main CIntruder
  40. """
  41. self.captcha = self.set_captcha(captcha)
  42. start = time.time()
  43. if not os.path.exists("core/images/previews/"):
  44. os.mkdir("core/images/previews/")
  45. else:
  46. shutil.rmtree("core/images/previews/")
  47. os.mkdir("core/images/previews/")
  48. def buildvector(self, im):
  49. d1 = {}
  50. count = 0
  51. for i in im.getdata():
  52. d1[count] = i
  53. count += 1
  54. return d1
  55. def set_captcha(self, captcha):
  56. """
  57. Set the captcha.
  58. """
  59. self.captcha = captcha
  60. return captcha
  61. def crack(self, options):
  62. v = VectorCompare()
  63. path, dirs, files = os.walk("dictionary/").next()
  64. dictionary = dirs
  65. imageset = []
  66. last_letter = None
  67. print "\n[Info] Loading dictionary...\n"
  68. for letter in dictionary:
  69. for img in os.listdir('dictionary/'+letter):
  70. temp = []
  71. temp.append(self.buildvector(Image.open("dictionary/%s/%s"%(letter, img))))
  72. imageset.append({letter:temp})
  73. try:
  74. im = Image.open(self.captcha)
  75. im.save("core/images/previews/last-preview.gif")
  76. im2 = Image.new("P", im.size, 255)
  77. im = im.convert("P")
  78. except:
  79. print "\nError during cracking!. Is that captcha supported?\n"
  80. if os.path.exists('core/images/previews'):
  81. shutil.rmtree('core/images/previews') # remove last OCR
  82. return
  83. temp = {}
  84. try:
  85. setids = int(options.setids)
  86. except:
  87. setids = 3 # default pixel colour id
  88. for x in range(im.size[1]):
  89. for y in range(im.size[0]):
  90. pix = im.getpixel((y, x))
  91. temp[pix] = pix
  92. if pix == setids: # pixel colour id
  93. im2.putpixel((y, x), 0)
  94. inletter = False
  95. foundletter = False
  96. start = 0
  97. end = 0
  98. letters = []
  99. for y in range(im2.size[0]): # slice across
  100. for x in range(im2.size[1]): # slice down
  101. pix = im2.getpixel((y, x))
  102. if pix == 0: # != 255
  103. inletter = True
  104. if foundletter == False and inletter == True:
  105. foundletter = True
  106. start = y
  107. if foundletter == True and inletter == False:
  108. foundletter = False
  109. end = y
  110. letters.append((start, end))
  111. inletter = False
  112. count = 0
  113. countid = 1
  114. word_sug = None
  115. end = time.time()
  116. elapsed = end - start
  117. words = {}
  118. for letter in letters:
  119. m = hashlib.md5()
  120. im3 = im2.crop((letter[0], 0, letter[1], im2.size[1]))
  121. guess = []
  122. for image in imageset:
  123. for x, y in image.iteritems():
  124. if len(y) != 0:
  125. guess.append(( v.relation(y[0], self.buildvector(im3)), x))
  126. guess.sort(reverse=True)
  127. word_per = guess[0][0] * 100
  128. if str(word_per) == "100.0":
  129. print "Image position :", countid
  130. print "Broken Percent :", int(round(float(word_per))), "%", "[+CRACKED!]"
  131. words[countid] = guess[0][1]
  132. else:
  133. print "Image position :", countid
  134. print "Broken Percent :", "%.4f" % word_per, "%"
  135. words[countid] = "_"
  136. print "Word suggested :", guess[0][1]
  137. print "-------------------"
  138. if word_sug == None:
  139. word_sug = str(guess[0][1])
  140. else:
  141. word_sug = word_sug + str(guess[0][1])
  142. count += 1
  143. countid = countid + 1
  144. print "\n========================================"
  145. if options.verbose:
  146. print "[Info] Elapsed OCR time :", elapsed
  147. print "========================================"
  148. if word_sug is None:
  149. print "Suggested Solution: ", "[ No idea!. Try to add more images to your dictionary/]"
  150. else:
  151. print "Cracked Words: ", words.values()
  152. print "Suggested Solution: ", "[", word_sug, "]"
  153. print "========================================\n"
  154. return word_sug