mirror of
https://github.com/usmannasir/cyberpanel.git
synced 2025-11-10 23:36:11 +01:00
124 lines
5.2 KiB
Python
Executable File
124 lines
5.2 KiB
Python
Executable File
#!/usr/local/CyberCP/bin/python2
|
|
import os,sys
|
|
sys.path.append('/usr/local/CyberCP')
|
|
import django
|
|
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "CyberCP.settings")
|
|
django.setup()
|
|
import threading as multi
|
|
from plogical.CyberCPLogFileWriter import CyberCPLogFileWriter as logging
|
|
from policyConstraint import policyConstraints
|
|
from emailPremium.models import DomainLimits, EmailLogs
|
|
from mailServer.models import Domains, EUsers
|
|
import time
|
|
from cacheManager import cacheManager
|
|
|
|
limitThreads = multi.BoundedSemaphore(10)
|
|
|
|
class HandleRequest(multi.Thread):
|
|
cleaningPath = '/home/cyberpanel/purgeCache'
|
|
def __init__(self, conn):
|
|
multi.Thread.__init__(self)
|
|
self.connection = conn
|
|
|
|
def __del__(self):
|
|
try:
|
|
self.connection.close()
|
|
except BaseException, msg:
|
|
logging.writeToFile(str(msg) + ' [HandleRequest.__del__]')
|
|
|
|
def run(self):
|
|
limitThreads.acquire()
|
|
dataComplete = ""
|
|
|
|
try:
|
|
while True:
|
|
|
|
Data = self.connection.recv(64)
|
|
|
|
# Wait for a connection
|
|
if os.path.exists(HandleRequest.cleaningPath):
|
|
readFromFile = open(HandleRequest.cleaningPath, 'r')
|
|
command = readFromFile.read()
|
|
cacheManager.handlePurgeRequest(command)
|
|
readFromFile.close()
|
|
os.remove(HandleRequest.cleaningPath)
|
|
cacheManager.flushCache()
|
|
|
|
if Data:
|
|
if len(Data) < 64:
|
|
dataComplete = dataComplete + Data
|
|
self.manageRequest(dataComplete)
|
|
dataComplete = ''
|
|
else:
|
|
dataComplete = dataComplete + Data
|
|
else:
|
|
self.connection.close()
|
|
break
|
|
|
|
limitThreads.release()
|
|
except BaseException, msg:
|
|
logging.writeToFile(str(msg) + ' [HandleRequest.run]')
|
|
limitThreads.release()
|
|
|
|
def manageRequest(self, completeData):
|
|
try:
|
|
completeData = completeData.split('\n')
|
|
|
|
for items in completeData:
|
|
tempData = items.split('=')
|
|
if tempData[0] == 'sender':
|
|
emailAddress = tempData[1]
|
|
domainName = emailAddress.split('@')[1]
|
|
elif tempData[0] == 'recipient':
|
|
destination = tempData[1]
|
|
|
|
if domainName in cacheManager.domains:
|
|
domainObj = cacheManager.domains[domainName]
|
|
emailObj = domainObj.findEmailOBJ(emailAddress)
|
|
else:
|
|
try:
|
|
domain = Domains.objects.get(domain=domainName)
|
|
domainLTS = DomainLimits.objects.get(domain=domain)
|
|
|
|
newDomain = policyConstraints(domainName, domainLTS.monthlyLimit, domainLTS.monthlyUsed, domainLTS.limitStatus)
|
|
cacheManager.domains[domainName] = newDomain
|
|
domainObj = newDomain
|
|
|
|
emailObj = newDomain.findEmailOBJ(emailAddress)
|
|
except:
|
|
raise BaseException
|
|
|
|
#logging.writeToFile('Domain Limit Status: ' + str(domainObj.limitStatus))
|
|
#logging.writeToFile('Email Limit Status: ' + str(domainObj.limitStatus))
|
|
#logging.writeToFile('Email Monthly Limit: ' + str(emailObj.monthlyLimits))
|
|
#logging.writeToFile('Email Monthly Used: ' + str(emailObj.monthlyUsed))
|
|
|
|
if domainObj.limitStatus == 1 and emailObj.limitStatus == 1:
|
|
if domainObj.monthlyLimits <= domainObj.monthlyUsed or emailObj.monthlyLimits <= emailObj.monthlyUsed or emailObj.hourlyLimits <= emailObj.hourlyUsed:
|
|
logging.writeToFile(emailAddress + ' either exceeded monthly or hourly sending limit.')
|
|
self.connection.sendall('action=defer_if_permit Service temporarily unavailable\n\n')
|
|
else:
|
|
email = EUsers.objects.get(email=emailAddress)
|
|
if emailObj.logStatus == 1:
|
|
logEntry = EmailLogs(email=email, destination=destination, timeStamp=time.strftime("%I-%M-%S-%a-%b-%Y"))
|
|
logEntry.save()
|
|
emailObj.monthlyUsed = emailObj.monthlyUsed + 1
|
|
emailObj.hourlyUsed = emailObj.hourlyUsed + 1
|
|
domainObj.monthlyUsed = domainObj.monthlyUsed + 1
|
|
self.connection.sendall('action=dunno\n\n')
|
|
else:
|
|
email = EUsers.objects.get(email=emailAddress)
|
|
if emailObj.logStatus == 1:
|
|
logEntry = EmailLogs(email=email, destination=destination,
|
|
timeStamp=time.strftime("%I-%M-%S-%a-%b-%Y"))
|
|
logEntry.save()
|
|
emailObj.monthlyUsed = emailObj.monthlyUsed + 1
|
|
emailObj.hourlyUsed = emailObj.hourlyUsed + 1
|
|
self.connection.sendall('action=dunno\n\n')
|
|
|
|
|
|
except BaseException, msg:
|
|
self.connection.sendall('action=dunno\n\n')
|
|
limitThreads.release()
|
|
logging.writeToFile(str(msg))
|