support symlinks in tftp and http

pypxe · Feb 21, 2015 · cebef93 · cebef93
1 parent d463204
commit cebef93
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pypxe/http.py b/pypxe/http.py
@@ -44,7 +44,7 @@ def handleRequest(self, connection, addr):
         startline = request.split('\r\n')[0].split(' ')
         method = startline[0]
         target = startline[1]
-        if not os.path.exists(target) or not os.path.isfile(target):
+        if not os.path.lexists(target) or not os.path.isfile(target):
             status = '404 Not Found'
         elif method not in ('GET', 'HEAD'):
             status = '501 Not Implemented'

diff --git a/pypxe/tftp.py b/pypxe/tftp.py
@@ -86,7 +86,7 @@ def read(self, address, message):
                 file does not exist -> reply with error
         '''
         filename = self.filename(message)
-        if not os.path.exists(filename):
+        if not os.path.lexists(filename):
             self.notFound(address)
             return
         self.ongoing[address]['filename'] = filename