Index: cloud_print/virtual_driver/posix/virtual_driver_posix.cc |
=================================================================== |
--- cloud_print/virtual_driver/posix/virtual_driver_posix.cc (revision 92378) |
+++ cloud_print/virtual_driver/posix/virtual_driver_posix.cc (working copy) |
@@ -1,156 +0,0 @@ |
-// Copyright (c) 2011 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 <pwd.h> |
-#include <stdio.h> |
- |
-#include <cups/backend.h> |
- |
-#include <sys/stat.h> |
-#include <sys/types.h> |
-#include <sys/wait.h> |
- |
-#include "base/at_exit.h" |
-#include "base/base_paths.h" |
-#include "base/command_line.h" |
-#include "base/file_path.h" |
-#include "base/logging.h" |
-#include "base/path_service.h" |
- |
-#include "cloud_print/virtual_driver/posix/printer_driver_util_posix.h" |
- |
-using namespace std; |
- |
-int WriteToTemp(FILE* input_pdf, FilePath output_path) { |
- FILE* output_pdf; |
- char buffer[128]; |
- output_pdf = fopen(output_path.value().c_str(),"w"); |
- if(output_pdf == NULL) { |
- LOG(ERROR) <<"Unable to open file handle for writing output file"; |
- exit(CUPS_BACKEND_CANCEL); |
- } |
- // Read from input file or stdin and write to output file. |
- while(fgets(buffer,128,input_pdf) != NULL){ |
- fputs(buffer, output_pdf); |
- } |
- |
- LOG(INFO) << "Successfully wrote output file"; |
- // ensure everything is written, then close the files. |
- fflush(output_pdf); |
- fclose(input_pdf); |
- fclose(output_pdf); |
-} |
- |
-void SetUser(const char* user) { |
- struct passwd* calling_user=NULL; |
- calling_user=getpwnam(user); |
- if(calling_user == NULL) { |
- LOG(ERROR) << "Unable to get calling user"; |
- exit(CUPS_BACKEND_CANCEL); |
- } |
- if(setgid(calling_user->pw_gid)==-1) { |
- LOG(ERROR) << "Unable to set group ID"; |
- exit(CUPS_BACKEND_CANCEL); |
- } |
- if(!setuid(calling_user->pw_uid)==-1) { |
- LOG(ERROR) << "Unable to set UID"; |
- exit(CUPS_BACKEND_CANCEL); |
- } |
- |
- LOG(INFO) << "Successfully set user and group ID"; |
-} |
- |
- |
- |
-// Main function for backend. |
-int main(int argc, const char* argv[]) { |
- // With no arguments, send identification string as required by CUPS. |
- if (argc==1) { |
- printf("file GCP-driver:/ \"GCP Virtual Driver\" \"GCP Virtual Driver\" " |
- "\"MFG:Google Inc.;MDL:GCP Virtual Driver;DES:GCP Virtual Driver;" |
- "CLS:PRINTER;CMD:POSTSCRIPT;\"\n"); |
- return 0; |
- } |
- |
- if (argc<6 || argc>7) { |
- fprintf(stderr,"Usage: GCP-Driver job-id user" |
- "title copies options [file]\n"); |
- return 0; |
- } |
- |
- //AtExitManager is necessary to use the path service. |
- base::AtExitManager aemanager; |
- // File handler for the temporary PDF we pass onto Chrome. |
- FILE* output_pdf; |
- //CommandLine is only setup to enable logging, never used subseqeuently. |
- CommandLine::Init(argc,argv); |
- // Location for the log file. |
- std::string log_location="/var/log/GCP-Driver.log"; |
- |
- // Set up logging. |
- logging::InitLogging(log_location.c_str(), |
- logging::LOG_ONLY_TO_FILE, |
- logging::LOCK_LOG_FILE, |
- logging::APPEND_TO_OLD_LOG_FILE, |
- logging::DISABLE_DCHECK_FOR_NON_OFFICIAL_RELEASE_BUILDS); |
- |
- // Temporary directory to hold the output file. |
- FilePath temp_dir; |
- // Strings to hold details of current job. |
- std::string current_user; |
- std::string set_var; |
- std::string job_title; |
- std::string job_id; |
- std::string file_name; |
- |
- // Get temp directory to hold spool file. |
- if(!PathService::Get(base::DIR_TEMP, &temp_dir)) { |
- LOG(ERROR) << "Unable to get DIR_TEMP"; |
- return CUPS_BACKEND_CANCEL; |
- } |
- // Get details from command line and set spool path. |
- current_user = argv[2]; |
- job_title = argv[3]; |
- job_id = argv[1]; |
- file_name = current_user + "-" + job_title + "-" + job_id; |
- FilePath output_path = temp_dir.Append(file_name); |
- |
- // Read from file if specified. |
- if(argc==7){ |
- FILE* input_pdf=fopen(argv[6],"r"); |
- if(input_pdf == NULL) { |
- LOG(ERROR) << "Unable to read input PDF"; |
- return CUPS_BACKEND_CANCEL; |
- } |
- // File is opened. |
- WriteToTemp(input_pdf,output_path); |
- } |
- // Otherwise, read from stdin. |
- else{ |
- WriteToTemp(stdin, output_path); |
- } |
- |
- // Set File permissions to allow non-sudo user to read spool file. |
- if(chmod(output_path.value().c_str(),S_IRUSR|S_IWUSR|S_IROTH|S_IWOTH)!=0) { |
- LOG(ERROR)<<"Unable to change file permissions"; |
- return CUPS_BACKEND_CANCEL; |
- } |
- |
- pid_t pid=fork(); |
- |
- if(!pid) { |
- // In child process. |
- |
- // Set the user to the one that initiated print job. |
- SetUser(argv[2]); |
- |
- // Launch Chrome and pass the print job onto Cloud Print. |
- LaunchPrintDialog(output_path.value(), job_title, current_user); |
- |
- return 0; |
- }// back in parent process. |
- // wait for child, then terminate. |
- waitpid(pid,NULL,0); |
- return 0; |
-}// main |