support traffic limit option.

Author: huiguangjun

doc/man/ossfs.1.in

@@ -316,6 +316,14 @@ Specify the path of the mime.types file.
 If this option is not specified, the existence of "/etc/mime.types" is checked, and that file is loaded as mime information.
 If this file does not exist on macOS, then "/etc/apache2/mime.types" is checked as well.
 .TP
+\fB\-o\fR upload_traffic_limit (default="0")
+The single-connection bandwidth throttling, in bit/s, for each upload request.
+The minimum value is 819200 bit/s and the maximum value is 838860800 bit/s. 0 value means disable.
+.TP
+\fB\-o\fR download_traffic_limit (default="0")
+The single-connection bandwidth throttling, in bit/s, for each download request.
+The minimum value is 819200 bit/s and the maximum value is 838860800 bit/s. 0 value means disable.
+.TP
 \fB\-o\fR logfile - specify the log output file.
 ossfs outputs the log file to syslog. Alternatively, if ossfs is started with the "-f" option specified, the log will be output to the stdout/stderr.
 You can use this option to specify the log file that ossfs outputs.

src/curl.cpp

@@ -120,6 +120,9 @@ bool             S3fsCurl::is_ua               = true;           // default
 bool             S3fsCurl::listobjectsv2       = false;          // default
 bool             S3fsCurl::requester_pays      = false;          // default
 
+long             S3fsCurl::upload_traffic_limit      = 0;          // default
+long             S3fsCurl::download_traffic_limit    = 0;          // default
+
 //-------------------------------------------------------------------
 // Class methods for S3fsCurl
 //-------------------------------------------------------------------
@@ -3344,6 +3347,11 @@ int S3fsCurl::PutRequest(const char* tpath, headers_t& meta, int fd)
         // set additional header by ahbe conf
         requestHeaders = AdditionalHeader::get()->AddHeader(requestHeaders, tpath);
     }
+    if(S3fsCurl::upload_traffic_limit != 0) {
+        char buff[64];
+        sprintf(buff, "%ld", S3fsCurl::upload_traffic_limit);
+        requestHeaders = curl_slist_sort_insert(requestHeaders, "x-oss-traffic-limit", buff);
+    }
 
     op = "PUT";
     type = REQTYPE_PUT;
@@ -3417,6 +3425,12 @@ int S3fsCurl::PreGetObjectRequest(const char* tpath, int fd, off_t start, off_t
         S3FS_PRN_WARN("Failed to set SSE header, but continue...");
     }
 
+    if(S3fsCurl::download_traffic_limit != 0) {
+        char buff[64];
+        sprintf(buff, "%ld", S3fsCurl::download_traffic_limit);
+        requestHeaders = curl_slist_sort_insert(requestHeaders, "x-oss-traffic-limit", buff);
+    }
+
     op = "GET";
     type = REQTYPE_GET;
 
@@ -3931,6 +3945,12 @@ int S3fsCurl::UploadMultipartPostSetup(const char* tpath, int part_num, const st
         }
     }
 
+    if(S3fsCurl::upload_traffic_limit != 0) {
+        char buff[64];
+        sprintf(buff, "%ld", S3fsCurl::upload_traffic_limit);
+        requestHeaders = curl_slist_sort_insert(requestHeaders, "x-oss-traffic-limit", buff);
+    }
+
     requestHeaders = curl_slist_sort_insert(requestHeaders, "Accept", NULL);
 
     op = "PUT";

src/curl.h

@@ -156,6 +156,8 @@ class S3fsCurl
         static bool             is_ua;             // User-Agent
         static bool             listobjectsv2;
         static bool             requester_pays;
+        static long             upload_traffic_limit;
+        static long             download_traffic_limit;
 
         // variables
         CURL*                hCurl;
@@ -334,6 +336,8 @@ class S3fsCurl
         static bool IsListObjectsV2() { return S3fsCurl::listobjectsv2; }
         static bool SetRequesterPays(bool flag) { bool old_flag = S3fsCurl::requester_pays; S3fsCurl::requester_pays = flag; return old_flag; }
         static bool IsRequesterPays() { return S3fsCurl::requester_pays; }
+        static bool SetUploadTrafficLimit(long value) { bool old_value = S3fsCurl::upload_traffic_limit; S3fsCurl::upload_traffic_limit = value; return old_value; }
+        static bool SetDownloadTrafficLimit(long value) { bool old_value = S3fsCurl::download_traffic_limit; S3fsCurl::download_traffic_limit = value; return old_value; }
 
         // methods
         bool CreateCurlHandle(bool only_pool = false, bool remake = false);

src/s3fs.cpp

@@ -4294,6 +4294,16 @@ static int my_fuse_opt_proc(void* data, const char* arg, int key, struct fuse_ar
             mimetype_file = strchr(arg, '=') + sizeof(char);
             return 0;
         }
+        if(is_prefix(arg, "upload_traffic_limit=")){
+            long traffic_limit = static_cast<long>(cvt_strtoofft(strchr(arg, '=') + sizeof(char), /*base=*/ 10));
+            S3fsCurl::SetUploadTrafficLimit(traffic_limit);
+            return 0;
+        }
+        if(is_prefix(arg, "download_traffic_limit=")){
+            long traffic_limit = static_cast<long>(cvt_strtoofft(strchr(arg, '=') + sizeof(char), /*base=*/ 10));
+            S3fsCurl::SetDownloadTrafficLimit(traffic_limit);
+            return 0;
+        }
         //
         // log file option
         //

src/s3fs_help.cpp

@@ -393,6 +393,16 @@ static const char help_string[] =
     "        If this file does not exist on macOS, then \"/etc/apache2/mime.types\"\n"
     "        is checked as well.\n"
     "\n"
+    "   upload_traffic_limit (default=\"0\")\n"
+    "        The single-connection bandwidth throttling, in bit/s, for each upload request.\n"
+    "        The minimum value is 819200 bit/s and the maximum value is 838860800 bit/s.\n"
+    "        0 value means disable.\n"
+    "\n"
+    "   download_traffic_limit (default=\"0\")\n"
+    "        The single-connection bandwidth throttling, in bit/s, for each download request.\n"
+    "        The minimum value is 819200 bit/s and the maximum value is 838860800 bit/s.\n"
+    "        0 value means disable.\n"
+    "\n"
     "   logfile - specify the log output file.\n"
     "        ossfs outputs the log file to syslog. Alternatively, if ossfs is\n"
     "        started with the \"-f\" option specified, the log will be output\n"