Add size argument to LCAC client demo (#113)
diff --git a/demo/client.cc b/demo/client.cc
index e059c5e..646810a 100644
--- a/demo/client.cc
+++ b/demo/client.cc
@@ -40,6 +40,7 @@
 std::string tag = "dlp";
 bool threaded = false;
 std::string digest = "sha256-123456";
+uint64_t size = 16;
 std::string url = "https://upload.example.com";
 std::string email = "me@example.com";
 std::string machine_user = "DOMAIN\\me";
@@ -63,6 +64,7 @@
 constexpr const char* kArgMachineUser = "--machine-user=";
 constexpr const char* kArgPath = "--path=";
 constexpr const char* kArgRequestToken = "--request-token=";
+constexpr const char* kArgSize = "--size=";
 constexpr const char* kArgTag = "--tag=";
 constexpr const char* kArgThreaded = "--threaded";
 constexpr const char* kArgUrl = "--url=";
@@ -98,6 +100,8 @@
       threaded = true;
     } else if (arg.find(kArgDigest) == 0) {
       digest = arg.substr(strlen(kArgDigest));
+    } else if (arg.find(kArgSize) == 0) {
+      size = std::stoull(arg.substr(strlen(kArgSize)));
     } else if (arg.find(kArgUrl) == 0) {
       url = arg.substr(strlen(kArgUrl));
     } else if (arg.find(kArgMachineUser) == 0) {
@@ -145,6 +149,7 @@
     << kArgPath << " <path> : Used the specified path instead of default. Must come after --user." << std::endl
     << kArgUserSpecific << " : Connects to an OS user specific agent" << std::endl
     << kArgDigest << "<digest> : defaults to 'sha256-123456'" << std::endl
+    << kArgSize << "<size> : file size, defaults to '16'" << std::endl
     << kArgGroup << " : Generate the same final action for all requests" << std::endl
     << kArgHelp << " : prints this help message" << std::endl;
 }
@@ -176,6 +181,7 @@
   request_data->set_url(url);
   request_data->set_email(email);
   request_data->set_digest(digest);
+  request_data->set_file_size(size);
   if (!filename.empty()) {
     request_data->set_filename(filename);
   }