get_file_size(): check -1

rapidsai · Sep 19, 2024 · c87ad73 · c87ad73
1 parent 4032fee
commit c87ad73
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/cpp/include/kvikio/remote_handle.hpp b/cpp/include/kvikio/remote_handle.hpp
@@ -184,6 +184,10 @@ inline std::size_t get_file_size(std::string url)
 
   curl_off_t cl;
   curl.getinfo(CURLINFO_CONTENT_LENGTH_DOWNLOAD_T, &cl);
+  if (cl < 0) {
+    throw std::runtime_error("cannot get size of " + url +
+                             ", content-length not provided by the server");
+  }
   return cl;
 }