9a689ca559473ed3b0154aa95cec6e51fa9a7d70
[WebKit-https.git] / Source / WebKitLegacy / mac / Storage / WebStorageManager.mm
1 /*
2  * Copyright (C) 2011 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #import "WebStorageManagerInternal.h"
27
28 #import "StorageTracker.h"
29 #import "WebKitVersionChecks.h"
30 #import "WebSecurityOriginInternal.h"
31 #import "WebStorageNamespaceProvider.h"
32 #import "WebStorageTrackerClient.h"
33 #import <WebCore/PageGroup.h>
34 #import <WebCore/SecurityOrigin.h>
35 #import <WebCore/SecurityOriginData.h>
36
37 using namespace WebCore;
38
39 NSString * const WebStorageDirectoryDefaultsKey = @"WebKitLocalStorageDatabasePathPreferenceKey";
40 NSString * const WebStorageDidModifyOriginNotification = @"WebStorageDidModifyOriginNotification";
41
42 @implementation WebStorageManager
43
44 + (WebStorageManager *)sharedWebStorageManager
45 {
46     static WebStorageManager *sharedManager = [[WebStorageManager alloc] init];
47     return sharedManager;
48 }
49
50 - (id)init
51 {
52     if (!(self = [super init]))
53         return nil;
54     
55     WebKitInitializeStorageIfNecessary();
56     
57     return self;
58 }
59
60 - (NSArray *)origins
61 {
62     auto coreOrigins = WebKit::StorageTracker::tracker().origins();
63
64     NSMutableArray *webOrigins = [[NSMutableArray alloc] initWithCapacity:coreOrigins.size()];
65
66     for (auto& origin : coreOrigins) {
67         WebSecurityOrigin *webOrigin = [[WebSecurityOrigin alloc] _initWithWebCoreSecurityOrigin:origin.securityOrigin().ptr()];
68         [webOrigins addObject:webOrigin];
69         [webOrigin release];
70     }
71
72     return [webOrigins autorelease];
73 }
74
75 - (void)deleteAllOrigins
76 {
77     WebKit::StorageTracker::tracker().deleteAllOrigins();
78 #if PLATFORM(IOS)
79     // FIXME: This needs to be removed once StorageTrackers in multiple processes
80     // are in sync: <rdar://problem/9567500> Remove Website Data pane is not kept in sync with Safari
81     [[NSFileManager defaultManager] removeItemAtPath:[WebStorageManager _storageDirectoryPath] error:NULL];
82 #endif
83 }
84
85 - (void)deleteOrigin:(WebSecurityOrigin *)origin
86 {
87     WebKit::StorageTracker::tracker().deleteOrigin(SecurityOriginData::fromSecurityOrigin(*[origin _core]));
88 }
89
90 - (unsigned long long)diskUsageForOrigin:(WebSecurityOrigin *)origin
91 {
92     return WebKit::StorageTracker::tracker().diskUsageForOrigin([origin _core]);
93 }
94
95 - (void)syncLocalStorage
96 {
97     WebKit::WebStorageNamespaceProvider::syncLocalStorage();
98 }
99
100 - (void)syncFileSystemAndTrackerDatabase
101 {
102     WebKit::StorageTracker::tracker().syncFileSystemAndTrackerDatabase();
103 }
104
105 + (NSString *)_storageDirectoryPath
106 {
107     static NSString *sLocalStoragePath;
108     static dispatch_once_t flag;
109     dispatch_once(&flag, ^{
110         NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
111         sLocalStoragePath = [defaults objectForKey:WebStorageDirectoryDefaultsKey];
112         if (!sLocalStoragePath || ![sLocalStoragePath isKindOfClass:[NSString class]]) {
113             NSArray *paths = NSSearchPathForDirectoriesInDomains(NSLibraryDirectory, NSUserDomainMask, YES);
114             NSString *libraryDirectory = [paths objectAtIndex:0];
115             sLocalStoragePath = [libraryDirectory stringByAppendingPathComponent:@"WebKit/LocalStorage"];
116         }
117         sLocalStoragePath = [[sLocalStoragePath stringByStandardizingPath] retain];
118     });
119     return sLocalStoragePath;
120 }
121
122 + (void)setStorageDatabaseIdleInterval:(double)interval
123 {
124     WebKit::StorageTracker::tracker().setStorageDatabaseIdleInterval(1_s * interval);
125 }
126
127 + (void)closeIdleLocalStorageDatabases
128 {
129     WebKit::WebStorageNamespaceProvider::closeIdleLocalStorageDatabases();
130 }
131
132 void WebKitInitializeStorageIfNecessary()
133 {
134     static BOOL initialized = NO;
135     if (initialized)
136         return;
137
138     auto *storagePath = [WebStorageManager _storageDirectoryPath];
139     WebKit::StorageTracker::initializeTracker(storagePath, WebStorageTrackerClient::sharedWebStorageTrackerClient());
140
141 #if PLATFORM(IOS)
142     if (linkedOnOrAfter(SDKVersion::FirstToExcludeLocalStorageFromBackup))
143         [[NSURL fileURLWithPath:storagePath] setResourceValue:@YES forKey:NSURLIsExcludedFromBackupKey error:nil];
144 #endif
145
146     initialized = YES;
147 }
148
149 @end