#ifndef _ENCRYPT_H
#define _ENCRYPT_H

#include <sys/socket.h>
#include <string.h>
#include <stdlib.h>
#include <stdio.h>
#include <openssl/md5.h>
#include <openssl/evp.h>

#define BUF_SIZE 4096

#define TABLE 0
#define RC4   1

union {
    struct {
        unsigned char *encrypt_table;
        unsigned char *decrypt_table;
    } table;

    struct {
        unsigned char *key;
        int key_len;
    } rc4;
} enc_ctx;

void get_table(const char* key);
void encrypt_ctx(char *buf, int len, EVP_CIPHER_CTX *ctx);
void decrypt_ctx(char *buf, int len, EVP_CIPHER_CTX *ctx);
void enc_ctx_init(EVP_CIPHER_CTX *ctx, int enc);
void enc_key_init(const char *pass);

unsigned int _i;
unsigned long long _a;
int _method;

#define LOGD(...) ((void)fprintf(stdout, __VA_ARGS__))
#define LOGE(...) ((void)fprintf(stderr, __VA_ARGS__))

#endif // _ENCRYPT_H