diff --git a/browser/src/client_win.cc b/browser/src/client_win.cc index 9d3d7e8..427ba8c 100644 --- a/browser/src/client_win.cc +++ b/browser/src/client_win.cc @@ -192,6 +192,8 @@ std::vector ReadNextMessageFromPipe( break; } + final_size += read; + // Reaching here means the error is "more data", that is, the buffer // specified in ReadFile() was too small to contain the entire response // message from the server. ReadFile() has placed the start of the diff --git a/demo/client.cc b/demo/client.cc index 84ca6e2..e059c5e 100644 --- a/demo/client.cc +++ b/demo/client.cc @@ -84,6 +84,8 @@ bool ParseCommandLine(int argc, char* argv[]) { connector = content_analysis::sdk::PRINT; } else if (connector_str == "file-transfer") { connector = content_analysis::sdk::FILE_TRANSFER; + } else if (connector_str == "data-copied") { + connector = content_analysis::sdk::DATA_COPIED; } else { std::cout << "[Demo] Incorrect command line arg: " << arg << std::endl; return false; @@ -133,7 +135,7 @@ void PrintHelp() { << "Otherwise the content is read from a file called 'content_or_file'." << std::endl << "Multiple [@]content_or_file arguments may be specified, each generates one request." << std::endl << std::endl << "Options:" << std::endl - << kArgConnector << " : one of 'download', 'attach' (default), 'bulk-data-entry', 'print', or 'file-transfer'" << std::endl + << kArgConnector << " : one of 'download', 'attach' (default), 'bulk-data-entry', 'print', 'file-transfer', or 'data-copied'" << std::endl << kArgRequestToken << " : defaults to 'req-' which auto increments" << std::endl << kArgTag << " : defaults to 'dlp'" << std::endl << kArgThreaded << " : handled multiple requests using threads" << std::endl diff --git a/demo/handler.h b/demo/handler.h index 9d1ccfd..96a4e20 100644 --- a/demo/handler.h +++ b/demo/handler.h @@ -208,6 +208,9 @@ class Handler : public content_analysis::sdk::AgentEventHandler { case content_analysis::sdk::FILE_TRANSFER: connector = "file-transfer"; break; + case content_analysis::sdk::DATA_COPIED: + connector = "data-copied"; + break; default: break; } @@ -240,6 +243,9 @@ class Handler : public content_analysis::sdk::AgentEventHandler { case content_analysis::sdk::ContentAnalysisRequest::SAVE_AS_DOWNLOAD: reason = "SAVE_AS_DOWNLOAD"; break; + case content_analysis::sdk::ContentAnalysisRequest::CLIPBOARD_COPY: + reason = "CLIPBOARD_COPY"; + break; } }