xpcom/ds/nsProperties.cpp

   1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
   2 /* This Source Code Form is subject to the terms of the Mozilla Public
   3  * License, v. 2.0. If a copy of the MPL was not distributed with this
   4  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   5
   6 #include "nsProperties.h"
   7
   8 ////////////////////////////////////////////////////////////////////////////////
   9
  10 NS_IMPL_AGGREGATED(nsProperties)
  11 NS_INTERFACE_MAP_BEGIN_AGGREGATED(nsProperties)
  12   NS_INTERFACE_MAP_ENTRY(nsIProperties)
  13 NS_INTERFACE_MAP_END
  14
  15 NS_IMETHODIMP
  16 nsProperties::Get(const char* prop, const nsIID& uuid, void** result)
  17 {
  18   if (NS_WARN_IF(!prop)) {
  19     return NS_ERROR_INVALID_ARG;
  20   }
  21
  22   nsCOMPtr<nsISupports> value;
  23   if (!nsProperties_HashBase::Get(prop, getter_AddRefs(value))) {
  24     return NS_ERROR_FAILURE;
  25   }
  26   return (value) ? value->QueryInterface(uuid, result) : NS_ERROR_NO_INTERFACE;
  27 }
  28
  29 NS_IMETHODIMP
  30 nsProperties::Set(const char* prop, nsISupports* value)
  31 {
  32   if (NS_WARN_IF(!prop)) {
  33     return NS_ERROR_INVALID_ARG;
  34   }
  35   Put(prop, value);
  36   return NS_OK;
  37 }
  38
  39 NS_IMETHODIMP
  40 nsProperties::Undefine(const char* prop)
  41 {
  42   if (NS_WARN_IF(!prop)) {
  43     return NS_ERROR_INVALID_ARG;
  44   }
  45
  46   nsCOMPtr<nsISupports> value;
  47   if (!nsProperties_HashBase::Get(prop, getter_AddRefs(value))) {
  48     return NS_ERROR_FAILURE;
  49   }
  50
  51   Remove(prop);
  52   return NS_OK;
  53 }
  54
  55 NS_IMETHODIMP
  56 nsProperties::Has(const char* prop, bool* result)
  57 {
  58   if (NS_WARN_IF(!prop)) {
  59     return NS_ERROR_INVALID_ARG;
  60   }
  61
  62   nsCOMPtr<nsISupports> value;
  63   *result = nsProperties_HashBase::Get(prop, getter_AddRefs(value));
  64   return NS_OK;
  65 }
  66
  67 struct GetKeysEnumData
  68 {
  69   char** keys;
  70   uint32_t next;
  71   nsresult res;
  72 };
  73
  74 PLDHashOperator
  75 GetKeysEnumerate(const char* aKey, nsISupports* aData, void* aArg)
  76 {
  77   GetKeysEnumData* gkedp = (GetKeysEnumData*)aArg;
  78   gkedp->keys[gkedp->next] = strdup(aKey);
  79
  80   if (!gkedp->keys[gkedp->next]) {
  81     gkedp->res = NS_ERROR_OUT_OF_MEMORY;
  82     return PL_DHASH_STOP;
  83   }
  84
  85   gkedp->next++;
  86   return PL_DHASH_NEXT;
  87 }
  88
  89 NS_IMETHODIMP
  90 nsProperties::GetKeys(uint32_t* aCount, char*** aKeys)
  91 {
  92   if (NS_WARN_IF(!aCount) || NS_WARN_IF(!aKeys)) {
  93     return NS_ERROR_INVALID_ARG;
  94   }
  95
  96   uint32_t n = Count();
  97   char** k = (char**)nsMemory::Alloc(n * sizeof(char*));
  98
  99   GetKeysEnumData gked;
 100   gked.keys = k;
 101   gked.next = 0;
 102   gked.res = NS_OK;
 103
 104   EnumerateRead(GetKeysEnumerate, &gked);
 105
 106   nsresult rv = gked.res;
 107   if (NS_FAILED(rv)) {
 108     // Free 'em all
 109     for (uint32_t i = 0; i < gked.next; i++) {
 110       nsMemory::Free(k[i]);
 111     }
 112     nsMemory::Free(k);
 113     return rv;
 114   }
 115
 116   *aCount = n;
 117   *aKeys = k;
 118   return NS_OK;
 119 }
 120
 121 ////////////////////////////////////////////////////////////////////////////////