Some of the repetitive violations were suppressed.
2.57KiB; Python | 2018-01-28 18:59:31+01 | SLOC 63
1
#!/bin/env python
2
3
##
4
# Copyright(c) 2010-2015 Intel Corporation.
5
# Copyright(c) 2016-2018 Viosoft Corporation.
6
# All rights reserved.
7
#
8
# Redistribution and use in source and binary forms, with or without
9
# modification, are permitted provided that the following conditions
10
# are met:
11
#
12
#   * Redistributions of source code must retain the above copyright
13
#     notice, this list of conditions and the following disclaimer.
14
#   * Redistributions in binary form must reproduce the above copyright
15
#     notice, this list of conditions and the following disclaimer in
16
#     the documentation and/or other materials provided with the
17
#     distribution.
18
#   * Neither the name of Intel Corporation nor the names of its
19
#     contributors may be used to endorse or promote products derived
20
#     from this software without specific prior written permission.
21
#
22
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
23
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
24
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
25
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
26
# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
27
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
28
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
29
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
30
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
31
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
32
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
33
##
34
35
import socket
36
37
class ProxSocket:
38
    def __init__(self, ip):
39 1
        self._ip = ip;
40
        self._dat = ""
41
42
        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
43
        try:
44
            sock.connect((self._ip, 8474))
45
        except:
46
            raise Exception("Failed to connect to prox on " + self._ip)
47 1
        self._sock = sock;
48
49
    def send(self, msg):
50 1
        self._sock.sendall(msg + "\n");
51
        return self
52
53
    def recv(self):
54 1
        ret_str = "";
55 1
        done = 0;
56
        while done == 0:
57 1
            if (len(self._dat) == 0):
58 1
                self._dat = self._sock.recv(256);
59 1
                if (self._dat == ''):
60 1
                    return '';
61
62 1
            while(len(self._dat)):
63 1
                if (self._dat[0] == '\n'):
64
                    done = 1
65
                    self._dat = self._dat[1:]
66 1
                    break;
67
                else:
68 1
                    ret_str += self._dat[0];
69
                    self._dat = self._dat[1:]
70 1
        return ret_str;