nichkara/ninachloe
1
0
Fork 0
mirror of https://github.com/nichkara/InfinitumBotty.git synced 2026-06-11 06:36:24 +02:00
Code Issues Packages Projects Releases Wiki Activity

add certfp feature for bot auth

Browse Source
This commit is contained in:
Context 77
2025-03-03 01:19:14 +01:00
parent a9a5e859ba
commit 1216660260
1 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
FaustBot/Communication/Connection.py
+10 -7
View File
@@ -3,6 +3,8 @@ import queue
import socket import socket
import time import time
import ssl import ssl
import os.path
from threading import Condition from threading import Condition
from FaustBot.Communication.JoinObservable import JoinObservable from FaustBot.Communication.JoinObservable import JoinObservable
@@ -137,12 +139,17 @@ class Connection(object):
""" """
establish the connection establish the connection
""" """
use_certfp = os.path.isfile("certfp.pem")
self.irc = socket.socket(socket.AF_INET, socket.SOCK_STREAM) self.irc = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
socker = socket.create_connection( socker = socket.create_connection(
(self.details.get_server(), self.details.get_port()) (self.details.get_server(), self.details.get_port())
) )
if self.details.get_ssl().lower() != "false": if self.details.get_ssl().lower() != "false":
self.wraper = ssl.create_default_context() if use_certfp:
self.wraper = ssl.create_default_context(ssl.Purpose.SERVER_AUTH)
self.wraper.load_cert_chain("certfp.pem")
else:
self.wraper = ssl.create_default_context()
self.irc = self.wraper.wrap_socket( self.irc = self.wraper.wrap_socket(
socker, server_hostname=self.details.get_server() socker, server_hostname=self.details.get_server()
) )
@@ -155,12 +162,8 @@ class Connection(object):
self.irc.send( self.irc.send(
f"PRIVMSG NICKSERV :identify {self.details.get_nick()} {self.details.get_pwd()} \r\n".encode() f"PRIVMSG NICKSERV :identify {self.details.get_nick()} {self.details.get_pwd()} \r\n".encode()
) )
# self.send_to_user( # Sleep a bit to ensure that the Bot is fully logged in.
# "NICKSERV", time.sleep(11.2233)
# f"identify {self.details.get_nick()} {self.details.get_pwd()} ",
# )
# Sleep a bit to ensure that the Bot is fully logged in.
time.sleep(11.2233)
self.irc.send(f"JOIN {self.details.get_channel()}\r\n".encode()) self.irc.send(f"JOIN {self.details.get_channel()}\r\n".encode())
self.irc.send(f"WHO {self.details.get_channel()}\r\n".encode()) self.irc.send(f"WHO {self.details.get_channel()}\r\n".encode())
self.irc.send(f"MODE {self.details.get_nick()} -R\r\n".encode()) self.irc.send(f"MODE {self.details.get_nick()} -R\r\n".encode())