75 lines
3.1 KiB
Diff
75 lines
3.1 KiB
Diff
diff --git a/Lib/logging/handlers.py b/Lib/logging/handlers.py
|
|
index 11ebcf1..ee3d960 100644
|
|
--- a/Lib/logging/handlers.py
|
|
+++ b/Lib/logging/handlers.py
|
|
@@ -181,14 +181,17 @@ class RotatingFileHandler(BaseRotatingHandler):
|
|
Basically, see if the supplied record would cause the file to exceed
|
|
the size limit we have.
|
|
"""
|
|
+ # See bpo-45401: Never rollover anything other than regular files
|
|
+ if os.path.exists(self.baseFilename) and not os.path.isfile(self.baseFilename):
|
|
+ return False
|
|
if self.stream is None: # delay was set...
|
|
self.stream = self._open()
|
|
if self.maxBytes > 0: # are we rolling over?
|
|
msg = "%s\n" % self.format(record)
|
|
self.stream.seek(0, 2) #due to non-posix-compliant Windows feature
|
|
if self.stream.tell() + len(msg) >= self.maxBytes:
|
|
- return 1
|
|
- return 0
|
|
+ return True
|
|
+ return False
|
|
|
|
class TimedRotatingFileHandler(BaseRotatingHandler):
|
|
"""
|
|
@@ -335,10 +338,13 @@ class TimedRotatingFileHandler(BaseRotatingHandler):
|
|
record is not used, as we are just comparing times, but it is needed so
|
|
the method signatures are the same
|
|
"""
|
|
+ # See bpo-45401: Never rollover anything other than regular files
|
|
+ if os.path.exists(self.baseFilename) and not os.path.isfile(self.baseFilename):
|
|
+ return False
|
|
t = int(time.time())
|
|
if t >= self.rolloverAt:
|
|
- return 1
|
|
- return 0
|
|
+ return True
|
|
+ return False
|
|
|
|
def getFilesToDelete(self):
|
|
"""
|
|
diff --git a/Lib/test/test_logging.py b/Lib/test/test_logging.py
|
|
index 45b72e3..055b8e3 100644
|
|
--- a/Lib/test/test_logging.py
|
|
+++ b/Lib/test/test_logging.py
|
|
@@ -4219,6 +4219,13 @@ class RotatingFileHandlerTest(BaseFileTest):
|
|
rh = logging.handlers.RotatingFileHandler(self.fn, maxBytes=0)
|
|
self.assertFalse(rh.shouldRollover(None))
|
|
rh.close()
|
|
+ # bpo-45401 - test with special file
|
|
+ # We set maxBytes to 1 so that rollover would normally happen, except
|
|
+ # for the check for regular files
|
|
+ rh = logging.handlers.RotatingFileHandler(
|
|
+ os.devnull, encoding="utf-8", maxBytes=1)
|
|
+ self.assertFalse(rh.shouldRollover(self.next_rec()))
|
|
+ rh.close()
|
|
|
|
def test_should_rollover(self):
|
|
rh = logging.handlers.RotatingFileHandler(self.fn, maxBytes=1)
|
|
@@ -4294,6 +4301,15 @@ class RotatingFileHandlerTest(BaseFileTest):
|
|
rh.close()
|
|
|
|
class TimedRotatingFileHandlerTest(BaseFileTest):
|
|
+ def test_should_not_rollover(self):
|
|
+ # See bpo-45401. Should only ever rollover regular files
|
|
+ fh = logging.handlers.TimedRotatingFileHandler(
|
|
+ os.devnull, 'S', encoding="utf-8", backupCount=1)
|
|
+ time.sleep(1.1) # a little over a second ...
|
|
+ r = logging.makeLogRecord({'msg': 'testing - device file'})
|
|
+ self.assertFalse(fh.shouldRollover(r))
|
|
+ fh.close()
|
|
+
|
|
# other test methods added below
|
|
def test_rollover(self):
|
|
fh = logging.handlers.TimedRotatingFileHandler(self.fn, 'S',
|