#include <stdio.h>
void rc4_init(unsigned char *key, int key_length, unsigned char *S)
{
    for (int i = 0; i < 256; i++) {
        S[i] = i;
    }
    int j = 0;
    for (int i = 0; i < 256; i++) {
        j = (j + S[i] + key[i % key_length]) % 256;
        // Swap S[i] and S[j]
        unsigned char temp = S[i];
        S[i] = S[j];
        S[j] = temp;
    }
}
void rc4_encrypt(unsigned char *data, int data_length, unsigned char *S)
{
    int i = 0;
    int j = 0;
    for (int k = 0; k < data_length; k++) {
        i = (i + 1) % 256;
        j = (j + S[i]) % 256;
        // Swap S[i] and S[j]
        unsigned char temp = S[i];
        S[i] = S[j];
        S[j] = temp;
        // Encrypt data
        data[k] ^= S[(S[i] + S[j]) % 256];
    }
}
int main()
{
    unsigned char key[] = "SecretKey";
    unsigned char data[] = "Hello, RC4!";
    int key_length = 9;
    int data_length = 12;
    unsigned char S[256];
    rc4_init(key, key_length, S);
    rc4_encrypt(data, data_length, S);
    printf("Encrypted data: ");
    for (int i = 0; i < data_length; i++) {
        printf("%02X ", data[i]);
    }
    printf("\n");
    return 0;
}

#include <iostream>
#include <vector>
void rc4_init(std::vector<unsigned char>& key, std::vector<unsigned char>& S)
{
    for (int i = 0; i < 256; i++) {
        S[i] = i;
    }
    int j = 0;
    for (int i = 0; i < 256; i++) {
        j = (j + S[i] + key[i % key.size()]) % 256;
        // Swap S[i] and S[j]
        std::swap(S[i], S[j]);
    }
}
void rc4_encrypt(std::vector<unsigned char>& data, std::vector<unsigned char>& S)
{
    int i = 0;
    int j = 0;
    for (size_t k = 0; k < data.size(); k++) {
        i = (i + 1) % 256;
        j = (j + S[i]) % 256;
        // Swap S[i] and S[j]
        std::swap(S[i], S[j]);
        // Encrypt data
        data[k] ^= S[(S[i] + S[j]) % 256];
    }
}
int main()
{
    std::vector<unsigned char> key = {'S', 'e', 'c', 'r', 'e', 't', 'K', 'e', 'y'};
    std::vector<unsigned char> data = {'H', 'e', 'l', 'l', 'o', ',', ' ', 'R', 'C', '4', '!'};
    std::vector<unsigned char> S(256);
    rc4_init(key, S);
    rc4_encrypt(data, S);
    std::cout << "Encrypted data: ";
    for (size_t i = 0; i < data.size(); i++) {
        std::cout << std::hex << static_cast<int>(data[i]) << ' ';
    }
    std::cout << std::dec << std::endl;
    return 0;}