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
135
136
137
138
139
140
|
/*
* UploadingNetworkResponder.cpp
*
* Created on: 3 Feb 2019
* Author: David
*/
#include "UploadingNetworkResponder.h"
#include "Socket.h"
#include <Platform/Platform.h>
UploadingNetworkResponder::UploadingNetworkResponder(NetworkResponder *n) noexcept : NetworkResponder(n)
#if HAS_MASS_STORAGE
, uploadError(false), dummyUpload(false)
#endif
{
}
// This is called when we lose a connection or when we are asked to terminate. Overridden in some derived classes.
void UploadingNetworkResponder::ConnectionLost() noexcept
{
CancelUpload();
NetworkResponder::ConnectionLost();
}
// If this responder has an upload in progress, cancel it
void UploadingNetworkResponder::CancelUpload() noexcept
{
#if HAS_MASS_STORAGE
if (fileBeingUploaded.IsLive())
{
fileBeingUploaded.Close();
if (!filenameBeingProcessed.IsEmpty())
{
MassStorage::Delete(filenameBeingProcessed.c_str(), true);
filenameBeingProcessed.Clear();
}
}
#endif
}
#if HAS_MASS_STORAGE
// Start writing to a new file, returning true if successful
bool UploadingNetworkResponder::StartUpload(const char* folder, const char *fileName, const OpenMode mode, const uint32_t preAllocSize) noexcept
{
if (fileName[0] == 0 || StringEndsWithIgnoreCase(fileName, ".dummy"))
{
filenameBeingProcessed.Clear();
dummyUpload = true;
}
else
{
if (!MassStorage::CombineName(filenameBeingProcessed.GetRef(), folder, fileName))
{
filenameBeingProcessed.Clear();
return false;
}
if (filenameBeingProcessed.cat(UPLOAD_EXTENSION))
{
filenameBeingProcessed.Clear();
return false;
}
FileStore * const file = GetPlatform().OpenFile(folder, filenameBeingProcessed.c_str(), mode, preAllocSize);
if (file == nullptr)
{
filenameBeingProcessed.Clear();
return false;
}
fileBeingUploaded.Set(file);
dummyUpload = false;
}
responderState = ResponderState::uploading;
uploadError = false;
return true;
}
// Finish a file upload. Set variable uploadError if anything goes wrong.
void UploadingNetworkResponder::FinishUpload(uint32_t fileLength, time_t fileLastModified, bool gotCrc, uint32_t expectedCrc) noexcept
{
if (!dummyUpload)
{
// Flush remaining data for FSO
if (!fileBeingUploaded.Flush())
{
uploadError = true;
GetPlatform().Message(ErrorMessage, "Could not flush remaining data while finishing upload\n");
}
// Check the file length is as expected
if (fileLength != 0 && fileBeingUploaded.Length() != fileLength)
{
uploadError = true;
GetPlatform().MessageF(ErrorMessage, "Uploaded file size is different (%lu vs. expected %lu bytes)\n", fileBeingUploaded.Length(), fileLength);
}
else if (gotCrc && expectedCrc != fileBeingUploaded.GetCrc32())
{
uploadError = true;
GetPlatform().MessageF(ErrorMessage, "Uploaded file CRC is different (%08" PRIx32 " vs. expected %08" PRIx32 ")\n", fileBeingUploaded.GetCrc32(), expectedCrc);
}
// Close the file
if (fileBeingUploaded.IsLive())
{
fileBeingUploaded.Close();
}
// Delete the file again if an error has occurred
if (!filenameBeingProcessed.IsEmpty())
{
const char *uploadFilename = filenameBeingProcessed.c_str();
if (uploadError)
{
MassStorage::Delete(uploadFilename, true);
}
else
{
String<MaxFilenameLength> origFilename;
origFilename.catn(uploadFilename, filenameBeingProcessed.GetRef().strlen() - strlen(UPLOAD_EXTENSION));
// Delete possibly existing files with that name (i.e. prepare "overwrite")
MassStorage::Delete(origFilename.c_str(), true);
// Rename the uploaded file to it's original name
MassStorage::Rename(uploadFilename, origFilename.c_str(), true);
if (fileLastModified != 0)
{
// Update the file timestamp if it was specified
(void)MassStorage::SetLastModifiedTime(origFilename.c_str(), fileLastModified);
}
}
filenameBeingProcessed.Clear();
}
}
}
#endif
// End
|
