added ssl support
This commit is contained in:
parent
b4825a3fb7
commit
de4223170b
@ -1,5 +1,5 @@
|
|||||||
all:
|
all:
|
||||||
cc *.c -omain -O0 -g3 -pedantic -Wall -Wextra
|
cc *.c -omain -O0 -g3 -pedantic -Wall -Wextra -DENABLE_SSL -lcrypto -lssl
|
||||||
|
|
||||||
clean:
|
clean:
|
||||||
rm main
|
rm main
|
||||||
|
80
C/irc.c
80
C/irc.c
@ -9,6 +9,24 @@
|
|||||||
#include <arpa/inet.h>
|
#include <arpa/inet.h>
|
||||||
#include "irc.h"
|
#include "irc.h"
|
||||||
|
|
||||||
|
int IRCC_urecv(IRCC_client *irc) {
|
||||||
|
#ifdef ENABLE_SSL
|
||||||
|
if (irc->irc_usingssl)
|
||||||
|
return SSL_read(irc->irc_ssl, irc->irc_raw, sizeof(irc->irc_raw));
|
||||||
|
#endif
|
||||||
|
|
||||||
|
return recv(irc->irc_socket, irc->irc_raw, sizeof(irc->irc_raw), 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
int IRCC_usend(IRCC_client *irc, off_t bytes) {
|
||||||
|
#ifdef ENABLE_SSL
|
||||||
|
if (irc->irc_usingssl)
|
||||||
|
return SSL_write(irc->irc_ssl, irc->irc_raw, bytes);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
return send(irc->irc_socket, irc->irc_raw, bytes, 0);
|
||||||
|
}
|
||||||
|
|
||||||
int IRCC_connect(IRCC_client *irc, const char *ip, const unsigned int port) {
|
int IRCC_connect(IRCC_client *irc, const char *ip, const unsigned int port) {
|
||||||
struct hostent *hp = gethostbyname(ip);
|
struct hostent *hp = gethostbyname(ip);
|
||||||
if (!hp)
|
if (!hp)
|
||||||
@ -38,6 +56,7 @@ int IRCC_connect(IRCC_client *irc, const char *ip, const unsigned int port) {
|
|||||||
|
|
||||||
void IRCC_parse_msg(char *tmp, IRCC_client *irc) {
|
void IRCC_parse_msg(char *tmp, IRCC_client *irc) {
|
||||||
irc->irc_raw[strcspn(irc->irc_raw, "\r\n")] = '\0';
|
irc->irc_raw[strcspn(irc->irc_raw, "\r\n")] = '\0';
|
||||||
|
|
||||||
if (tmp != NULL) {
|
if (tmp != NULL) {
|
||||||
//Message
|
//Message
|
||||||
char *val = strchr(tmp, ':');
|
char *val = strchr(tmp, ':');
|
||||||
@ -67,17 +86,14 @@ void IRCC_parse_msg(char *tmp, IRCC_client *irc) {
|
|||||||
|
|
||||||
int IRCC_recv(IRCC_client *irc) {
|
int IRCC_recv(IRCC_client *irc) {
|
||||||
memset(irc->irc_raw, '\0', IRCC_MSG_MAX);
|
memset(irc->irc_raw, '\0', IRCC_MSG_MAX);
|
||||||
irc->irc_channel = NULL;
|
|
||||||
irc->irc_msg = NULL;
|
|
||||||
irc->irc_nick = NULL;
|
|
||||||
|
|
||||||
int msg_size = recv(irc->irc_socket, irc->irc_raw, sizeof(irc->irc_raw), 0);
|
int msg_size = IRCC_urecv(irc);
|
||||||
if (msg_size == 0 || msg_size == -1)
|
if (msg_size == 0 || msg_size == -1)
|
||||||
return IRCC_DISCONNECTED;
|
return IRCC_DISCONNECTED;
|
||||||
|
|
||||||
else if (!strncmp(irc->irc_raw, "PING", 4)) {
|
else if (!strncmp(irc->irc_raw, "PING", 4)) {
|
||||||
*(strchr(irc->irc_raw, 'I')) = 'O';
|
*(strchr(irc->irc_raw, 'I')) = 'O';
|
||||||
if (send(irc->irc_socket, irc->irc_raw, strlen(irc->irc_raw), 0) == -1)
|
if (IRCC_usend(irc, strlen(irc->irc_raw)) == -1)
|
||||||
return IRCC_ERROR;
|
return IRCC_ERROR;
|
||||||
|
|
||||||
return IRCC_PING;
|
return IRCC_PING;
|
||||||
@ -87,6 +103,9 @@ int IRCC_recv(IRCC_client *irc) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
int IRCC_parse(IRCC_client *irc) {
|
int IRCC_parse(IRCC_client *irc) {
|
||||||
|
irc->irc_channel = NULL;
|
||||||
|
irc->irc_msg = NULL;
|
||||||
|
irc->irc_nick = NULL;
|
||||||
|
|
||||||
//Check end of motd
|
//Check end of motd
|
||||||
if (strstr(irc->irc_raw, "PRIVMSG ") == NULL && strstr(irc->irc_raw, "MOTD"))
|
if (strstr(irc->irc_raw, "PRIVMSG ") == NULL && strstr(irc->irc_raw, "MOTD"))
|
||||||
@ -129,11 +148,11 @@ int IRCC_parse(IRCC_client *irc) {
|
|||||||
|
|
||||||
int IRCC_register(IRCC_client *irc, const char *nickname) {
|
int IRCC_register(IRCC_client *irc, const char *nickname) {
|
||||||
off_t bytes = snprintf(irc->irc_raw, sizeof(irc->irc_raw), "NICK %s\r\n", nickname);
|
off_t bytes = snprintf(irc->irc_raw, sizeof(irc->irc_raw), "NICK %s\r\n", nickname);
|
||||||
if (send(irc->irc_socket, irc->irc_raw, bytes, 0) == -1)
|
if (IRCC_usend(irc, bytes) == -1)
|
||||||
return IRCC_ERROR;
|
return IRCC_ERROR;
|
||||||
|
|
||||||
bytes = snprintf(irc->irc_raw, sizeof(irc->irc_raw), "USER %s 0 localhost :%s\r\n", nickname, nickname);
|
bytes = snprintf(irc->irc_raw, sizeof(irc->irc_raw), "USER %s 0 localhost :%s\r\n", nickname, nickname);
|
||||||
if (send(irc->irc_socket, irc->irc_raw, bytes, 0) == -1)
|
if (IRCC_usend(irc, bytes) == -1)
|
||||||
return IRCC_ERROR;
|
return IRCC_ERROR;
|
||||||
|
|
||||||
//Motd skip
|
//Motd skip
|
||||||
@ -152,7 +171,7 @@ int IRCC_register(IRCC_client *irc, const char *nickname) {
|
|||||||
|
|
||||||
int IRCC_join(IRCC_client *irc, const char *channel, const char *key) {
|
int IRCC_join(IRCC_client *irc, const char *channel, const char *key) {
|
||||||
off_t bytes = snprintf(irc->irc_raw, sizeof(irc->irc_raw), "JOIN %s %s\r\n", channel, (key) ? key : "");
|
off_t bytes = snprintf(irc->irc_raw, sizeof(irc->irc_raw), "JOIN %s %s\r\n", channel, (key) ? key : "");
|
||||||
if (send(irc->irc_socket, irc->irc_raw, bytes, 0) == -1)
|
if (IRCC_usend(irc, bytes) == -1)
|
||||||
return IRCC_ERROR;
|
return IRCC_ERROR;
|
||||||
|
|
||||||
return IRCC_SUCCESS;
|
return IRCC_SUCCESS;
|
||||||
@ -160,18 +179,57 @@ int IRCC_join(IRCC_client *irc, const char *channel, const char *key) {
|
|||||||
|
|
||||||
int IRCC_send(IRCC_client *irc, const char *channel, const char *msg) {
|
int IRCC_send(IRCC_client *irc, const char *channel, const char *msg) {
|
||||||
off_t bytes = snprintf(irc->irc_raw, sizeof(irc->irc_raw), "PRIVMSG %s :%s\r\n", channel, msg);
|
off_t bytes = snprintf(irc->irc_raw, sizeof(irc->irc_raw), "PRIVMSG %s :%s\r\n", channel, msg);
|
||||||
if (send(irc->irc_socket, irc->irc_raw, bytes, 0) == -1)
|
if (IRCC_usend(irc, bytes) == -1)
|
||||||
return IRCC_ERROR;
|
return IRCC_ERROR;
|
||||||
|
|
||||||
return IRCC_SUCCESS;
|
return IRCC_SUCCESS;
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
void IRCC_init(IRCC_client *irc) {
|
int IRCC_initssl(IRCC_client *irc, int *ircc_errno) {
|
||||||
irc->irc_msg = irc->irc_nick = irc->irc_channel = NULL;
|
#ifdef ENABLE_SSL
|
||||||
|
OpenSSL_add_all_algorithms();
|
||||||
|
|
||||||
|
if (SSL_library_init() < 0) {
|
||||||
|
*ircc_errno = IRCC_SSL_INIT;
|
||||||
|
return IRCC_ERROR;
|
||||||
|
}
|
||||||
|
|
||||||
|
irc->irc_sslmethod = (SSL_METHOD *)SSLv23_client_method();
|
||||||
|
irc->irc_sslctx = SSL_CTX_new(irc->irc_sslmethod);
|
||||||
|
if (irc->irc_sslctx == NULL) {
|
||||||
|
*ircc_errno = IRCC_SSL_METHOD;
|
||||||
|
return IRCC_ERROR;
|
||||||
|
}
|
||||||
|
|
||||||
|
SSL_CTX_set_options(irc->irc_sslctx, SSL_OP_NO_SSLv2);
|
||||||
|
irc->irc_ssl = SSL_new(irc->irc_sslctx);
|
||||||
|
|
||||||
|
SSL_set_fd(irc->irc_ssl, irc->irc_socket);
|
||||||
|
if (SSL_connect(irc->irc_ssl) != 1) {
|
||||||
|
*ircc_errno = IRCC_SSL_SESSION;
|
||||||
|
return IRCC_ERROR;
|
||||||
|
}
|
||||||
|
|
||||||
|
irc->irc_usingssl = 1;
|
||||||
|
|
||||||
|
#else
|
||||||
|
(void)irc;
|
||||||
|
*ircc_errno = 0;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
return IRCC_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
void IRCC_close(IRCC_client *irc) {
|
void IRCC_close(IRCC_client *irc) {
|
||||||
close(irc->irc_socket);
|
close(irc->irc_socket);
|
||||||
irc->irc_msg = irc->irc_nick = irc->irc_channel = NULL;
|
irc->irc_msg = irc->irc_nick = irc->irc_channel = NULL;
|
||||||
|
|
||||||
|
#ifdef ENABLE_SSL
|
||||||
|
if (!irc->irc_usingssl)
|
||||||
|
return;
|
||||||
|
|
||||||
|
SSL_free(irc->irc_ssl);
|
||||||
|
SSL_CTX_free(irc->irc_sslctx);
|
||||||
|
#endif
|
||||||
}
|
}
|
||||||
|
39
C/irc.h
39
C/irc.h
@ -1,16 +1,22 @@
|
|||||||
/*
|
/*
|
||||||
LICENSE: WTFPL
|
LICENSE: WTFPL
|
||||||
Simple libary for working with ipv4 irc servers
|
Simple library for working with ipv4 irc servers
|
||||||
Dont support:
|
Dont support:
|
||||||
ssl
|
|
||||||
ddc
|
ddc
|
||||||
ipv6
|
ipv6
|
||||||
motd
|
motd
|
||||||
*/
|
*/
|
||||||
|
|
||||||
|
#ifndef _IRCC_H
|
||||||
|
#define _IRCC_H
|
||||||
|
|
||||||
#define IRCC_MSG_MAX 512
|
#define IRCC_MSG_MAX 512
|
||||||
#define IRCC_PING_TIMEOUT 600
|
#define IRCC_PING_TIMEOUT 600
|
||||||
|
|
||||||
|
#ifdef ENABLE_SSL
|
||||||
|
#include <openssl/ssl.h>
|
||||||
|
#endif
|
||||||
|
|
||||||
enum {
|
enum {
|
||||||
IRCC_NICK = 1,
|
IRCC_NICK = 1,
|
||||||
IRCC_PRIVMSG,
|
IRCC_PRIVMSG,
|
||||||
@ -26,6 +32,19 @@ enum {
|
|||||||
IRCC_SUCCESS
|
IRCC_SUCCESS
|
||||||
};
|
};
|
||||||
|
|
||||||
|
enum {
|
||||||
|
IRCC_SSL_INIT = 1,
|
||||||
|
IRCC_SSL_METHOD,
|
||||||
|
IRCC_SSL_SESSION
|
||||||
|
};
|
||||||
|
|
||||||
|
static const char *IRCC_errno[] = {
|
||||||
|
"",
|
||||||
|
"Could not initialize the SSL library",
|
||||||
|
"Unable to create a new SSL context struct",
|
||||||
|
"Could not build a SSL session"
|
||||||
|
};
|
||||||
|
|
||||||
typedef struct {
|
typedef struct {
|
||||||
int irc_socket;
|
int irc_socket;
|
||||||
|
|
||||||
@ -33,6 +52,14 @@ typedef struct {
|
|||||||
char *irc_msg;
|
char *irc_msg;
|
||||||
char *irc_channel;
|
char *irc_channel;
|
||||||
char *irc_nick;
|
char *irc_nick;
|
||||||
|
|
||||||
|
#ifdef ENABLE_SSL
|
||||||
|
int irc_usingssl;
|
||||||
|
|
||||||
|
SSL_METHOD *irc_sslmethod;
|
||||||
|
SSL_CTX *irc_sslctx;
|
||||||
|
SSL *irc_ssl;
|
||||||
|
#endif
|
||||||
} IRCC_client;
|
} IRCC_client;
|
||||||
|
|
||||||
int IRCC_connect(IRCC_client *irc, const char *ip, const unsigned int port);
|
int IRCC_connect(IRCC_client *irc, const char *ip, const unsigned int port);
|
||||||
@ -41,5 +68,11 @@ int IRCC_parse(IRCC_client *irc);
|
|||||||
int IRCC_join(IRCC_client *irc, const char *channel, const char *key);
|
int IRCC_join(IRCC_client *irc, const char *channel, const char *key);
|
||||||
int IRCC_send(IRCC_client *irc, const char *msg, const char *channel);
|
int IRCC_send(IRCC_client *irc, const char *msg, const char *channel);
|
||||||
int IRCC_register(IRCC_client *irc, const char *nickname);
|
int IRCC_register(IRCC_client *irc, const char *nickname);
|
||||||
void IRCC_init(IRCC_client *irc);
|
int IRCC_initssl(IRCC_client *irc, int *ircc_errno);
|
||||||
void IRCC_close(IRCC_client *irc);
|
void IRCC_close(IRCC_client *irc);
|
||||||
|
|
||||||
|
//u - mean universal. Functions uses internal fields in structure
|
||||||
|
int IRCC_usend(IRCC_client *irc, off_t bytes);
|
||||||
|
int IRCC_urecv(IRCC_client *irc);
|
||||||
|
|
||||||
|
#endif
|
||||||
|
23
C/main.c
23
C/main.c
@ -5,17 +5,16 @@
|
|||||||
#include "irc.h"
|
#include "irc.h"
|
||||||
|
|
||||||
//Config
|
//Config
|
||||||
#define HOST "192.168.0.184"
|
#define HOST "irc.rizon.net"
|
||||||
#define PORT 6667
|
#define PORT 6697
|
||||||
#define NICK "tester134"
|
#define NICK "tester134"
|
||||||
char *channels[] = {"#test"};
|
char *channels[] = {"#testssl"};
|
||||||
|
|
||||||
|
|
||||||
#define CHECK_NULL() (client.irc_nick != NULL && client.irc_channel != NULL && client.irc_msg != NULL)
|
#define CHECK_NULL() (client.irc_nick != NULL && client.irc_channel != NULL && client.irc_msg != NULL)
|
||||||
IRCC_client client;
|
IRCC_client client;
|
||||||
|
|
||||||
void die(char *msg) {
|
void die(const char *msg) {
|
||||||
puts(msg);
|
perror(msg);
|
||||||
IRCC_close(&client);
|
IRCC_close(&client);
|
||||||
exit(1);
|
exit(1);
|
||||||
}
|
}
|
||||||
@ -30,7 +29,7 @@ void recvinfo(void) {
|
|||||||
|
|
||||||
//Print
|
//Print
|
||||||
if (CHECK_NULL() && irc_status == IRCC_PRIVMSG)
|
if (CHECK_NULL() && irc_status == IRCC_PRIVMSG)
|
||||||
printf("[%s %s] %s\n", client.irc_channel, client.irc_nick, client.irc_msg);
|
printf("%s\n", client.irc_msg);
|
||||||
|
|
||||||
else if (client.irc_nick != NULL && client.irc_msg != NULL && irc_status == IRCC_NICK)
|
else if (client.irc_nick != NULL && client.irc_msg != NULL && irc_status == IRCC_NICK)
|
||||||
printf("[N] %s is know as %s\n", client.irc_nick, client.irc_msg);
|
printf("[N] %s is know as %s\n", client.irc_nick, client.irc_msg);
|
||||||
@ -47,17 +46,17 @@ void recvinfo(void) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
int main(void) {
|
int main(void) {
|
||||||
|
|
||||||
//512 - size of raw buffer (max irc)
|
|
||||||
IRCC_init(&client);
|
|
||||||
int status = IRCC_connect(&client, HOST, PORT);
|
int status = IRCC_connect(&client, HOST, PORT);
|
||||||
if (status == IRCC_ERROR)
|
if (status == IRCC_ERROR)
|
||||||
die("Conn refused");
|
die("Conn refused");
|
||||||
|
|
||||||
//Register
|
int ie = 0;
|
||||||
|
if (IRCC_initssl(&client, &ie) == IRCC_ERROR)
|
||||||
|
die(IRCC_errno[ie]);
|
||||||
|
|
||||||
|
//Register and skip motd
|
||||||
IRCC_register(&client, NICK);
|
IRCC_register(&client, NICK);
|
||||||
|
|
||||||
//Recv motd and join in channel
|
|
||||||
sleep(5);
|
sleep(5);
|
||||||
for (size_t i = 0; i < sizeof(channels) / sizeof(char *); i++) {
|
for (size_t i = 0; i < sizeof(channels) / sizeof(char *); i++) {
|
||||||
if (channels[i] == NULL)
|
if (channels[i] == NULL)
|
||||||
|
@ -1 +0,0 @@
|
|||||||
Delete me
|
|
@ -1,55 +0,0 @@
|
|||||||
import socket
|
|
||||||
import time
|
|
||||||
import sys
|
|
||||||
|
|
||||||
class IRC:
|
|
||||||
def __init__(self, channel, code):
|
|
||||||
self.channel = channel
|
|
||||||
self.sock = socket.socket()
|
|
||||||
self.code = code
|
|
||||||
|
|
||||||
self.sender = {"ip":"", "nickname":""}
|
|
||||||
self.data = {"type":""}
|
|
||||||
|
|
||||||
def connect(self, host, nick):
|
|
||||||
self.sock.connect(host)
|
|
||||||
|
|
||||||
self.sock.send(f"USER {nick} 0 0 {nick}\r\n".encode(self.code))
|
|
||||||
self.sock.send(f"NICK {nick}\r\n".encode(self.code))
|
|
||||||
time.sleep(7)
|
|
||||||
self.sock.send(f"JOIN {self.channel}\r\n".encode(self.code))
|
|
||||||
|
|
||||||
def recv(self, size):
|
|
||||||
try:
|
|
||||||
data = self.sock.recv(size).decode()
|
|
||||||
except UnicodeDecodeError:
|
|
||||||
return None
|
|
||||||
|
|
||||||
if len(data) == 0:
|
|
||||||
self.sock.close()
|
|
||||||
raise BrokenPipeError("Disconnected")
|
|
||||||
|
|
||||||
if "PING" in data.split(":")[0]:
|
|
||||||
self.sock.send(f"PONG {data.split()[-1]}\r\n".encode())
|
|
||||||
return data
|
|
||||||
|
|
||||||
def parser(self, data):
|
|
||||||
data = data.split(" ", 3)
|
|
||||||
self.sender["ip"] = data[0].split("@", 1)[-1]
|
|
||||||
self.sender["nickname"] = data[0].split("!", 1)[0][1:]
|
|
||||||
|
|
||||||
if data[1] == "PRIVMSG":
|
|
||||||
self.data["channel"] = data[2]
|
|
||||||
self.data["msg"] = data[-1][1:].strip()
|
|
||||||
self.data["type"] = "PRIVMSG"
|
|
||||||
|
|
||||||
elif data[1] == "JOIN":
|
|
||||||
self.data["type"] = "JOIN"
|
|
||||||
elif data[1] == "PART":
|
|
||||||
self.data["type"] = "PART"
|
|
||||||
|
|
||||||
def sendMSG(self, msg, channel):
|
|
||||||
self.sock.send(f"PRIVMSG {channel} :{msg}\r\n".encode(self.code))
|
|
||||||
|
|
||||||
def sendRAW(self, msg):
|
|
||||||
self.sock.send(f"{msg}\r\n".encode(self.code))
|
|
@ -1,28 +0,0 @@
|
|||||||
#!/usr/bin/python3
|
|
||||||
import sys
|
|
||||||
import time
|
|
||||||
from irc import IRC
|
|
||||||
|
|
||||||
def main():
|
|
||||||
main = IRC("#mainchannel", "cp1251")
|
|
||||||
main.connect(("irc..ru", 6667), "tester")
|
|
||||||
|
|
||||||
while True:
|
|
||||||
data = main.recv(512)
|
|
||||||
if data is None:
|
|
||||||
continue
|
|
||||||
|
|
||||||
main.parser(data)
|
|
||||||
print(main.sender)
|
|
||||||
print(main.data)
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
while True:
|
|
||||||
try:
|
|
||||||
main()
|
|
||||||
|
|
||||||
except KeyboardInterrupt:
|
|
||||||
break
|
|
||||||
|
|
||||||
except:
|
|
||||||
pass
|
|
Loading…
Reference in New Issue
Block a user