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
127
128
129
130
131
132
133
134
| @PJL FSUPLOAD NAME="jetdirect.py" SIZE=5119
import os
import sys
import socket
import logging
import re
import hashlib
class PJLServer:
def __init__(self):
self._server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self._server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
def listen(self, port=9100, backlog=100):
self._server.bind(("127.0.0.1", port))
self._server.listen(backlog)
logging.info("Listening on port %d" % port)
def accept(self):
client, addr = self._server.accept()
logging.info("[%s] connected" % addr[0])
return PJLClient(client, addr[0])
class PJLClient:
def __init__(self, client, address):
self._client = client
self._address = address
def get_line(self):
"""Reads until a newline to get a single PJL command."""
line = b""
while True:
char = self._client.recv(1)
if not char: return None
line += char
if char == b"\n": break
return line
def reply(self, message):
if isinstance(message, str):
message = message.encode("utf-8")
self._client.sendall(message)
def close(self):
self._client.close()
class Filesystem:
def __init__(self, root_dir):
self._root = os.path.abspath(root_dir)
def _translate(self, path):
clean = path.replace("0:", "").replace("\\", "/").lstrip("/")
return os.path.normpath(os.path.join(self._root, clean))
def listdir(self, name=""):
target = self._translate(name)
if not os.path.exists(target): return "FILEERROR=1"
try:
items = os.listdir(target)
res = [". TYPE=DIR", ".. TYPE=DIR"]
for i in items:
p = os.path.join(target, i)
res.append(f"{i} TYPE={'DIR' if os.path.isdir(p) else 'FILE'} SIZE={os.path.getsize(p)}")
return "\n".join(res)
except: return "FILEERROR=1"
def read(self, path):
target = self._translate(path)
if os.path.isfile(target):
with open(target, "rb") as f: return f.read()
return None
def write(self, path, data):
target = self._translate(path)
try:
os.makedirs(os.path.dirname(target), exist_ok=True)
with open(target, "wb") as f: f.write(data)
return "OK"
except: return "FILEERROR=1"
fs = None
def handle_download(command, client):
m = re.search(r'NAME\s*=\s*"([^"]+)"\s*SIZE\s*=\s*(\d+)', command, re.I)
if not m: return "FILEERROR=1"
path, size = m.group(1), int(m.group(2))
logging.info(f"Receiving file: {path} ({size} bytes)")
data = b""
while len(data) < size:
chunk = client._client.recv(min(size - len(data), 4096))
if not chunk: break
data += chunk
return fs.write(path, data)
def handle_upload(command):
m = re.search(r'NAME\s*=\s*"([^"]+)"', command, re.I)
if not m: return "FILEERROR=1"
path = m.group(1)
data = fs.read(path)
if data is None: return "FILEERROR=1"
header = f'@PJL FSUPLOAD NAME="{path}" SIZE={len(data)}\n'.encode("utf-8")
return header + data
if __name__ == "__main__":
if len(sys.argv) < 3:
print(f"Usage: {sys.argv[0]} <PORT> <ROOT_DIR>")
sys.exit(1)
fs = Filesystem(sys.argv[2])
LOG_FILE = "/home/archivist/printer/logs/commands.log"
logging.basicConfig(
level=logging.INFO,
format="%(message)s",
handlers=[
logging.FileHandler(LOG_FILE),
logging.StreamHandler(sys.stdout)
]
)
server = PJLServer()
server.listen(int(sys.argv[1]))
while True:
client = server.accept()
while True:
line_bytes = client.get_line()
if not line_bytes: break
if b"@" not in line_bytes: continue
line = line_bytes[line_bytes.find(b"@"):].decode("utf-8", errors="ignore").strip()
|
