george/SocksBin
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Directory creation complete

Browse Source
This commit is contained in:
Vividh Mariy
2020-04-11 21:09:58 +05:30
parent 932632b8bd
commit 7ae206ff9b
1 changed files with 81 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

120
paster.py
View File

@@ -1,43 +1,40 @@
#!/usr/bin/python3
import os,socket,secrets
import sys,getopt,re
import os
import socket
import secrets
import sys
import getopt
import re
# Config
def main(theargs):
# where to serve this page.
# 127.0.0.1 / localhost => only accessible locally, on this machine
# 0.0.0.0 => accessible to the internet, by any host
# 0.0.0.0 => accessible to the internet, by any host
hostname = "0.0.0.0"
# which port number to bind to
# choose a port number > 1000, to prevent interference with system processes
port = "1234"
# maximum number of clients this can handle at any given time
queue_depth = 10
# directory path
# default is $HOME i.e. /home/username/sockbin
output_directory = os.path.expanduser("~") + "sockbin"
# specify the length of generated filename. multiples of 2 only
# specify the length of generated filename. multiples of 2 only
slug_size = 8
# amount of data transferred at a time, in bytes
# set between 4096 and 64000.
# set between 4096 and 64000.
buffer_size = 32768
# path for log file
log_file = "/tmp/socklog.txt"
helpmessage = """Welcome to SockBin, the command line pastebin !
Released under GNU GPL.
@@ -53,7 +50,8 @@ def main(theargs):
"""
try:
opts, args = getopt.getopt(theargs, "n:p:q:o:s:b:l:", ["hostname=","port=", "queue_depth=", "output_directory=", "slug_size=","buffer_size=", "log_file="])
opts, args = getopt.getopt(theargs, "n:p:q:o:s:b:l:h:", [
"hostname=", "port=", "queue_depth=", "output_directory=", "slug_size=", "buffer_size=", "log_file=", "help="])
for opt, arg in opts:
if opt in ['-n', '--hostname']:
if arg.find('/') != -1:
@@ -63,46 +61,79 @@ def main(theargs):
hostname = arg
elif opt in ['-p', "--port"]:
if re.match("^ *\d[\d ]*$", arg) is None:
print("Incorrect port format. Choose a number between 0 and 64738. Warning: ports below 1000 could interfere with system processes.")
print(
"Incorrect port format. Choose a number between 0 and 64738. Warning: ports below 1000 could interfere with system processes.")
sys.exit()
elif int(arg) > 64738:
print("Incorrect port format. Choose a number between 0 and 64738. Warning: ports below 1000 could interfere with system processes.")
print(
"Incorrect port format. Choose a number between 0 and 64738. Warning: ports below 1000 could interfere with system processes.")
sys.exit()
elif int(arg) < 1000:
choice = input("Warning: ports below 1000 could interfere with system processes. Do you want to choose a different port ? (y/n): ")
choice = input(
"Warning: ports below 1000 could interfere with system processes. Do you want to choose a different port ? (y/n): ")
if choice.lower() == "y":
arg = input("Please enter your new port number: ")
if re.match("^ *\d[\d ]*$", arg) is None:
print("Incorrect port format. Choose a number between 1000 and 64738")
print(
"Incorrect port format. Choose a number between 1000 and 64738")
sys.exit()
else:
port = arg
elif choice.lower() == "n":
pass
port = arg
port = int(arg)
elif opt in ['-q', "--queue_depth"]:
queue_depth = arg
if not arg.isdigit():
print(
"Queue depth can only be an integer. Run with -h for more options")
sys.exit()
queue_depth = int(arg)
elif opt in ['-o', "--output_directory"]:
output_directory = arg
directory = arg
if os.path.isdir(directory):
permission = (isWritable(directory))
if not permission:
print(f"You do not have permissions to write to {directory}")
sys.exit()
else:
output_directory = arg
else:
parent_dir = (os.path.abspath(os.path.join(directory, os.pardir)))
if not os.path.isdir(parent_dir):
print(f"The parent folder {parent_dir} does not exist. Please try again after creating it.")
sys.exit()
permission = (isWritable(parent_dir))
if not permission:
print(f"You do not have permissions to write to {directory}, or the parent folder {parent_dir} does not exist")
sys.exit()
else:
try:
os.makedirs(directory)
print(f"> Directory {directory} created !")
output_directory = arg
except OSError:
print("unable to create directory. Please check your permissions.")
sys.exit()
elif opt in ['-s', "--slug_size"]:
slug_size = arg
elif opt in ['-b', "--buffer_size"]:
buffer_size = arg
elif opt in ['-l', "--log_file"]:
log_file = arg
print({
"hostname":hostname,
"port":port,
"queue":queue_depth,
"output":output_directory,
"slug":slug_size,
"buff":buffer_size,
"log":log_file,
"hostname": hostname,
"port": port,
"queue": queue_depth,
"output": output_directory,
"slug": slug_size,
"buff": buffer_size,
"log": log_file,
})
except getopt.GetoptError as err:
@@ -110,8 +141,6 @@ def main(theargs):
print("Run with -h or --help to see the various options")
#######################################################################################
def server():
@@ -120,21 +149,18 @@ def server():
# SOCK_STREAM => TCP Connections
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.bind(("0.0.0.0", 1234))
s.listen(5)
while True:
clientSocket, address = s.accept()
print(f"Connection from {address} has been established")
filepath = secrets.token_hex(8)
clientSocket.sendall(bytes("https://static.magnum.wtf/"+filepath, "utf-8"))
clientSocket.sendall(
bytes("https://static.magnum.wtf/"+filepath, "utf-8"))
clientSocket.shutdown(socket.SHUT_WR)
full_message = ""
while True:
data = clientSocket.recv(4096)
@@ -144,12 +170,28 @@ def server():
print("ingesting")
full_message += data.decode('utf-8')
with open(filepath+".txt", 'w') as writer:
writer.write(full_message)
def isWritable(directory):
try:
tmp_prefix = "write_tester";
count = 0
filename = os.path.join(directory, tmp_prefix)
while(os.path.exists(filename)):
filename = "{}.{}".format(os.path.join(directory, tmp_prefix),count)
count = count + 1
f = open(filename,"w")
f.close()
os.remove(filename)
return True
except Exception as e:
return False
if __name__ == "__main__":
main(sys.argv[1:])
main(sys.argv[1:])