This source file includes following definitions.
- stats_ptr_factory_
- InitPrintSystem
- ScheduleStatsReport
- ReportStats
- Start
- Stop
- IsRunning
- GetPrinterIds
- RegisterPrinters
- CheckForJobs
- UpdatePrinterSettings
- OnPrinterAdded
- OnPrinterDeleted
- OnAuthError
- HandleRawData
- HandleJSONData
- OnRequestAuthError
- GetAuthHeader
- HandlePrinterListResponse
- HandlePrinterListResponseSettingsUpdate
- HandlePrinterDeleteResponse
- HandleRegisterPrinterResponse
- StartGetRequest
- StartPostRequest
- ReportUserMessage
- RemovePrinterFromList
- InitJobHandlerForPrinter
- UpdateSettingsFromPrintersList
- AddPendingAvailableTask
- AddPendingDeleteTask
- AddPendingRegisterTask
- AddPendingTask
- ProcessPendingTask
- ContinuePendingTaskProcessing
- OnPrintersAvailable
- OnPrinterRegister
- OnPrinterDelete
- OnReceivePrinterCaps
- IsSamePrinter
#include "chrome/service/cloud_print/cloud_print_connector.h"
#include "base/bind.h"
#include "base/bind_helpers.h"
#include "base/md5.h"
#include "base/rand_util.h"
#include "base/strings/string_number_conversions.h"
#include "base/strings/string_split.h"
#include "base/strings/stringprintf.h"
#include "base/strings/utf_string_conversions.h"
#include "base/values.h"
#include "chrome/common/cloud_print/cloud_print_constants.h"
#include "chrome/common/cloud_print/cloud_print_helpers.h"
#include "chrome/service/cloud_print/cloud_print_service_helpers.h"
#include "grit/generated_resources.h"
#include "net/base/mime_util.h"
#include "ui/base/l10n/l10n_util.h"
namespace cloud_print {
CloudPrintConnector::CloudPrintConnector(Client* client,
const ConnectorSettings& settings)
: client_(client),
next_response_handler_(NULL),
stats_ptr_factory_(this) {
settings_.CopyFrom(settings);
}
bool CloudPrintConnector::InitPrintSystem() {
if (print_system_.get())
return true;
print_system_ = PrintSystem::CreateInstance(
settings_.print_system_settings());
if (!print_system_.get()) {
NOTREACHED();
return false;
}
PrintSystem::PrintSystemResult result = print_system_->Init();
if (!result.succeeded()) {
print_system_ = NULL;
ReportUserMessage(kPrintSystemFailedMessageId, result.message());
return false;
}
return true;
}
void CloudPrintConnector::ScheduleStatsReport() {
base::MessageLoop::current()->PostDelayedTask(
FROM_HERE,
base::Bind(&CloudPrintConnector::ReportStats,
stats_ptr_factory_.GetWeakPtr()),
base::TimeDelta::FromHours(1));
}
void CloudPrintConnector::ReportStats() {
PrinterJobHandler::ReportsStats();
ScheduleStatsReport();
}
bool CloudPrintConnector::Start() {
VLOG(1) << "CP_CONNECTOR: Starting connector"
<< ", proxy id: " << settings_.proxy_id();
pending_tasks_.clear();
if (!InitPrintSystem())
return false;
ScheduleStatsReport();
print_server_watcher_ = print_system_->CreatePrintServerWatcher();
print_server_watcher_->StartWatching(this);
AddPendingAvailableTask();
return true;
}
void CloudPrintConnector::Stop() {
VLOG(1) << "CP_CONNECTOR: Stopping connector"
<< ", proxy id: " << settings_.proxy_id();
DCHECK(IsRunning());
stats_ptr_factory_.InvalidateWeakPtrs();
pending_tasks_.clear();
print_server_watcher_ = NULL;
request_ = NULL;
}
bool CloudPrintConnector::IsRunning() {
return print_server_watcher_.get() != NULL;
}
void CloudPrintConnector::GetPrinterIds(std::list<std::string>* printer_ids) {
DCHECK(printer_ids);
printer_ids->clear();
for (JobHandlerMap::const_iterator iter = job_handler_map_.begin();
iter != job_handler_map_.end(); ++iter) {
printer_ids->push_back(iter->first);
}
}
void CloudPrintConnector::RegisterPrinters(
const printing::PrinterList& printers) {
if (!IsRunning())
return;
printing::PrinterList::const_iterator it;
for (it = printers.begin(); it != printers.end(); ++it) {
if (settings_.ShouldConnect(it->printer_name))
AddPendingRegisterTask(*it);
}
}
void CloudPrintConnector::CheckForJobs(const std::string& reason,
const std::string& printer_id) {
if (!IsRunning())
return;
if (!printer_id.empty()) {
JobHandlerMap::iterator index = job_handler_map_.find(printer_id);
if (index != job_handler_map_.end()) {
index->second->CheckForJobs(reason);
} else {
std::string status_message = l10n_util::GetStringUTF8(
IDS_CLOUD_PRINT_ZOMBIE_PRINTER);
LOG(ERROR) << "CP_CONNECTOR: " << status_message <<
" Printer_id: " << printer_id;
ReportUserMessage(kZombiePrinterMessageId, status_message);
}
} else {
for (JobHandlerMap::iterator index = job_handler_map_.begin();
index != job_handler_map_.end(); index++) {
index->second->CheckForJobs(reason);
}
}
}
void CloudPrintConnector::UpdatePrinterSettings(const std::string& printer_id) {
GURL printer_list_url = GetUrlForPrinterList(
settings_.server_url(), settings_.proxy_id());
StartGetRequest(
printer_list_url,
kCloudPrintRegisterMaxRetryCount,
&CloudPrintConnector::HandlePrinterListResponseSettingsUpdate);
}
void CloudPrintConnector::OnPrinterAdded() {
AddPendingAvailableTask();
}
void CloudPrintConnector::OnPrinterDeleted(const std::string& printer_id) {
AddPendingDeleteTask(printer_id);
}
void CloudPrintConnector::OnAuthError() {
client_->OnAuthFailed();
}
CloudPrintURLFetcher::ResponseAction CloudPrintConnector::HandleRawData(
const net::URLFetcher* source,
const GURL& url,
const std::string& data) {
if (user_message_request_.get() &&
user_message_request_->IsSameRequest(source))
return CloudPrintURLFetcher::STOP_PROCESSING;
return CloudPrintURLFetcher::CONTINUE_PROCESSING;
}
CloudPrintURLFetcher::ResponseAction CloudPrintConnector::HandleJSONData(
const net::URLFetcher* source,
const GURL& url,
base::DictionaryValue* json_data,
bool succeeded) {
if (!IsRunning())
return CloudPrintURLFetcher::STOP_PROCESSING;
DCHECK(next_response_handler_);
return (this->*next_response_handler_)(source, url, json_data, succeeded);
}
CloudPrintURLFetcher::ResponseAction CloudPrintConnector::OnRequestAuthError() {
OnAuthError();
return CloudPrintURLFetcher::STOP_PROCESSING;
}
std::string CloudPrintConnector::GetAuthHeader() {
return GetCloudPrintAuthHeaderFromStore();
}
CloudPrintConnector::~CloudPrintConnector() {}
CloudPrintURLFetcher::ResponseAction
CloudPrintConnector::HandlePrinterListResponse(
const net::URLFetcher* source,
const GURL& url,
base::DictionaryValue* json_data,
bool succeeded) {
DCHECK(succeeded);
if (!succeeded)
return CloudPrintURLFetcher::RETRY_REQUEST;
UpdateSettingsFromPrintersList(json_data);
printing::PrinterList local_printers;
PrintSystem::PrintSystemResult result =
print_system_->EnumeratePrinters(&local_printers);
bool full_list = result.succeeded();
if (!full_list) {
std::string message = result.message();
if (message.empty())
message = l10n_util::GetStringFUTF8(IDS_CLOUD_PRINT_ENUM_FAILED,
l10n_util::GetStringUTF16(IDS_GOOGLE_CLOUD_PRINT));
ReportUserMessage(kEnumPrintersFailedMessageId, message);
}
base::ListValue* printer_list = NULL;
if (json_data->GetList(kPrinterListValue, &printer_list) && printer_list) {
for (size_t index = 0; index < printer_list->GetSize(); index++) {
base::DictionaryValue* printer_data = NULL;
if (printer_list->GetDictionary(index, &printer_data)) {
std::string printer_name;
printer_data->GetString(kNameValue, &printer_name);
std::string printer_id;
printer_data->GetString(kIdValue, &printer_id);
if (!settings_.ShouldConnect(printer_name)) {
VLOG(1) << "CP_CONNECTOR: Deleting " << printer_name <<
" id: " << printer_id << " as blacklisted";
AddPendingDeleteTask(printer_id);
} else if (RemovePrinterFromList(printer_name, &local_printers)) {
InitJobHandlerForPrinter(printer_data);
} else {
if (full_list || settings_.delete_on_enum_fail()) {
VLOG(1) << "CP_CONNECTOR: Deleting " << printer_name <<
" id: " << printer_id <<
" full_list: " << full_list <<
" delete_on_enum_fail: " << settings_.delete_on_enum_fail();
AddPendingDeleteTask(printer_id);
} else {
LOG(ERROR) << "CP_CONNECTOR: Printer: " << printer_name <<
" id: " << printer_id <<
" not found in print system and full printer list was" <<
" not received. Printer will not be able to process" <<
" jobs.";
}
}
} else {
NOTREACHED();
}
}
}
request_ = NULL;
RegisterPrinters(local_printers);
ContinuePendingTaskProcessing();
return CloudPrintURLFetcher::STOP_PROCESSING;
}
CloudPrintURLFetcher::ResponseAction
CloudPrintConnector::HandlePrinterListResponseSettingsUpdate(
const net::URLFetcher* source,
const GURL& url,
base::DictionaryValue* json_data,
bool succeeded) {
DCHECK(succeeded);
if (!succeeded)
return CloudPrintURLFetcher::RETRY_REQUEST;
UpdateSettingsFromPrintersList(json_data);
return CloudPrintURLFetcher::STOP_PROCESSING;
}
CloudPrintURLFetcher::ResponseAction
CloudPrintConnector::HandlePrinterDeleteResponse(
const net::URLFetcher* source,
const GURL& url,
base::DictionaryValue* json_data,
bool succeeded) {
VLOG(1) << "CP_CONNECTOR: Handler printer delete response"
<< ", succeeded: " << succeeded
<< ", url: " << url;
ContinuePendingTaskProcessing();
return CloudPrintURLFetcher::STOP_PROCESSING;
}
CloudPrintURLFetcher::ResponseAction
CloudPrintConnector::HandleRegisterPrinterResponse(
const net::URLFetcher* source,
const GURL& url,
base::DictionaryValue* json_data,
bool succeeded) {
VLOG(1) << "CP_CONNECTOR: Handler printer register response"
<< ", succeeded: " << succeeded
<< ", url: " << url;
if (succeeded) {
base::ListValue* printer_list = NULL;
if (json_data->GetList(kPrinterListValue, &printer_list)) {
base::DictionaryValue* printer_data = NULL;
if (printer_list->GetDictionary(0, &printer_data))
InitJobHandlerForPrinter(printer_data);
}
}
ContinuePendingTaskProcessing();
return CloudPrintURLFetcher::STOP_PROCESSING;
}
void CloudPrintConnector::StartGetRequest(const GURL& url,
int max_retries,
ResponseHandler handler) {
next_response_handler_ = handler;
request_ = CloudPrintURLFetcher::Create();
request_->StartGetRequest(CloudPrintURLFetcher::REQUEST_UPDATE_JOB,
url, this, max_retries, std::string());
}
void CloudPrintConnector::StartPostRequest(
CloudPrintURLFetcher::RequestType type,
const GURL& url,
int max_retries,
const std::string& mime_type,
const std::string& post_data,
ResponseHandler handler) {
next_response_handler_ = handler;
request_ = CloudPrintURLFetcher::Create();
request_->StartPostRequest(
type, url, this, max_retries, mime_type, post_data, std::string());
}
void CloudPrintConnector::ReportUserMessage(const std::string& message_id,
const std::string& failure_msg) {
std::string mime_boundary;
CreateMimeBoundaryForUpload(&mime_boundary);
GURL url = GetUrlForUserMessage(settings_.server_url(), message_id);
std::string post_data;
net::AddMultipartValueForUpload(kMessageTextValue, failure_msg, mime_boundary,
std::string(), &post_data);
net::AddMultipartFinalDelimiterForUpload(mime_boundary, &post_data);
std::string mime_type("multipart/form-data; boundary=");
mime_type += mime_boundary;
user_message_request_ = CloudPrintURLFetcher::Create();
user_message_request_->StartPostRequest(
CloudPrintURLFetcher::REQUEST_USER_MESSAGE, url, this, 1, mime_type,
post_data, std::string());
}
bool CloudPrintConnector::RemovePrinterFromList(
const std::string& printer_name,
printing::PrinterList* printer_list) {
for (printing::PrinterList::iterator index = printer_list->begin();
index != printer_list->end(); index++) {
if (IsSamePrinter(index->printer_name, printer_name)) {
index = printer_list->erase(index);
return true;
}
}
return false;
}
void CloudPrintConnector::InitJobHandlerForPrinter(
base::DictionaryValue* printer_data) {
DCHECK(printer_data);
PrinterJobHandler::PrinterInfoFromCloud printer_info_cloud;
printer_data->GetString(kIdValue, &printer_info_cloud.printer_id);
DCHECK(!printer_info_cloud.printer_id.empty());
VLOG(1) << "CP_CONNECTOR: Init job handler"
<< ", printer id: " << printer_info_cloud.printer_id;
JobHandlerMap::iterator index = job_handler_map_.find(
printer_info_cloud.printer_id);
if (index != job_handler_map_.end())
return;
printing::PrinterBasicInfo printer_info;
printer_data->GetString(kNameValue, &printer_info.printer_name);
DCHECK(!printer_info.printer_name.empty());
printer_data->GetString(kPrinterDescValue,
&printer_info.printer_description);
std::string printer_status;
if (printer_data->GetString(kPrinterStatusValue, &printer_status)) {
base::StringToInt(printer_status, &printer_info.printer_status);
}
printer_data->GetString(kPrinterCapsHashValue,
&printer_info_cloud.caps_hash);
base::ListValue* tags_list = NULL;
if (printer_data->GetList(kTagsValue, &tags_list) && tags_list) {
for (size_t index = 0; index < tags_list->GetSize(); index++) {
std::string tag;
if (tags_list->GetString(index, &tag) &&
StartsWithASCII(tag, kCloudPrintServiceTagsHashTagName, false)) {
std::vector<std::string> tag_parts;
base::SplitStringDontTrim(tag, '=', &tag_parts);
DCHECK_EQ(tag_parts.size(), 2U);
if (tag_parts.size() == 2)
printer_info_cloud.tags_hash = tag_parts[1];
}
}
}
int xmpp_timeout = 0;
printer_data->GetInteger(kLocalSettingsPendingXmppValue, &xmpp_timeout);
printer_info_cloud.current_xmpp_timeout = settings_.xmpp_ping_timeout_sec();
printer_info_cloud.pending_xmpp_timeout = xmpp_timeout;
scoped_refptr<PrinterJobHandler> job_handler;
job_handler = new PrinterJobHandler(printer_info,
printer_info_cloud,
settings_.server_url(),
print_system_.get(),
this);
job_handler_map_[printer_info_cloud.printer_id] = job_handler;
job_handler->Initialize();
}
void CloudPrintConnector::UpdateSettingsFromPrintersList(
base::DictionaryValue* json_data) {
base::ListValue* printer_list = NULL;
int min_xmpp_timeout = std::numeric_limits<int>::max();
if (json_data->GetList(kPrinterListValue, &printer_list) && printer_list) {
for (size_t index = 0; index < printer_list->GetSize(); index++) {
base::DictionaryValue* printer_data = NULL;
if (printer_list->GetDictionary(index, &printer_data)) {
int xmpp_timeout = 0;
if (printer_data->GetInteger(kLocalSettingsPendingXmppValue,
&xmpp_timeout)) {
min_xmpp_timeout = std::min(xmpp_timeout, min_xmpp_timeout);
}
}
}
}
if (min_xmpp_timeout != std::numeric_limits<int>::max()) {
DCHECK(min_xmpp_timeout >= kMinXmppPingTimeoutSecs);
settings_.SetXmppPingTimeoutSec(min_xmpp_timeout);
client_->OnXmppPingUpdated(min_xmpp_timeout);
}
}
void CloudPrintConnector::AddPendingAvailableTask() {
PendingTask task;
task.type = PENDING_PRINTERS_AVAILABLE;
AddPendingTask(task);
}
void CloudPrintConnector::AddPendingDeleteTask(const std::string& id) {
PendingTask task;
task.type = PENDING_PRINTER_DELETE;
task.printer_id = id;
AddPendingTask(task);
}
void CloudPrintConnector::AddPendingRegisterTask(
const printing::PrinterBasicInfo& info) {
PendingTask task;
task.type = PENDING_PRINTER_REGISTER;
task.printer_info = info;
AddPendingTask(task);
}
void CloudPrintConnector::AddPendingTask(const PendingTask& task) {
pending_tasks_.push_back(task);
if (pending_tasks_.size() == 1) {
base::MessageLoop::current()->PostTask(
FROM_HERE, base::Bind(&CloudPrintConnector::ProcessPendingTask, this));
}
}
void CloudPrintConnector::ProcessPendingTask() {
if (!IsRunning())
return;
if (pending_tasks_.size() == 0)
return;
PendingTask task = pending_tasks_.front();
switch (task.type) {
case PENDING_PRINTERS_AVAILABLE :
OnPrintersAvailable();
break;
case PENDING_PRINTER_REGISTER :
OnPrinterRegister(task.printer_info);
break;
case PENDING_PRINTER_DELETE :
OnPrinterDelete(task.printer_id);
break;
default:
NOTREACHED();
}
}
void CloudPrintConnector::ContinuePendingTaskProcessing() {
if (pending_tasks_.size() == 0)
return;
pending_tasks_.pop_front();
if (pending_tasks_.size() != 0) {
base::MessageLoop::current()->PostTask(
FROM_HERE, base::Bind(&CloudPrintConnector::ProcessPendingTask, this));
}
}
void CloudPrintConnector::OnPrintersAvailable() {
GURL printer_list_url = GetUrlForPrinterList(
settings_.server_url(), settings_.proxy_id());
StartGetRequest(printer_list_url,
kCloudPrintRegisterMaxRetryCount,
&CloudPrintConnector::HandlePrinterListResponse);
}
void CloudPrintConnector::OnPrinterRegister(
const printing::PrinterBasicInfo& info) {
for (JobHandlerMap::iterator it = job_handler_map_.begin();
it != job_handler_map_.end(); ++it) {
if (IsSamePrinter(it->second->GetPrinterName(), info.printer_name)) {
ContinuePendingTaskProcessing();
return;
}
}
print_system_->GetPrinterCapsAndDefaults(
info.printer_name.c_str(),
base::Bind(&CloudPrintConnector::OnReceivePrinterCaps,
base::Unretained(this)));
}
void CloudPrintConnector::OnPrinterDelete(const std::string& printer_id) {
JobHandlerMap::iterator it = job_handler_map_.find(printer_id);
if (it != job_handler_map_.end()) {
it->second->Shutdown();
job_handler_map_.erase(it);
}
GURL url = GetUrlForPrinterDelete(
settings_.server_url(), printer_id, "printer_deleted");
StartGetRequest(url,
kCloudPrintAPIMaxRetryCount,
&CloudPrintConnector::HandlePrinterDeleteResponse);
}
void CloudPrintConnector::OnReceivePrinterCaps(
bool succeeded,
const std::string& printer_name,
const printing::PrinterCapsAndDefaults& caps_and_defaults) {
if (!IsRunning())
return;
DCHECK(pending_tasks_.size() > 0 &&
pending_tasks_.front().type == PENDING_PRINTER_REGISTER);
if (!succeeded) {
LOG(ERROR) << "CP_CONNECTOR: Failed to get printer info"
<< ", printer name: " << printer_name;
base::string16 printer_name_utf16 = base::UTF8ToUTF16(printer_name);
std::string status_message = l10n_util::GetStringFUTF8(
IDS_CLOUD_PRINT_REGISTER_PRINTER_FAILED,
printer_name_utf16,
l10n_util::GetStringUTF16(IDS_GOOGLE_CLOUD_PRINT));
ReportUserMessage(kGetPrinterCapsFailedMessageId, status_message);
ContinuePendingTaskProcessing();
return;
}
const printing::PrinterBasicInfo& info = pending_tasks_.front().printer_info;
DCHECK(IsSamePrinter(info.printer_name, printer_name));
std::string mime_boundary;
CreateMimeBoundaryForUpload(&mime_boundary);
std::string post_data;
net::AddMultipartValueForUpload(kProxyIdValue,
settings_.proxy_id(), mime_boundary, std::string(), &post_data);
net::AddMultipartValueForUpload(kPrinterNameValue,
info.printer_name, mime_boundary, std::string(), &post_data);
net::AddMultipartValueForUpload(kPrinterDescValue,
info.printer_description, mime_boundary, std::string(), &post_data);
net::AddMultipartValueForUpload(kPrinterStatusValue,
base::StringPrintf("%d", info.printer_status),
mime_boundary, std::string(), &post_data);
net::AddMultipartValueForUpload(kPrinterLocalSettingsValue,
base::StringPrintf(kCreateLocalSettingsXmppPingFormat,
settings_.xmpp_ping_timeout_sec()),
mime_boundary, std::string(), &post_data);
post_data += GetPostDataForPrinterInfo(info, mime_boundary);
if (caps_and_defaults.caps_mime_type == kContentTypeJSON) {
net::AddMultipartValueForUpload(kUseCDD, "true", mime_boundary,
std::string(), &post_data);
}
net::AddMultipartValueForUpload(kPrinterCapsValue,
caps_and_defaults.printer_capabilities, mime_boundary,
caps_and_defaults.caps_mime_type, &post_data);
net::AddMultipartValueForUpload(kPrinterDefaultsValue,
caps_and_defaults.printer_defaults, mime_boundary,
caps_and_defaults.defaults_mime_type, &post_data);
net::AddMultipartValueForUpload(kPrinterCapsHashValue,
base::MD5String(caps_and_defaults.printer_capabilities),
mime_boundary, std::string(), &post_data);
net::AddMultipartFinalDelimiterForUpload(mime_boundary, &post_data);
std::string mime_type("multipart/form-data; boundary=");
mime_type += mime_boundary;
GURL post_url = GetUrlForPrinterRegistration(settings_.server_url());
StartPostRequest(CloudPrintURLFetcher::REQUEST_REGISTER, post_url,
kCloudPrintAPIMaxRetryCount, mime_type, post_data,
&CloudPrintConnector::HandleRegisterPrinterResponse);
}
bool CloudPrintConnector::IsSamePrinter(const std::string& name1,
const std::string& name2) const {
return (0 == base::strcasecmp(name1.c_str(), name2.c_str()));
}
}