StorageTrackerDelegate.mm 2.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283
  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. #import "config.h"
  26. #import "StorageTrackerDelegate.h"
  27. #import "TestRunner.h"
  28. #import <WebKit/WebSecurityOriginPrivate.h>
  29. #import <WebKit/WebStorageManagerPrivate.h>
  30. @implementation StorageTrackerDelegate
  31. - (id)init
  32. {
  33. self = [super init];
  34. if (!self)
  35. return nil;
  36. [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(originModified:) name:WebStorageDidModifyOriginNotification object:nil];
  37. return self;
  38. }
  39. - (void)logNotifications:(unsigned)number controller:(TestRunner*)controller
  40. {
  41. controllerToNotifyDone = controller;
  42. numberOfNotificationsToLog = number;
  43. }
  44. - (void)originModified:(NSNotification *)notification
  45. {
  46. if (!numberOfNotificationsToLog)
  47. return;
  48. numberOfNotificationsToLog--;
  49. if (numberOfNotificationsToLog == 0 && controllerToNotifyDone) {
  50. NSArray *origins = [[WebStorageManager sharedWebStorageManager] origins];
  51. for (WebSecurityOrigin *origin in origins)
  52. printf("Origin identifier: '%s'\n", [[origin databaseIdentifier] UTF8String]);
  53. controllerToNotifyDone->notifyDone();
  54. }
  55. }
  56. - (void)dealloc
  57. {
  58. [[NSNotificationCenter defaultCenter] removeObserver:self name:WebStorageDidModifyOriginNotification object:nil];
  59. [super dealloc];
  60. }
  61. - (void)setControllerToNotifyDone:(TestRunner*)controller
  62. {
  63. controllerToNotifyDone = controller;
  64. }
  65. @end