_ Git - youtube-dl/blob - devscripts/youtube_genalgo.py

   1 #!/usr/bin/env python
   2 # encoding: utf-8
   3
   4 # Generate youtube signature algorithm from test cases
   5
   6 import sys
   7
   8 tests = [
   9     # 93 - vfl79wBKW 2013/07/20
  10     (u"qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={[]}|:;?/>.<'`~\"€",
  11      u".>/?;:|}][{=+-_)(*&^%$#@!MNBVCXZASDFGHJKLPOIUYTREWQ098765'321mnbvcxzasdfghjklpoiu"),
  12     # 92 - vflQw-fB4 2013/07/17
  13     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={[]}|:;?/>.<'`~\"",
  14      "mrtyuioplkjhgfdsazxcvbnq1234567890QWERTY}IOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={[]\"|:;"),
  15     # 90
  16     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={[]}|:;?/>.<'`",
  17      "mrtyuioplkjhgfdsazxcvbne1234567890QWER[YUIOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={`]}|"),
  18     # 89
  19     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={[]}|:;?/>.<'",
  20      "/?;:|}<[{=+-_)(*&^%$#@!MqBVCXZASDFGHJKLPOIUYTREWQ0987654321mnbvcxzasdfghjklpoiuyt"),
  21     # 88 - vflapUV9V 2013/08/28
  22     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={[]}|:;?/>.<",
  23      "ioplkjhgfdsazxcvbnm12<4567890QWERTYUIOZLKJHGFDSAeXCVBNM!@#$%^&*()_-+={[]}|:;?/>.3"),
  24     # 87
  25     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$^&*()_-+={[]}|:;?/>.<",
  26      "uioplkjhgfdsazxcvbnm1t34567890QWE2TYUIOPLKJHGFDSAZXCVeNM!@#$^&*()_-+={[]}|:;?/>.<"),
  27     # 86 - vfluy6kdb 2013/09/06
  28     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={[|};?/>.<",
  29      "yuioplkjhgfdsazxcvbnm12345678q0QWrRTYUIOELKJHGFD-AZXCVBNM!@#$%^&*()_<+={[|};?/>.S"),
  30     # 85 - vflkuzxcs 2013/09/11
  31     ('0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ!"#$%&\'()*+,-./:;<=>?@[',
  32      '3456789a0cdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRS[UVWXYZ!"#$%&\'()*+,-./:;<=>?@'),
  33     # 84 - vflg0g8PQ 2013/08/29 (sporadic)
  34     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={[};?>.<",
  35      ">?;}[{=+-_)(*&^%$#@!MNBVCXZASDFGHJKLPOIUYTREWq0987654321mnbvcxzasdfghjklpoiuytr"),
  36     # 83
  37     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!#$%^&*()_+={[};?/>.<",
  38      ".>/?;}[{=+_)(*&^%<#!MNBVCXZASPFGHJKLwOIUYTREWQ0987654321mnbvcxzasdfghjklpoiuytreq"),
  39     # 82 - vflGNjMhJ 2013/09/12
  40     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKHGFDSAZXCVBNM!@#$%^&*(-+={[};?/>.<",
  41      ".>/?;}[<=+-(*&^%$#@!MNBVCXeASDFGHKLPOqUYTREWQ0987654321mnbvcxzasdfghjklpoiuytrIwZ"),
  42     # 81 - vflLC8JvQ 2013/07/25
  43     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKHGFDSAZXCVBNM!@#$%^&*(-+={[};?/>.",
  44      "C>/?;}[{=+-(*&^%$#@!MNBVYXZASDFGHKLPOIU.TREWQ0q87659321mnbvcxzasdfghjkl4oiuytrewp"),
  45     # 80 - vflZK4ZYR 2013/08/23 (sporadic)
  46     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKHGFDSAZXCVBNM!@#$%^&*(-+={[};?/>",
  47      "wertyuioplkjhgfdsaqxcvbnm1234567890QWERTYUIOPLKHGFDSAZXCVBNM!@#$%^&z(-+={[};?/>"),
  48     # 79 - vflLC8JvQ 2013/07/25 (sporadic)
  49     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKHGFDSAZXCVBNM!@#$%^&*(-+={[};?/",
  50      "Z?;}[{=+-(*&^%$#@!MNBVCXRASDFGHKLPOIUYT/EWQ0q87659321mnbvcxzasdfghjkl4oiuytrewp"),
  51 ]
  52
  53 tests_age_gate = [
  54     # 86 - vflqinMWD
  55     ("qwertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!@#$%^&*()_-+={[|};?/>.<",
  56      "ertyuioplkjhgfdsazxcvbnm1234567890QWERTYUIOPLKJHGFDSAZXCVBNM!/#$%^&*()_-+={[|};?@"),
  57 ]
  58
  59 def find_matching(wrong, right):
  60     idxs = [wrong.index(c) for c in right]
  61     return compress(idxs)
  62     return ('s[%d]' % i for i in idxs)
  63
  64 def compress(idxs):
  65     def _genslice(start, end, step):
  66         starts = '' if start == 0 else str(start)
  67         ends = ':%d' % (end+step)
  68         steps = '' if step == 1 else (':%d' % step)
  69         return 's[%s%s%s]' % (starts, ends, steps)
  70
  71     step = None
  72     for i, prev in zip(idxs[1:], idxs[:-1]):
  73         if step is not None:
  74             if i - prev == step:
  75                 continue
  76             yield _genslice(start, prev, step)
  77             step = None
  78             continue
  79         if i - prev in [-1, 1]:
  80             step = i - prev
  81             start = prev
  82             continue
  83         else:
  84             yield 's[%d]' % prev
  85     if step is None:
  86         yield 's[%d]' % i
  87     else:
  88         yield _genslice(start, i, step)
  89
  90 def _assert_compress(inp, exp):
  91     res = list(compress(inp))
  92     if res != exp:
  93         print('Got %r, expected %r' % (res, exp))
  94         assert res == exp
  95 _assert_compress([0,2,4,6], ['s[0]', 's[2]', 's[4]', 's[6]'])
  96 _assert_compress([0,1,2,4,6,7], ['s[:3]', 's[4]', 's[6:8]'])
  97 _assert_compress([8,0,1,2,4,7,6,9], ['s[8]', 's[:3]', 's[4]', 's[7:5:-1]', 's[9]'])
  98
  99 def gen(wrong, right, indent):
 100     code = ' + '.join(find_matching(wrong, right))
 101     return 'if len(s) == %d:\n%s    return %s\n' % (len(wrong), indent, code)
 102
 103 def genall(tests):
 104     indent = ' ' * 8
 105     return indent + (indent + 'el').join(gen(wrong, right, indent) for wrong,right in tests)
 106
 107 def main():
 108     print(genall(tests))
 109     print(u'    Age gate:')
 110     print(genall(tests_age_gate))
 111
 112 if __name__ == '__main__':
 113     main()