Tools/DumpRenderTree/StorageTrackerDelegate.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. ``AS IS'' AND ANY
  14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
  17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
  21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  24  */
  25
  26 #import "config.h"
  27 #import "StorageTrackerDelegate.h"
  28
  29 #import "LayoutTestController.h"
  30 #import <WebKit/WebSecurityOriginPrivate.h>
  31 #import <WebKit/WebStorageManagerPrivate.h>
  32
  33 @implementation StorageTrackerDelegate
  34
  35 - (id)init
  36 {
  37     self = [super init];
  38     if (!self)
  39         return nil;
  40
  41     [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(originModified:) name:WebStorageDidModifyOriginNotification object:nil];
  42
  43     return self;
  44 }
  45
  46 - (void)logNotifications:(unsigned)number controller:(LayoutTestController*)controller
  47 {
  48     controllerToNotifyDone = controller;
  49
  50     numberOfNotificationsToLog = number;
  51 }
  52
  53 - (void)originModified:(NSNotification *)notification
  54 {
  55     if (!numberOfNotificationsToLog)
  56         return;
  57
  58     numberOfNotificationsToLog--;
  59
  60     if (numberOfNotificationsToLog == 0 && controllerToNotifyDone) {
  61         NSArray *origins = [[WebStorageManager sharedWebStorageManager] origins];
  62         for (WebSecurityOrigin *origin in origins)
  63             printf("Origin identifier: '%s'\n", [[origin databaseIdentifier] UTF8String]);
  64
  65         controllerToNotifyDone->notifyDone();
  66     }
  67 }
  68
  69 - (void)dealloc
  70 {
  71     [[NSNotificationCenter defaultCenter] removeObserver:self name:WebStorageDidModifyOriginNotification object:nil];
  72
  73     [super dealloc];
  74 }
  75
  76 - (void)setControllerToNotifyDone:(LayoutTestController*)controller
  77 {
  78     controllerToNotifyDone = controller;
  79 }
  80
  81
  82 @end