#! /usr/bin/python
# -*- coding: utf-8 -*- #
#
# @author: G1yuK
# @url: twitter.com/G1yuK
#
# shells checkers
#
# Thx xaknet.ru && gfs-team.ru

import re
import urllib2

###############################CONFIG##############################
source_file = "shells.txt"
result_file = "result_shells.txt"
###############################CONFIG##############################

def site_connect(need_url):
    arr_ans = [0, 0]
    try:
        source_page = urllib2.urlopen(need_url)
        arr_ans = [1, source_page]
    except urllib2.URLError, e:
        arr_ans[1] = e

    return arr_ans

def parse_shell(site):
    shell_info = []
    site       = site.read()
    re_all     = r'<b>Software:&nbsp;(.*?)<a.*?<b>uname -a:&nbsp;(.*?)</b>.*?<b>Safe-mode:&nbsp;<font color=.*?>(.*?)</font>'
    re_all     = re.compile(re_all)
    shell_info = re.findall(re_all, site)[0]

    return shell_info

def main(file_in):
    file_out = open(result_file, "w+")
    file_out.write("Url\tPing\t\tSoftWare\t\tOS\t\tSafeMode\n")
    for one_shell in open(file_in):
        one_shell.strip()
        re_url   = r'(http://[\w.]+\.+[a-z]{2,4})/'
        re_url   = re.compile(re_url)
        main_url = re.findall(re_url, one_shell)
        file_out.write(main_url)
        if site[0]:
            shell_info = parse_shell(site[1])
            file_out.write("\tOk")
            for part in shell_info:
                file_out.write("\t" + part)
        else:
#            file_out.write("\t" + str(site[1]))
            file_out.write("\tOff?")
        file_out.write("\n")

print "\nThx xaknet.ru && gfs-team.ru\n"

file_in = raw_input("Enter shell's file(default: shells.txt):")
if file_in:
    source_file = file_in

main(source_file)

raw_input("\nFinish\nPress Enter...")