Foro de elhacker.net

Buenas,
Feliz año  ;D

Estoy con un socket servidor que al estar corriendo quiero que si el usuario hace Ctrl+C en el CMD este sale limpiamente (con un SIGINT).

main.py

from sources.handlerSignals import HandlerSignals
class SocketServer:
    def __init__(self,**kwargs):
        try:
            HandlerSignals()
            # [...]
 
        # When the service shutdown successfully
        except (KeyboardInterrupt, SystemExit,GeneratorExit) as err:
            print("KeyboardInterrupt")
            None
 
        finally:
            try:
                self.sock.close()
            except:
                None
            print("Turnning off ClassAdminS")
            sys.exit(0)
 
    # continue code
if __name__=="__main__":
    SocketServer(args=sys.argv)
 

El HandlerSignals() esta en una carpeta 'sources' en el archivo handlerSignals
  -main.py
  |
  - sources/
      \-- handlerSignals.py


handlerSignals.py

import signal, sys, platform
class HandlerSignals:
    def __init__(self):
        signal.signal(signal.SIGTERM,self.shutdown)
 
        # shutdown signal in Windows
        if platform.system().upper()=="WINDOWS":
            import win32api
            win32api.SetConsoleCtrlHandler(self.shutdownWin, True)
 
    def shutdownWin(self,a):
        print("Ctrl+c")
        raise SystemExit
 
    def shutdown(self,code,msg):
        raise SystemExit
 

Como se puede ver si el sistema es Windows, se ejecuta Que esta en escucha de Ctrl+C en windows CMD. Así que al ser presionado se llama a la funcion shutdownWin que este imprime "Ctrl+C" y lanza una excepcion SystemExit.

Que esta excepcion teoricamente es recogida por main.py que y sale de la terminal.

Pero no ocurre eso. Al lanzarse el SystemExit no imprime "KeyboardInterrupt". Y no se porque pasa....


https://imgur.com/gallery/x5jxQhM

Parece como si la excepcion no se pueda pasar al main.py

Como lo puedo solucionar...? Gracias por vuestra atencion.

Buenos dias!
Ya esta resuelto. Puede que no sea la mejor opcion, ni la más adecuada, pero me veo obligado ha hacerlo así pues Windows tiende a matar los procesos.

Así como lo queria hacer tendría que funcionar para mis ojos, pero la realidad es que no, y no se porque (en Linux funcionaria....).

Lo que hice es añadir el HandlerSignals() justo despues de crear el socket (dentro del __createSocket, despues del listen)

main.py:

#!/usr/bin/python3
import sys,socket,urllib3,time,multiprocessing, threading
from sources.ClientListener import ClientListener
from sources.handlerSignals import HandlerSignals
from sources.utils import logFile, Hosts, getIpAddress, Notify
from sources.Server import Server
from sources.Requests import Requests
 
class SocketServer:
    def __init__(self,**kwargs):
        try:
            # [...]
            self.__createSocket()
 
        # When the service shutdown successfully
        except (KeyboardInterrupt, SystemExit,GeneratorExit) as err:
                None
 
        finally:
            try:
                self.sock.close()
            except:
                None
            Notify("Turnning off ClassAdminS",logFile().message("Good Bye. What you have a good day :)", True, "INFO"))
 
    def __createSocket(self):
        self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.sock.bind(("",self.PORT))
        self.sock.listen(self.CLIENTS)
        HandlerSignals(self.sock)
        Notify("Start ClassAdminS",logFile().message(f"Good days :D. I am listenning {self.CLIENTS} clients by port {self.PORT}.",True,"INFO"))
        self.__handlerClients()
 
if __name__=="__main__":
    SocketServer(args=sys.argv)
 

La diferencia es que la funcion HnadlerSignals(sock) coge como parametro la variable que contiene el objecto socket.

handlerSignals.py

import signal, platform
class HandlerSignals:
    def __init__(self,sock):
        self.sock = sock
        signal.signal(signal.SIGTERM,self.shutdown)
        # shutdown signal in Windows
        if platform.system().upper()=="WINDOWS":
            import win32api
            win32api.SetConsoleCtrlHandler(self.shutdownWin, True)
 
    def shutdownWin(self,a):
        self.sock.close()
 
    def shutdown(self,code,msg):
        raise SystemExit
 

La gran diferencia de handlerSignals.py con el anterior es que, al hacer Ctrl+C y llamar a la funcion shutdownWin(), este cierra el socket y me muestra el Notify de el try/except/finally del main.py.

Así me funciona correctamente, aunque no se si es muy buena practica,pero me funciona.

Hay que tener en cuenta que el HandlerSignals() debe estar justo después de la declaración de la variable self.sock. Yo lo pongo despues del self.sock.listen().

https://imgur.com/gallery/B2VR7Lp

Por dos cosas:

• Pues para salir limpiamente y correctamente
• Pues este script lo ejecuto como servicio con NSSM

NSSM al parar un servicio, implicitamente hace un Ctrl+C y espera una reaccion. Si no hay reaccion este se cierra forzosamente y muestra el siguiente mensaje.
https://imgur.com/gallery/n5UO4sL

Ahora si el script detecta el Ctrl+C en CMD....
https://imgur.com/gallery/qLnBGOi

En el NSSM este es configurado así
https://imgur.com/gallery/ZEwvmOq

Gracias por vuestra atencion! ;-)