File: probe_printer.py

package info (click to toggle)
system-config-printer 1.0.0-4lenny1
  • links: PTS
  • area: main
  • in suites: lenny
  • size: 6,868 kB
  • ctags: 815
  • sloc: python: 10,430; sh: 3,752; makefile: 171; xml: 144
file content (126 lines) | stat: -rw-r--r-- 3,678 bytes parent folder | download
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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
## system-config-printer

## Copyright (C) 2006, 2007, 2008 Red Hat, Inc.
## Copyright (C) 2006 Florian Festi <ffesti@redhat.com>
## Copyright (C) 2007, 2008 Tim Waugh <twaugh@redhat.com>

## This program is free software; you can redistribute it and/or modify
## it under the terms of the GNU General Public License as published by
## the Free Software Foundation; either version 2 of the License, or
## (at your option) any later version.

## This program is distributed in the hope that it will be useful,
## but WITHOUT ANY WARRANTY; without even the implied warranty of
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
## GNU General Public License for more details.

## You should have received a copy of the GNU General Public License
## along with this program; if not, write to the Free Software
## Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.

import socket, time

class LpdServer:
    def __init__(self, hostname):
        self.hostname = hostname
        self.max_lpt_com = 8

    def _open_socket(self):
        port = 515
        try:
            host, port = self.hostname.split(":", 1)
        except ValueError:
            host = self.hostname
        
        s = None
        try:
            ai = socket.getaddrinfo(host, port, socket.AF_UNSPEC,
                                    socket.SOCK_STREAM)
        except socket.gaierror:
            ai = []

        for res in ai:
            af, socktype, proto, canonname, sa = res
            try:
                s = socket.socket(af, socktype, proto)
            except socket.error, msg:
                s = None
                continue
            try:
                s.connect(sa)
            except socket.error, msg:
                s.close()
                s = None
                continue
            break
        return s

    def _probe_queue(self,name, result):
        s = self._open_socket()
        if not s: return False
        print name
        
        try:
            s.send('\0x02%s\n' % name) # cmd send job to queue
            data = s.recv(1024) # receive status
            print repr(data)
        except socket.error, msg:
            print msg
            try:
                s.close ()
            except:
                pass

            return False

        if len(data)>0 and data[0]==0:
            try:
                s.send('\0x01\n') # abort job again
                s.close ()
            except:
                pass

            result.append(name)
            return True

        try:
            s.close()
        except:
            pass

        return False

    def probe(self):
        #s = self._open_socket()        
        #if s is None:
        #    return []
        result = []
        for name in ["PASSTHRU", "ps", "lp", "PORT1", ""]:
            self._probe_queue(name, result)
            time.sleep(0.1) # avoid DOS and following counter messures 
        for nr in range(self.max_lpt_com):
            self._probe_queue("LPT%d" % nr, result)
            time.sleep(0.1)
            self._probe_queue("LPT%d_PASSTHRU" % nr, result)
            time.sleep(0.1)
            self._probe_queue("COM%d" % nr, result)
            time.sleep(0.1)
            self._probe_queue("COM%d_PASSTHRU" % nr, result)
            time.sleep(0.1)

        nr = 1
        while nr<50:
            found =  self._probe_queue("pr%d" % nr, result)
            time.sleep(0.1)
            if not found: break
            nr += 1

        return result

class SocketServer:
    def __init__(self, hostname):
        self.hostname = hostname
    
class IppServer:
    def __init__(self, hostname):
        self.hostname = hostname