SecureClick/controller.py at main · KrishayNair/SecureClick · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
from urllib.parse import urlparse, urlencode, quote, unquote
import tldextract
import model

global BASE_SCORE
BASE_SCORE = 50  # default trust_ score of url out of 100


def main(url):

    try:

        # input validattion
        url = model.include_protocol(url)
        url_validation = model.validate_url(url)

        # default data
        domain = tldextract.extract(url).domain + '.' + tldextract.extract(url).suffix
        response = {'status': 'SUCCESS', 'url': url}
        trust_score = BASE_SCORE


        # ================== starting url assessment ==================

        # phishtank check
        phishtank_response = model.phishtank_search(url)
        if phishtank_response:
            response['msg'] = "This is a verified phishing link."


        # website status
        response['response_status'] = url_validation


        # domain_rank
        domain_rank = model.get_domain_rank(domain)
        trust_score = model.calculate_trust_score(trust_score, 'domain_rank', domain_rank)
        if domain_rank:
            response['rank'] = domain_rank
        else:
            response['rank'] = '10,00,000+'


        # domain_age and whois_data
        whois_data = model.whois_data(domain)
        trust_score = model.calculate_trust_score(trust_score, 'domain_age', whois_data['age'])
        if whois_data['age'] == 'Not Given':
            response['age'] = whois_data['age']
        else:
            response['age'] = str(round(whois_data['age'],1)) + ' year(s)'
        response['whois'] = whois_data['data']


        # is_url_shortened
        is_url_shortened = model.is_url_shortened(url)
        trust_score = model.calculate_trust_score(trust_score, 'is_url_shortened', is_url_shortened)
        response['is_url_shortened'] = is_url_shortened


        # hsts_support
        hsts_support = model.hsts_support(url)
        trust_score = model.calculate_trust_score(trust_score, 'hsts_support', hsts_support)
        response['hsts_support'] = hsts_support


        # ip_present
        ip_present = model.ip_present(url)
        trust_score = model.calculate_trust_score(trust_score, 'ip_present', ip_present)
        response['ip_present'] = ip_present


        # url_redirects
        url_redirects = model.url_redirects(url)
        trust_score = model.calculate_trust_score(trust_score, 'url_redirects', url_redirects)
        response['url_redirects'] = url_redirects


        # too_long_url
        too_long_url = model.too_long_url(url)
        trust_score = model.calculate_trust_score(trust_score, 'too_long_url', too_long_url)
        response['too_long_url'] = too_long_url


        # too_deep_url
        too_deep_url = model.too_deep_url(url)
        trust_score = model.calculate_trust_score(trust_score, 'too_deep_url', too_deep_url)
        response['too_deep_url'] = too_deep_url


        # get ip address
        ip = model.get_ip(domain)
        if ip == 0:
            response['ip'] = 'Unavailable'
        else:
            response['ip'] = ip


        # get_certificate_details
        ssl = model.get_certificate_details(domain)
        response['ssl'] = ssl


        trust_score = int(max(min(trust_score, 100), 0))
        response['trust_score']= trust_score
        return response


    except Exception as e:
        print(f"Error: {e}")
        response = {'status': 'ERROR', 'url': url, 'msg': "Some error occurred, please check the URL."}
        return response