michael@0: /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this file,
michael@0:  * You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 
michael@0: #include "AutoMounter.h"
michael@0: #include "AutoMounterSetting.h"
michael@0: 
michael@0: #include "base/message_loop.h"
michael@0: #include "jsapi.h"
michael@0: #include "mozilla/Services.h"
michael@0: #include "nsCOMPtr.h"
michael@0: #include "nsDebug.h"
michael@0: #include "nsIObserverService.h"
michael@0: #include "nsCxPusher.h"
michael@0: #include "nsISettingsService.h"
michael@0: #include "nsJSUtils.h"
michael@0: #include "nsPrintfCString.h"
michael@0: #include "nsServiceManagerUtils.h"
michael@0: #include "nsString.h"
michael@0: #include "nsThreadUtils.h"
michael@0: #include "xpcpublic.h"
michael@0: #include "mozilla/Attributes.h"
michael@0: 
michael@0: #undef LOG
michael@0: #define LOG(args...)  __android_log_print(ANDROID_LOG_INFO, "AutoMounterSetting" , ## args)
michael@0: #define ERR(args...)  __android_log_print(ANDROID_LOG_ERROR, "AutoMounterSetting" , ## args)
michael@0: 
michael@0: #define UMS_MODE                  "ums.mode"
michael@0: #define UMS_STATUS                "ums.status"
michael@0: #define UMS_VOLUME_ENABLED_PREFIX "ums.volume."
michael@0: #define UMS_VOLUME_ENABLED_SUFFIX ".enabled"
michael@0: #define MOZSETTINGS_CHANGED       "mozsettings-changed"
michael@0: 
michael@0: namespace mozilla {
michael@0: namespace system {
michael@0: 
michael@0: class SettingsServiceCallback MOZ_FINAL : public nsISettingsServiceCallback
michael@0: {
michael@0: public:
michael@0:   NS_DECL_THREADSAFE_ISUPPORTS
michael@0: 
michael@0:   SettingsServiceCallback() {}
michael@0: 
michael@0:   NS_IMETHOD Handle(const nsAString& aName, JS::Handle<JS::Value> aResult)
michael@0:   {
michael@0:     if (JSVAL_IS_INT(aResult)) {
michael@0:       int32_t mode = JSVAL_TO_INT(aResult);
michael@0:       SetAutoMounterMode(mode);
michael@0:     }
michael@0:     return NS_OK;
michael@0:   }
michael@0: 
michael@0:   NS_IMETHOD HandleError(const nsAString& aName)
michael@0:   {
michael@0:     ERR("SettingsCallback::HandleError: %s\n", NS_LossyConvertUTF16toASCII(aName).get());
michael@0:     return NS_OK;
michael@0:   }
michael@0: };
michael@0: 
michael@0: NS_IMPL_ISUPPORTS(SettingsServiceCallback, nsISettingsServiceCallback)
michael@0: 
michael@0: class CheckVolumeSettingsCallback MOZ_FINAL : public nsISettingsServiceCallback
michael@0: {
michael@0: public:
michael@0:   NS_DECL_THREADSAFE_ISUPPORTS
michael@0: 
michael@0:   CheckVolumeSettingsCallback(const nsACString& aVolumeName)
michael@0:   : mVolumeName(aVolumeName) {}
michael@0: 
michael@0:   NS_IMETHOD Handle(const nsAString& aName, JS::Handle<JS::Value> aResult)
michael@0:   {
michael@0:     if (JSVAL_IS_BOOLEAN(aResult)) {
michael@0:       bool isSharingEnabled = JSVAL_TO_BOOLEAN(aResult);
michael@0:       SetAutoMounterSharingMode(mVolumeName, isSharingEnabled);
michael@0:     }
michael@0:     return NS_OK;
michael@0:   }
michael@0: 
michael@0:   NS_IMETHOD HandleError(const nsAString& aName)
michael@0:   {
michael@0:     ERR("CheckVolumeSettingsCallback::HandleError: %s\n", NS_LossyConvertUTF16toASCII(aName).get());
michael@0:     return NS_OK;
michael@0:   }
michael@0: private:
michael@0:   nsCString mVolumeName;
michael@0: };
michael@0: 
michael@0: NS_IMPL_ISUPPORTS(CheckVolumeSettingsCallback, nsISettingsServiceCallback)
michael@0: 
michael@0: AutoMounterSetting::AutoMounterSetting()
michael@0:   : mStatus(AUTOMOUNTER_STATUS_DISABLED)
michael@0: {
michael@0:   MOZ_ASSERT(NS_IsMainThread());
michael@0: 
michael@0:   // Setup an observer to watch changes to the setting
michael@0:   nsCOMPtr<nsIObserverService> observerService =
michael@0:     mozilla::services::GetObserverService();
michael@0:   if (!observerService) {
michael@0:     ERR("GetObserverService failed");
michael@0:     return;
michael@0:   }
michael@0:   nsresult rv;
michael@0:   rv = observerService->AddObserver(this, MOZSETTINGS_CHANGED, false);
michael@0:   if (NS_FAILED(rv)) {
michael@0:     ERR("AddObserver failed");
michael@0:     return;
michael@0:   }
michael@0: 
michael@0:   // Force ums.mode to be 0 initially. We do this because settings are persisted.
michael@0:   // We don't want UMS to be enabled until such time as the phone is unlocked,
michael@0:   // and gaia/apps/system/js/storage.js takes care of detecting when the phone
michael@0:   // becomes unlocked and changes ums.mode appropriately.
michael@0:   nsCOMPtr<nsISettingsService> settingsService =
michael@0:     do_GetService("@mozilla.org/settingsService;1");
michael@0:   if (!settingsService) {
michael@0:     ERR("Failed to get settingsLock service!");
michael@0:     return;
michael@0:   }
michael@0:   nsCOMPtr<nsISettingsServiceLock> lock;
michael@0:   settingsService->CreateLock(nullptr, getter_AddRefs(lock));
michael@0:   nsCOMPtr<nsISettingsServiceCallback> callback = new SettingsServiceCallback();
michael@0:   mozilla::AutoSafeJSContext cx;
michael@0:   JS::Rooted<JS::Value> value(cx);
michael@0:   value.setInt32(AUTOMOUNTER_DISABLE);
michael@0:   lock->Set(UMS_MODE, value, callback, nullptr);
michael@0:   value.setInt32(mStatus);
michael@0:   lock->Set(UMS_STATUS, value, nullptr, nullptr);
michael@0: }
michael@0: 
michael@0: AutoMounterSetting::~AutoMounterSetting()
michael@0: {
michael@0:   nsCOMPtr<nsIObserverService> observerService =
michael@0:     mozilla::services::GetObserverService();
michael@0:   if (observerService) {
michael@0:     observerService->RemoveObserver(this, MOZSETTINGS_CHANGED);
michael@0:   }
michael@0: }
michael@0: 
michael@0: NS_IMPL_ISUPPORTS(AutoMounterSetting, nsIObserver)
michael@0: 
michael@0: const char *
michael@0: AutoMounterSetting::StatusStr(int32_t aStatus)
michael@0: {
michael@0:   switch (aStatus) {
michael@0:     case AUTOMOUNTER_STATUS_DISABLED:   return "Disabled";
michael@0:     case AUTOMOUNTER_STATUS_ENABLED:    return "Enabled";
michael@0:     case AUTOMOUNTER_STATUS_FILES_OPEN: return "FilesOpen";
michael@0:   }
michael@0:   return "??? Unknown ???";
michael@0: }
michael@0: 
michael@0: class CheckVolumeSettingsRunnable : public nsRunnable
michael@0: {
michael@0: public:
michael@0:   CheckVolumeSettingsRunnable(const nsACString& aVolumeName)
michael@0:     : mVolumeName(aVolumeName) {}
michael@0: 
michael@0:   NS_IMETHOD Run()
michael@0:   {
michael@0:     MOZ_ASSERT(NS_IsMainThread());
michael@0:     nsCOMPtr<nsISettingsService> settingsService =
michael@0:       do_GetService("@mozilla.org/settingsService;1");
michael@0:     NS_ENSURE_TRUE(settingsService, NS_ERROR_FAILURE);
michael@0:     nsCOMPtr<nsISettingsServiceLock> lock;
michael@0:     settingsService->CreateLock(nullptr, getter_AddRefs(lock));
michael@0:     nsCOMPtr<nsISettingsServiceCallback> callback =
michael@0:       new CheckVolumeSettingsCallback(mVolumeName);
michael@0:     nsPrintfCString setting(UMS_VOLUME_ENABLED_PREFIX "%s" UMS_VOLUME_ENABLED_SUFFIX,
michael@0:                             mVolumeName.get());
michael@0:     lock->Get(setting.get(), callback);
michael@0:     return NS_OK;
michael@0:   }
michael@0: 
michael@0: private:
michael@0:   nsCString mVolumeName;
michael@0: };
michael@0: 
michael@0: //static
michael@0: void
michael@0: AutoMounterSetting::CheckVolumeSettings(const nsACString& aVolumeName)
michael@0: {
michael@0:   NS_DispatchToMainThread(new CheckVolumeSettingsRunnable(aVolumeName));
michael@0: }
michael@0: 
michael@0: class SetStatusRunnable : public nsRunnable
michael@0: {
michael@0: public:
michael@0:   SetStatusRunnable(int32_t aStatus) : mStatus(aStatus) {}
michael@0: 
michael@0:   NS_IMETHOD Run()
michael@0:   {
michael@0:     MOZ_ASSERT(NS_IsMainThread());
michael@0:     nsCOMPtr<nsISettingsService> settingsService =
michael@0:       do_GetService("@mozilla.org/settingsService;1");
michael@0:     NS_ENSURE_TRUE(settingsService, NS_ERROR_FAILURE);
michael@0:     nsCOMPtr<nsISettingsServiceLock> lock;
michael@0:     settingsService->CreateLock(nullptr, getter_AddRefs(lock));
michael@0:     // lock may be null if this gets called during shutdown.
michael@0:     if (lock) {
michael@0:       mozilla::AutoSafeJSContext cx;
michael@0:       JS::Rooted<JS::Value> value(cx, JS::Int32Value(mStatus));
michael@0:       lock->Set(UMS_STATUS, value, nullptr, nullptr);
michael@0:     }
michael@0:     return NS_OK;
michael@0:   }
michael@0: 
michael@0: private:
michael@0:   int32_t mStatus;
michael@0: };
michael@0: 
michael@0: //static
michael@0: void
michael@0: AutoMounterSetting::SetStatus(int32_t aStatus)
michael@0: {
michael@0:   if (aStatus != mStatus) {
michael@0:     LOG("Changing status from '%s' to '%s'",
michael@0:         StatusStr(mStatus), StatusStr(aStatus));
michael@0:     mStatus = aStatus;
michael@0:     NS_DispatchToMainThread(new SetStatusRunnable(aStatus));
michael@0:   }
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: AutoMounterSetting::Observe(nsISupports* aSubject,
michael@0:                             const char* aTopic,
michael@0:                             const char16_t* aData)
michael@0: {
michael@0:   if (strcmp(aTopic, MOZSETTINGS_CHANGED) != 0) {
michael@0:     return NS_OK;
michael@0:   }
michael@0: 
michael@0:   // Note that this function gets called for any and all settings changes,
michael@0:   // so we need to carefully check if we have the one we're interested in.
michael@0:   //
michael@0:   // The string that we're interested in will be a JSON string that looks like:
michael@0:   //  {"key":"ums.autoMount","value":true}
michael@0: 
michael@0:   mozilla::AutoSafeJSContext cx;
michael@0:   nsDependentString dataStr(aData);
michael@0:   JS::Rooted<JS::Value> val(cx);
michael@0:   if (!JS_ParseJSON(cx, dataStr.get(), dataStr.Length(), &val) ||
michael@0:       !val.isObject()) {
michael@0:     return NS_OK;
michael@0:   }
michael@0:   JS::Rooted<JSObject*> obj(cx, &val.toObject());
michael@0:   JS::Rooted<JS::Value> key(cx);
michael@0:   if (!JS_GetProperty(cx, obj, "key", &key) ||
michael@0:       !key.isString()) {
michael@0:     return NS_OK;
michael@0:   }
michael@0: 
michael@0:   JSString *jsKey = JS::ToString(cx, key);
michael@0:   nsDependentJSString keyStr;
michael@0:   if (!keyStr.init(cx, jsKey)) {
michael@0:     return NS_OK;
michael@0:   }
michael@0: 
michael@0:   JS::Rooted<JS::Value> value(cx);
michael@0:   if (!JS_GetProperty(cx, obj, "value", &value)) {
michael@0:     return NS_OK;
michael@0:   }
michael@0: 
michael@0:   // Check for ums.mode changes
michael@0:   if (keyStr.EqualsLiteral(UMS_MODE)) {
michael@0:     if (!value.isInt32()) {
michael@0:       return NS_OK;
michael@0:     }
michael@0:     int32_t mode = value.toInt32();
michael@0:     SetAutoMounterMode(mode);
michael@0:     return NS_OK;
michael@0:   }
michael@0: 
michael@0:   // Check for ums.volume.NAME.enabled
michael@0:   if (StringBeginsWith(keyStr, NS_LITERAL_STRING(UMS_VOLUME_ENABLED_PREFIX)) &&
michael@0:       StringEndsWith(keyStr, NS_LITERAL_STRING(UMS_VOLUME_ENABLED_SUFFIX))) {
michael@0:     if (!value.isBoolean()) {
michael@0:       return NS_OK;
michael@0:     }
michael@0:     const size_t prefixLen = sizeof(UMS_VOLUME_ENABLED_PREFIX) - 1;
michael@0:     const size_t suffixLen = sizeof(UMS_VOLUME_ENABLED_SUFFIX) - 1;
michael@0:     nsDependentSubstring volumeName =
michael@0:       Substring(keyStr, prefixLen, keyStr.Length() - prefixLen - suffixLen);
michael@0:     bool isSharingEnabled = value.toBoolean();
michael@0:     SetAutoMounterSharingMode(NS_LossyConvertUTF16toASCII(volumeName), isSharingEnabled);
michael@0:     return NS_OK;
michael@0:   }
michael@0: 
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: }   // namespace system
michael@0: }   // namespace mozilla