Service state file processing

cloud_print/service/win/service_state.cc

+// Copyright (c) 2012 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "cloud_print/service/win/service_state.h"
+
+#include "base/json/json_reader.h"
+#include "base/json/json_writer.h"
+#include "base/logging.h"
+#include "base/message_loop.h"
+#include "base/string_util.h"
+#include "base/utf_string_conversions.h"
+#include "net/base/escape.h"
+#include "net/base/io_buffer.h"
+#include "net/url_request/url_request.h"
+#include "net/url_request/url_request_context.h"
+#include "net/url_request/url_request_context_builder.h"
+
+namespace {
+
+const char kCloudPrintJsonName[] = "cloud_print";
+const char kEnabledOptionName[] = "enabled";
+
+const char kEmailOptionName[] = "email";
+const char kPasswordOptionName[] = "password";
+const char kProxyIdOptionName[] = "proxy_id";
+const char kRobotEmailOptionName[] = "robot_email";
+const char kRobotTokenOptionName[] = "robot_refresh_token";
+const char kAuthTokenOptionName[] = "auth_token";
+const char kXmppAuthTokenOptionName[] = "xmpp_auth_token";
+
+class ServiceStateURLRequestDelegate : public net::URLRequest::Delegate {
+ public:
+  virtual void OnResponseStarted(net::URLRequest* request) {
+    if (request->GetResponseCode() == 200)
+      Read(request);
+    request->Cancel();
+  };
+
+  virtual void OnReadCompleted(net::URLRequest* request, int bytes_read) {
+    MessageLoop::current()->Quit();
+  };
+
+  const std::string& data() const {
+    return data_;
+  }
+
+ private:
+  void Read(net::URLRequest* request) {
+    // Read as many bytes as are available synchronously.
+    const int kBufSize = 100000;
+    scoped_refptr<net::IOBuffer> buf(new net::IOBuffer(kBufSize));
+    while (true) {
+      int num_bytes = 0;
+      if (!request->Read(buf, kBufSize, &num_bytes)) {

[gene, 2012/05/21 23:08:06]

from url_request.h

  // Read() must be called at least once. Read() returns true when it completed
  // immediately, and false if an IO is pending or if there is an error.  When
  // Read() returns false, the caller can check the Request's status() to see
  // if an error occurred, or if the IO is just pending.  When Read() returns
  // true with zero bytes read, it indicates the end of the response.

[Vitaly Buka (NO REVIEWS), 2012/05/22 05:47:20]

Done.

+        return;
+      }
+      data_.append(buf->data(), buf->data() + num_bytes);
+    }
+  }
+  std::string data_;
+};
+
+
+void SetNotEmptyJsonString(base::DictionaryValue* dictionary,
+                           const std::string& name,
+                           const std::string& value) {
+  if (!value.empty())
+    dictionary->SetString(name, value);
+}
+
+}  // namespace
+
+ServiceState::OptionsDelegate::~OptionsDelegate() {
+}
+
+ServiceState::ServiceState() {
+  Reset();
+}
+
+ServiceState::~ServiceState() {
+}
+
+void ServiceState::Reset() {
+  email_.clear();
+  password_.clear();
+  proxy_id_.clear();
+  robot_email_.clear();
+  robot_token_.clear();
+  auth_token_.clear();
+  xmpp_auth_token_.clear();
+}
+
+bool ServiceState::FromString(const std::string& json) {
+  Reset();
+  scoped_ptr<base::Value> data(base::JSONReader::Read(json));
+  if (!data.get())
+    return false;
+
+  const base::DictionaryValue* services = NULL;
+  if (!data->GetAsDictionary(&services))
+    return false;
+
+  base::DictionaryValue* cloud_print = NULL;
+  if (!services->GetDictionary(kCloudPrintJsonName, &cloud_print))
+    return false;
+
+  bool valid_file = true;
+  // Don't exit on fail. Collect all data for re-use by user later.
+  if (!cloud_print->GetBoolean(kEnabledOptionName, &valid_file))
+    valid_file = false;
+
+  cloud_print->GetString(kEmailOptionName, &email_);
+  cloud_print->GetString(kProxyIdOptionName, &proxy_id_);
+  cloud_print->GetString(kRobotEmailOptionName, &robot_email_);
+  cloud_print->GetString(kRobotTokenOptionName, &robot_token_);
+  cloud_print->GetString(kAuthTokenOptionName, &auth_token_);
+  cloud_print->GetString(kXmppAuthTokenOptionName, &xmpp_auth_token_);
+
+  return valid_file && IsValid();
+}
+
+bool ServiceState::IsValid() const {
+  if (email_.empty() || proxy_id_.empty())
+    return false;
+  bool valid_robot = !robot_email_.empty() && !robot_token_.empty();
+  bool valid_auth = !auth_token_.empty() && !xmpp_auth_token_.empty();
+  return valid_robot || valid_auth;
+}
+
+std::string ServiceState::ToString() {
+  scoped_ptr<base::DictionaryValue> services(new DictionaryValue());
+
+  scoped_ptr<base::DictionaryValue> cloud_print(new DictionaryValue());
+  cloud_print->SetBoolean(kEnabledOptionName, true);
+
+  SetNotEmptyJsonString(cloud_print.get(), kEmailOptionName, email_);
+  SetNotEmptyJsonString(cloud_print.get(), kProxyIdOptionName, proxy_id_);
+  SetNotEmptyJsonString(cloud_print.get(), kRobotEmailOptionName, robot_email_);
+  SetNotEmptyJsonString(cloud_print.get(), kRobotTokenOptionName, robot_token_);
+  SetNotEmptyJsonString(cloud_print.get(), kAuthTokenOptionName, auth_token_);
+  SetNotEmptyJsonString(cloud_print.get(), kXmppAuthTokenOptionName,
+                        xmpp_auth_token_);
+
+  services->Set(kCloudPrintJsonName, cloud_print.release());
+
+  std::string json;
+  base::JSONWriter::WriteWithOptions(services.get(),
+                                     base::JSONWriter::OPTIONS_PRETTY_PRINT,
+                                     &json);
+  return json;
+}
+
+std::string ServiceState::LoginToGoogle(const std::string& service,
+                                        const std::string& email,
+                                        const std::string& password) {
+  MessageLoop loop(MessageLoop::TYPE_IO);
+
+  net::URLRequestContextBuilder builder;
+  scoped_ptr<net::URLRequestContext> context(builder.Build());
+
+  ServiceStateURLRequestDelegate fetcher_delegate;
+  GURL url("https://www.google.com/accounts/ClientLogin");

[gene, 2012/05/21 23:18:37]

Move this to consts in the begining of the file.

[Vitaly Buka (NO REVIEWS), 2012/05/22 05:47:20]

Done.

+
+  std::string post_body;
+  post_body += "accountType=GOOGLE";
+  post_body += "&Email=" + net::EscapeUrlEncodedData(email, true);
+  post_body += "&Passwd=" + net::EscapeUrlEncodedData(password, true);
+  post_body += "&source=" + net::EscapeUrlEncodedData("CP-Service", true);
+  post_body += "&service=" + net::EscapeUrlEncodedData(service, true);
+
+  net::URLRequest request(url, &fetcher_delegate);
+
+  request.AppendBytesToUpload(post_body.c_str(), post_body.size());
+  request.SetExtraRequestHeaderByName(
+      "Content-Type", "application/x-www-form-urlencoded", true);
+  request.set_context(context.get());
+  request.set_method("POST");
+  request.Start();
+
+  MessageLoop::current()->Run();
+
+  const char kAuthStart[] = "Auth=";
+  std::vector<std::string> lines;
+  Tokenize(fetcher_delegate.data(), "\r\n", &lines);
+  for (size_t i = 0; i < lines.size(); ++i) {
+    std::vector<std::string> tokens;
+    if (StartsWithASCII(lines[i], kAuthStart, false))
+      return lines[i].substr(arraysize(kAuthStart) - 1);
+  }
+
+  return std::string();
+}
+
+bool ServiceState::Configure(OptionsDelegate* delegate) {
+  if (!delegate) {
+    NOTREACHED();
+    return false;
+  }
+
+  robot_token_.clear();
+  robot_email_.clear();
+  email_ = delegate->GetOption(kEmailOptionName, email_, false);
+  password_ = delegate->GetOption(kPasswordOptionName, "", true);
+  proxy_id_ = delegate->GetOption(kProxyIdOptionName, proxy_id_, false);
+  auth_token_ = LoginToGoogle("cloudprint", email_, password_);
+  xmpp_auth_token_ = LoginToGoogle("chromiumsync", email_, password_);

[gene, 2012/05/21 23:18:37]

You need to clear robot accounts here, otherwise it will not work.

[Vitaly Buka (NO REVIEWS), 2012/05/22 05:47:20]

Done.

+
+  return IsValid();
+}
+