#include <stdio.h>
#include <stdint.h>
#include <stdlib.h>
#include <time.h>
#define delta 0x61C88647
void Detea(uint32_t* e, int* key) {
    uint32_t v1, v2;
    int sum = 0;
    uint32_t randNum[64] = {};
    for (int i = 0; i < 32; i++) {

        randNum[i] = rand();
        //printf("0x%x,", randNum[i]);
    }
    v1 = e[0];
    v2 = e[1];
    for (int i = 0; i < 32; i++) {
        sum += randNum[i];
    }

    for (int i = 0; i < 32; ++i)
    {
        v2 -= (key[3] + (v1 >> 5)) ^ (sum + v1) ^ (key[2] + 16 * v1);
        v1 -= (key[1] + (v2 >> 5)) ^ (sum + v2) ^ (key[0] + 16 * v2);
        sum -= randNum[32 - i - 1];
    }
    e[0] = v1;
    e[1] = v2;
}



int main() {
    uint32_t enc[] = { 0xea2063f8, 0x8f66f252, 0x902a72ef, 0x411fda74, 0x19590d4d, 0xcae74317, 0x63870f3f, 0xd753ae61,0 };

    int key[] = { 2,2,3,3 };
    int index = 0;
    srand(0);


    for (index = 0; index < 4; index++) {
        Detea(&enc[index * 2], key);
    }
    puts((char*)&enc);
    return 0;
}

a=True
d=len
G=list
g=range
s=next
R=bytes
o=input
Y=print
def l(S):
 i=0
 j=0
 while a:
  i=(i+1)%256
  j=(j+S[i])%256
  S[i],S[j]=S[j],S[i]
  K=S[(S[i]+S[j])%256]
  yield K
def N(key,O):
 I=d(key)
 S=G(g(256))
 j=0
 for i in g(256):
  j=(j+S[i]+key[i%I])%256
  S[i],S[j]=S[j],S[i]
 z=l(S)
 n=[]
 for k in O:
  n.append(k^s(z)+2)
 return R(n)
def E(s,parts_num):
 Q=d(s.decode())
 S=Q//parts_num
 u=Q%parts_num
 W=[]
 j=0
 for i in g(parts_num):
  T=j+S
  if u>0:
   T+=1
   u-=1
  W.append(s[j:T])
  j=T
 return W
if __name__=='__main__':
 L=o('input the flag: >>> ').encode()
 assert d(L)%2==0,'flag length should be even'
 t=b'v3ry_s3cr3t_p@ssw0rd'
 O=E(L,2)
 U=[]
 for i in O:
  U.append(N(t,i).hex())
 if U==['1796972c348bc4fe7a1930b833ff10a80ab281627731ab705dacacfef2e2804d74ab6bc19f60',2ea999141a8cc9e47975269340c177c726a8aa732953a66a6af183bcd9cec8464a']:
  Y('Congratulations! You got the flag!')
 else:
  Y('Wrong flag!')

class RC4:
    def __init__(self, key):
        self.key = key
        self.key_schedule()

    def key_schedule(self):
        """初始化密钥调度"""
        key_length = len(self.key)
        self.S = list(range(256))  # 初始化S盒，包含0到255
        j = 0
        # 根据密钥对S盒进行初始置换
        for i in range(256):
            j = (j + self.S[i] + self.key[i % key_length]) % 256
            self.S[i], self.S[j] = self.S[j], self.S[i]

    def keystream(self):
        """生成伪随机字节流"""
        i = j = 0
        while True:
            i = (i + 1) % 256
            j = (j + self.S[i]) % 256
            self.S[i], self.S[j] = self.S[j], self.S[i]
            k = self.S[(self.S[i] + self.S[j]) % 256]
            yield k

    def encrypt(self, plaintext):
        """加密函数"""
        plaintext_bytes = bytearray(plaintext, 'utf-8')
        keystream = self.keystream()
        ciphertext = bytearray()

        for byte in plaintext_bytes:
            ciphertext.append(byte ^ (next(keystream)+2))  # 异或操作

        return bytes(ciphertext)

    def decrypt(self, ciphertext):
        """解密函数"""
        keystream = self.keystream()
        plaintext = bytearray()

        for byte in ciphertext:
            plaintext.append(byte ^ (next(keystream)+2))  # 异或操作

        return bytes(plaintext)


# 测试RC4加密和解密
if __name__ == "__main__":
    # enc = "1796972c348bc4fe7a1930b833ff10a80ab281627731ab705dacacfef2e2804d74ab6bc19f60"
    enc = "2ea999141a8cc9e47975269340c177c726a8aa732953a66a6af183bcd9cec8464a"
    key = b"v3ry_s3cr3t_p@ssw0rd"  # 密钥
    rc4 = RC4(key)
    # enc = "2ea999141a8cc9e47975269340c177c726a8aa732953a66a6af183bcd9cec8464a"
    ciphertext = bytes.fromhex(enc)
    #
    decrypted_text = rc4.decrypt(ciphertext)
    print(f"解密后的文本: {decrypted_text}")