2008-01-18 Mark Rowe <mrowe@apple.com>
[WebKit-https.git] / WebCore / platform / win / FileSystemWin.cpp
1 /*
2  * Copyright (C) 2007 Apple Inc. All rights reserved.
3  * Copyright (C) 2008 Collabora, Ltd. All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  *
9  * 1.  Redistributions of source code must retain the above copyright
10  *     notice, this list of conditions and the following disclaimer. 
11  * 2.  Redistributions in binary form must reproduce the above copyright
12  *     notice, this list of conditions and the following disclaimer in the
13  *     documentation and/or other materials provided with the distribution. 
14  * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of
15  *     its contributors may be used to endorse or promote products derived
16  *     from this software without specific prior written permission. 
17  *
18  * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
19  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
20  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
21  * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
22  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
23  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
24  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
25  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
27  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28  */
29
30 #include "config.h"
31 #include "FileSystem.h"
32
33 #include "CString.h"
34 #include "NotImplemented.h"
35 #include "PlatformString.h"
36
37 #include <windows.h>
38 #include <winbase.h>
39 #include <shlobj.h>
40 #include <shlwapi.h>
41
42 namespace WebCore {
43
44 bool fileSize(const String& path, long long& result)
45 {
46     struct _stat32i64 sb;
47     String filename = path;
48     int statResult = _wstat32i64(filename.charactersWithNullTermination(), &sb);
49     if (statResult != 0 || (sb.st_mode & S_IFMT) != S_IFREG)
50         return false;
51     result = sb.st_size;
52     return true;
53 }
54
55 bool fileExists(const String& path) 
56 {
57     struct _stat32i64 sb;
58     String filename = path;
59     return !_wstat32i64(filename.charactersWithNullTermination(), &sb);
60 }
61
62 bool deleteFile(const String& path)
63 {
64     String filename = path;
65     return !!DeleteFileW(filename.charactersWithNullTermination());
66 }
67
68 String pathByAppendingComponent(const String& path, const String& component)
69 {
70     Vector<UChar> buffer(MAX_PATH);
71
72     if (path.length() + 1 > buffer.size())
73         return String();
74
75     memcpy(buffer.data(), path.characters(), path.length() * sizeof(UChar));
76     buffer[path.length()] = '\0';
77
78     String componentCopy = component;
79     if (!PathAppendW(buffer.data(), componentCopy.charactersWithNullTermination()))
80         return String();
81
82     buffer.resize(wcslen(buffer.data()));
83
84     return String::adopt(buffer);
85 }
86
87 CString fileSystemRepresentation(const String&)
88 {
89     return "";
90 }
91
92 bool makeAllDirectories(const String& path)
93 {
94     String fullPath = path;
95     if (SHCreateDirectoryEx(0, fullPath.charactersWithNullTermination(), 0) != ERROR_SUCCESS) {
96         DWORD error = GetLastError();
97         if (error != ERROR_FILE_EXISTS && error != ERROR_ALREADY_EXISTS) {
98             LOG_ERROR("Failed to create path %s", path.ascii().data());
99             return false;
100         }
101     }
102     return true;
103 }
104
105 String homeDirectoryPath()
106 {
107     notImplemented();
108     return "";
109 }
110
111 static String bundleName()
112 {
113     static bool initialized;
114     static String name = "WebKit";
115
116     if (!initialized) {
117         initialized = true;
118
119         if (CFBundleRef bundle = CFBundleGetMainBundle())
120             if (CFTypeRef bundleExecutable = CFBundleGetValueForInfoDictionaryKey(bundle, kCFBundleExecutableKey))
121                 if (CFGetTypeID(bundleExecutable) == CFStringGetTypeID())
122                     name = reinterpret_cast<CFStringRef>(bundleExecutable);
123     }
124
125     return name;
126 }
127
128 static String storageDirectory(DWORD pathIdentifier)
129 {
130     Vector<UChar> buffer(MAX_PATH);
131     if (FAILED(SHGetFolderPathW(0, pathIdentifier | CSIDL_FLAG_CREATE, 0, 0, buffer.data())))
132         return String();
133     buffer.resize(wcslen(buffer.data()));
134     String directory = String::adopt(buffer);
135
136     static const String companyNameDirectory = "Apple Computer\\";
137     directory = pathByAppendingComponent(directory, companyNameDirectory + bundleName());
138     if (!makeAllDirectories(directory))
139         return String();
140
141     return directory;
142 }
143
144 static String cachedStorageDirectory(DWORD pathIdentifier)
145 {
146     static HashMap<DWORD, String> directories;
147
148     HashMap<DWORD, String>::iterator it = directories.find(pathIdentifier);
149     if (it != directories.end())
150         return it->second;
151
152     String directory = storageDirectory(pathIdentifier);
153     directories.add(pathIdentifier, directory);
154
155     return directory;
156 }
157
158
159 CString openTemporaryFile(const char* prefix, PlatformFileHandle& handle)
160 {
161     char tempPath[MAX_PATH];
162     int tempPathLength = ::GetTempPathA(_countof(tempPath), tempPath);
163     if (tempPathLength <= 0 || tempPathLength > _countof(tempPath))
164         return 0;
165
166     char tempFile[MAX_PATH];
167     if (::GetTempFileNameA(tempPath, prefix, 0, tempFile) > 0) {
168         HANDLE tempHandle = ::CreateFileA(tempFile, GENERIC_READ | GENERIC_WRITE, 0, 0, 
169             CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
170
171         if (isHandleValid(tempHandle)) {
172             handle = tempHandle;
173             return tempFile;
174         }
175     }
176     return 0;
177 }
178
179 void closeFile(PlatformFileHandle& handle)
180 {
181     if (isHandleValid(handle)) {
182         ::CloseHandle(handle);
183         handle = invalidPlatformFileHandle;
184     }
185 }
186
187 int writeToFile(PlatformFileHandle handle, const char* data, int length)
188 {
189     if (!isHandleValid(handle))
190         return -1;
191
192     DWORD bytesWritten;
193     bool success = WriteFile(handle, data, length, &bytesWritten, 0);
194
195     if (!success)
196         return -1;
197     return static_cast<int>(bytesWritten);
198 }
199 String localUserSpecificStorageDirectory()
200 {
201     return cachedStorageDirectory(CSIDL_LOCAL_APPDATA);
202 }
203
204 String roamingUserSpecificStorageDirectory()
205 {
206     return cachedStorageDirectory(CSIDL_APPDATA);
207 }
208
209 bool safeCreateFile(const String& path, CFDataRef data)
210 {
211     // Create a temporary file.
212     WCHAR tempDirPath[MAX_PATH];
213     if (!GetTempPathW(ARRAYSIZE(tempDirPath), tempDirPath))
214         return false;
215
216     WCHAR tempPath[MAX_PATH];
217     if (!GetTempFileNameW(tempDirPath, L"WEBKIT", 0, tempPath))
218         return false;
219
220     HANDLE tempFileHandle = CreateFileW(tempPath, GENERIC_READ | GENERIC_WRITE, 0, 0, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, 0);
221     if (tempFileHandle == INVALID_HANDLE_VALUE)
222         return false;
223
224     // Write the data to this temp file.
225     DWORD written;
226     if (!WriteFile(tempFileHandle, CFDataGetBytePtr(data), static_cast<DWORD>(CFDataGetLength(data)), &written, 0))
227         return false;
228
229     CloseHandle(tempFileHandle);
230
231     // Copy the temp file to the destination file.
232     String destination = path;
233     if (!MoveFileExW(tempPath, destination.charactersWithNullTermination(), MOVEFILE_REPLACE_EXISTING | MOVEFILE_COPY_ALLOWED))
234         return false;
235
236     return true;
237 }
238
239 } // namespace WebCore