blob: 515440494b69462e85a911bcad6ca398803b2ea1 (
plain)
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
|
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#ifndef nsQuarantinedOutputStream_h__
#define nsQuarantinedOutputStream_h__
// #include "nsISupports.h"
#include "nsIOutputStream.h"
#include "nsISafeOutputStream.h"
#include "nsCOMPtr.h"
class nsIFile;
/**
* nsQuarantinedOutputStream layers on top of an existing target output stream.
* The idea is to let an OS virus checker quarantine individual messages
* _before_ they hit the mbox. You don't want entire mboxes embargoed if
* you can avoid it.
*
* It works by buffering all writes to a temporary file.
* When finish() is called the temporary file is closed, reopened,
* then copied into a pre-existing target stream. There's no special OS
* virus-checker integration - the assumption is that the checker will hook
* into the filesystem and prevent us from opening a file it has flagged as
* dodgy. Hence the temp file close/reopen before the final write.
*
* If the nsQuarantinedOutputStream is closed (or released) without calling
* finish(), the write is discarded (as per nsISafeOutputStream requirements).
*
* Upon close() or finish(), the underlying target file is also closed.
*/
class nsQuarantinedOutputStream : public nsIOutputStream, nsISafeOutputStream {
public:
NS_DECL_ISUPPORTS
NS_DECL_NSIOUTPUTSTREAM
NS_DECL_NSISAFEOUTPUTSTREAM
/**
* Pass the target output stream in during construction. Upon Close(),
* the written data will be copied here.
*/
explicit nsQuarantinedOutputStream(nsIOutputStream* target)
: mTarget(target) {}
nsQuarantinedOutputStream() = delete;
protected:
virtual ~nsQuarantinedOutputStream();
// Set up mTempFile and mTempStream (called at
// (lazily set up, upon first write).
nsresult InitTemp();
nsresult PerformAppend();
void EnterErrorState(nsresult status);
// The temporary file and stream we're writing to.
nsCOMPtr<nsIFile> mTempFile;
nsCOMPtr<nsIOutputStream> mTempStream;
// The stream we'll be appending to if it all succeeds.
nsCOMPtr<nsIOutputStream> mTarget;
enum {
eUninitialized, // No temp file yet.
eOpen, // We're up and running.
eClosed, // The file has been closed.
eError // An error has occurred (stored in mError).
} mState{eUninitialized};
nsresult mError{NS_OK};
};
#endif // nsQuarantinedOutputStream_h__
|
