root / vncauthproxy / rfb.py @ 4a1dd7af
History | View | Annotate | Download (2.5 kB)
1 |
# Copyright (c) 2010-2013 GRNET SA
|
---|---|
2 |
#
|
3 |
# This program is free software; you can redistribute it and/or modify
|
4 |
# it under the terms of the GNU General Public License as published by
|
5 |
# the Free Software Foundation; either version 2 of the License, or
|
6 |
# (at your option) any later version.
|
7 |
#
|
8 |
# This program is distributed in the hope that it will be useful, but
|
9 |
# WITHOUT ANY WARRANTY; without even the implied warranty of
|
10 |
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
11 |
# General Public License for more details.
|
12 |
#
|
13 |
# You should have received a copy of the GNU General Public License
|
14 |
# along with this program; if not, write to the Free Software
|
15 |
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
|
16 |
# 02110-1301, USA.
|
17 |
|
18 |
import d3des |
19 |
from struct import pack, unpack |
20 |
|
21 |
RFB_AUTH_SUCCESS = 0
|
22 |
RFB_AUTH_ERROR = 1
|
23 |
RFB_AUTHTYPE_VNC = 2
|
24 |
RFB_AUTHTYPE_NONE = 1
|
25 |
RFB_AUTHTYPE_ERROR = 0
|
26 |
RFB_SUPPORTED_AUTHTYPES = [RFB_AUTHTYPE_NONE, RFB_AUTHTYPE_VNC] |
27 |
RFB_VERSION_3_8 = "RFB 003.008"
|
28 |
RFB_VERSION_3_7 = "RFB 003.007"
|
29 |
RFB_VERSION_3_3 = "RFB 003.003"
|
30 |
RFB_VALID_VERSIONS = [RFB_VERSION_3_3, |
31 |
# RFB_VERSION_3_7,
|
32 |
RFB_VERSION_3_8] |
33 |
|
34 |
|
35 |
class RfbError(Exception): |
36 |
pass
|
37 |
|
38 |
|
39 |
def check_version(version): |
40 |
if version.strip()[:11] in RFB_VALID_VERSIONS: |
41 |
return version.strip()[:11] |
42 |
else:
|
43 |
return None |
44 |
|
45 |
|
46 |
def make_auth_request(*args, **kwargs): |
47 |
auth_methods = args |
48 |
version = kwargs['version']
|
49 |
if version == RFB_VERSION_3_3:
|
50 |
if len(auth_methods) != 1: |
51 |
raise RfbError("Only one auth type may be specified for RFB 3.3") |
52 |
auth_methods = set(auth_methods)
|
53 |
for method in auth_methods: |
54 |
if method not in RFB_SUPPORTED_AUTHTYPES: |
55 |
raise RfbError("Unsupported authentication type: %d" % method) |
56 |
if version == RFB_VERSION_3_3:
|
57 |
return pack('>I', *auth_methods) |
58 |
else:
|
59 |
return pack('B' + 'B' * len(auth_methods), len(auth_methods), |
60 |
*auth_methods) |
61 |
|
62 |
|
63 |
def parse_auth_request(request): |
64 |
length = unpack('B', request[0])[0] |
65 |
if length == 0: |
66 |
return []
|
67 |
return unpack('B' * length, request[1:]) |
68 |
|
69 |
|
70 |
def parse_client_authtype(authtype): |
71 |
return unpack('B', authtype[0])[0] |
72 |
|
73 |
|
74 |
def from_u32(val): |
75 |
return unpack('>L', val)[0] |
76 |
|
77 |
|
78 |
def to_u32(val): |
79 |
return pack('>L', val) |
80 |
|
81 |
|
82 |
def from_u8(val): |
83 |
return unpack('B', val)[0] |
84 |
|
85 |
|
86 |
def to_u8(val): |
87 |
return pack('B', val) |
88 |
|
89 |
|
90 |
def check_password(challenge, response, password): |
91 |
return d3des.generate_response((password + '\0' * 8)[:8], |
92 |
challenge) == response |