56 static const std::string
VERSION =
"0.2.0";
58 static const std::string
gUserAgent =
"ROOT/" + std::string(
gROOT->GetVersion()) +
59 " TDavixFile/" +
VERSION +
" davix/" + Davix::version();
62 #define ENVPFX "Davix." 66 using namespace Davix;
87 if (!str)
return false;
89 if (!strcmp(str,
"n") || !strcmp(str,
"no") || !strcmp(str,
"0") || !strcmp(str,
"false"))
return true;
100 old_flag |= O_RDONLY;
103 old_flag |= (O_CREAT | O_WRONLY | O_TRUNC);
106 old_flag |= (O_RDWR);
119 davix_set_log_level(0);
122 davix_set_log_level(DAVIX_LOG_WARNING);
125 davix_set_log_level(DAVIX_LOG_VERBOSE);
128 davix_set_log_level(DAVIX_LOG_DEBUG);
131 davix_set_log_level(DAVIX_LOG_ALL);
143 char default_proxy[64];
144 const char *genvvar = 0, *genvvar1 = 0;
148 ucert = ukey = genvvar;
150 Info(
"TDavixFile_http_get_ucert",
"Found proxy in gEnv");
155 if (getenv(
"X509_USER_PROXY")) {
157 Info(
"TDavixFile_http_get_ucert",
"Found proxy in X509_USER_PROXY");
158 ucert = ukey = getenv(
"X509_USER_PROXY");
163 snprintf(default_proxy,
sizeof(default_proxy),
"/tmp/x509up_u%d",
166 if (access(default_proxy, R_OK) == 0) {
168 Info(
"TDavixFile_http_get_ucert",
"Found proxy in /tmp");
169 ucert = ukey = default_proxy;
174 genvvar =
gEnv->
GetValue(
"Davix.GSI.UserCert", (
const char *) NULL);
175 genvvar1 =
gEnv->
GetValue(
"Davix.GSI.UserKey", (
const char *) NULL);
176 if (genvvar || genvvar1) {
178 Info(
"TDavixFile_http_get_ucert",
"Found cert and key in gEnv");
186 if (getenv(
"X509_USER_CERT"))
187 ucert = getenv(
"X509_USER_CERT");
188 if (getenv(
"X509_USER_KEY"))
189 ukey = getenv(
"X509_USER_KEY");
191 if ((ucert.size() > 0) || (ukey.size() > 0)) {
193 Info(
"TDavixFile_http_get_ucert",
"Found cert and key in gEnv");
202 Davix::X509Credential *cert, Davix::DavixError **err)
206 std::string ucert, ukey;
209 if (ucert.empty() || ukey.empty()) {
210 Davix::DavixError::setupError(err,
"TDavixFile",
211 Davix::StatusCode::AuthentificationError,
212 "Could not set the user's proxy or certificate");
215 return cert->loadFromFilePEM(ukey, ucert,
"", err);
231 if (davix_context_s ==
NULL) {
233 if (davix_context_s ==
NULL) {
234 davix_context_s =
new Context();
244 DavixError *davixErr =
NULL;
245 Davix_fd *fd = davixPosix->open(davixParam, fUrl.GetUrl(), oflags, &davixErr);
252 DavixError *davixErr2 =
NULL;
254 DavFile
file(*davixContext, Davix::Uri(fUrl.GetUrl()));
255 std::vector<DavFile> replicasLocal = file.getReplicas(
NULL, &davixErr2);
256 for(
size_t i = 0; i < replicasLocal.size(); i++) {
257 replicas.push_back(replicasLocal[i].getUri().getString());
261 DavixError::clearError(&davixErr2);
263 if(replicas.empty()) {
266 Error(
"DavixOpen",
"can not open file with davix: %s (%d)",
267 davixErr->getErrMsg().c_str(), davixErr->getStatus());
269 DavixError::clearError(&davixErr);
272 davixPosix->fadvise(fd, 0, 300, Davix::AdviseRandom);
282 DavixError *davixErr =
NULL;
283 if (davixFd !=
NULL && davixPosix->close(davixFd, &davixErr)) {
284 Error(
"DavixClose",
"can not to close file with davix: %s (%d)",
285 davixErr->getErrMsg().c_str(), davixErr->getStatus());
286 DavixError::clearError(&davixErr);
294 const char *env_var =
NULL;
297 Info(
"enableGridMode",
" grid mode enabled !");
299 if( ( env_var = getenv(
"X509_CERT_DIR")) ==
NULL){
300 env_var=
"/etc/grid-security/certificates/";
302 davixParam->addCertificateAuthorityPath(env_var);
304 Info(
"enableGridMode",
"Adding CAdir %s", env_var);
317 template<
typename TRequestParams = Davix::RequestParams>
318 static auto awsRegion(TRequestParams *parameters,
const char *region)
319 -> decltype(parameters->setAwsRegion(region),
void())
321 if (
gDebug > 1)
Info(
"awsRegion",
"Setting S3 Region to '%s' - v4 signature will be used", region);
322 parameters->setAwsRegion(region);
325 template<
typename TRequestParams = Davix::RequestParams>
327 Warning(
"setAwsRegion",
"Unable to set AWS region, not supported by this version of davix");
331 template<
typename TRequestParams = Davix::RequestParams>
332 static auto awsToken(TRequestParams *parameters,
const char *token)
333 -> decltype(parameters->setAwsToken(token),
void())
335 if (
gDebug > 1)
Info(
"awsToken",
"Setting S3 STS temporary credentials");
336 parameters->setAwsToken(token);
339 template<
typename TRequestParams = Davix::RequestParams>
341 Warning(
"awsToken",
"Unable to set AWS token, not supported by this version of davix");
345 if(!region.empty()) {
352 awsToken(davixParam, token.c_str());
357 const std::string ®ion,
const std::string &token)
360 Info(
"setS3Auth",
" Aws S3 tokens configured");
362 davixParam->setAwsAuthorizationKeys(secret, access);
363 davixParam->setProtocol(RequestProtocol::AwsS3);
365 setAwsRegion(region);
373 const char *env_var =
NULL, *env_var2 =
NULL;
375 davixParam->setTransparentRedirectionSupport(
true);
381 davixParam->addCertificateAuthorityPath(env_var);
383 Info(
"parseConfig",
"Add CAdir: %s", env_var);
386 bool ca_check_local = !
isno(
gEnv->
GetValue(
"Davix.GSI.CACheck", (
const char *)
"y"));
387 davixParam->setSSLCAcheck(ca_check_local);
389 Info(
"parseConfig",
"Setting CAcheck to %s", ((ca_check_local) ? (
"true") : (
"false")));
392 if (((env_var =
gEnv->
GetValue(
"Davix.S3.SecretKey", getenv(
"S3_SECRET_KEY"))) != NULL)
393 && ((env_var2 =
gEnv->
GetValue(
"Davix.S3.AccessKey", getenv(
"S3_ACCESS_KEY"))) != NULL)) {
394 Info(
"parseConfig",
"Setting S3 SecretKey and AccessKey. Access Key : %s ", env_var2);
395 davixParam->setAwsAuthorizationKeys(env_var, env_var2);
398 if ( (env_var =
gEnv->
GetValue(
"Davix.S3.Region", getenv(
"S3_REGION"))) != NULL) {
399 setAwsRegion(env_var);
402 if( (env_var =
gEnv->
GetValue(
"Davix.S3.Token", getenv(
"S3_TOKEN"))) != NULL) {
403 setAwsToken(env_var);
407 env_var =
gEnv->
GetValue(
"Davix.GSI.GridMode", (
const char *)
"y");
417 std::stringstream ss(option);
419 std::vector<std::string> parsed_options;
421 std::string s3seckey, s3acckey, s3region, s3token;
423 while (std::getline(ss, item,
' ')) {
424 parsed_options.push_back(item);
427 for (std::vector<std::string>::iterator it = parsed_options.begin(); it < parsed_options.end(); ++it) {
434 davixParam->setSSLCAcheck(
false);
437 if (strncasecmp(it->c_str(),
s3_seckey_opt, strlen(s3_seckey_opt)) == 0) {
438 s3seckey = std::string(it->c_str() + strlen(s3_seckey_opt));
441 if (strncasecmp(it->c_str(),
s3_acckey_opt, strlen(s3_acckey_opt)) == 0) {
442 s3acckey = std::string(it->c_str() + strlen(s3_acckey_opt));
445 if (strncasecmp(it->c_str(),
s3_region_opt, strlen(s3_region_opt)) == 0) {
446 s3region = std::string(it->c_str() + strlen(s3_region_opt));
449 if (strncasecmp(it->c_str(),
s3_token_opt, strlen(s3_token_opt)) == 0) {
450 s3token = std::string(it->c_str() + strlen(s3_token_opt));
456 if (s3seckey.size() > 0) {
457 setS3Auth(s3seckey, s3acckey, s3region, s3token);
468 davixPosix =
new DavPosix(davixContext);
469 davixParam =
new RequestParams();
471 davixParam->setMetalinkMode(Davix::MetalinkMode::Disable);
481 DavixError *davixErr =
NULL;
483 if (davixPosix->stat(davixParam, url, st, &davixErr) < 0) {
485 Error(
"DavixStat",
"can not stat the file with davix: %s (%d)",
486 davixErr->getErrMsg().c_str(), davixErr->getStatus());
487 DavixError::clearError(&davixErr);
534 if(!replicas.empty()) {
535 std::stringstream ss;
536 for(
size_t i = 0; i < replicas.size(); i++) {
538 if(i != replicas.size()-1) ss <<
"|";
561 Error(
"Seek",
"seeking from end in archive is not (yet) supported");
567 Info(
"Seek",
" move cursor to %lld" 586 Info(
"ReadBuffer",
"%lld bytes of data read sequentially" 587 " (%d requested)", ret, len);
605 Info(
"ReadBuffer",
"%lld bytes of data read from offset" 606 " %lld (%d requested)", ret, pos, len);
618 d_ptr->
davixPosix->fadvise(fd, static_cast<dav_off_t>(offs), static_cast<dav_size_t>(len), Davix::AdviseRandom);
621 Info(
"ReadBufferAsync",
"%d bytes of data prefected from offset" 622 " %lld ", len, offs);
639 Info(
"ReadBuffers",
"%lld bytes of data read from a list of %d buffers",
658 Info(
"WriteBuffer",
"%lld bytes of data write" 659 " %d requested", ret, len);
682 std::vector<void *>::iterator
f = std::find(dirdVec.begin(), dirdVec.end(), fd);
683 return (f != dirdVec.end());
691 dirdVec.push_back(fd);
699 std::vector<void *>::iterator
f = std::find(dirdVec.begin(), dirdVec.end(), fd);
700 if (f != dirdVec.end())
712 Info(
"GetSize",
"file size requested: %lld", (
Long64_t)st.st_size);
751 DavixError *davixErr =
NULL;
756 Error(
"DavixReadBuffer",
"can not read data with davix: %s (%d)",
757 davixErr->getErrMsg().c_str(), davixErr->getStatus());
758 DavixError::clearError(&davixErr);
771 DavixError *davixErr =
NULL;
776 Error(
"DavixWriteBuffer",
"can not write data with davix: %s (%d)",
777 davixErr->getErrMsg().c_str(), davixErr->getStatus());
778 DavixError::clearError(&davixErr);
791 DavixError *davixErr =
NULL;
796 Error(
"DavixPReadBuffer",
"can not read data with davix: %s (%d)",
797 davixErr->getErrMsg().c_str(), davixErr->getStatus());
798 DavixError::clearError(&davixErr);
811 DavixError *davixErr =
NULL;
813 DavIOVecInput in[nbuf];
814 DavIOVecOuput out[nbuf];
817 for (
Int_t i = 0; i < nbuf; ++i) {
818 in[i].diov_buffer = &buf[lastPos];
819 in[i].diov_offset = pos[i];
820 in[i].diov_size = len[i];
826 Error(
"DavixReadBuffers",
"can not read data with davix: %s (%d)",
827 davixErr->getErrMsg().c_str(), davixErr->getStatus());
828 DavixError::clearError(&davixErr);
static void ConfigureDavixLogLevel()
double read(const std::string &file_name)
reading
virtual Long64_t GetSize() const
Returns the current file size.
virtual void Info(const char *method, const char *msgfmt,...) const
Issue info message.
static const std::string gUserAgent
const char * s3_seckey_opt
Long64_t fBytesWrite
Number of bytes written to this file.
const char * open_mode_read
const char * open_mode_update
void Init(Bool_t init)
Initialize a TFile object.
static void TDavixFile_http_get_ucert(std::string &ucert, std::string &ukey)
void setS3Auth(const std::string &secret, const std::string &access, const std::string ®ion, const std::string &token)
static auto awsToken(TRequestParams *parameters, const char *token) -> decltype(parameters->setAwsToken(token), void())
Davix::RequestParams * davixParam
Small helper to keep current directory context.
static void SetFileReadCalls(Int_t readcalls=0)
virtual Bool_t ReadBuffer(char *buf, Int_t len)
Read specified byte range from remote file via HTTP.
TDavixFileInternal * d_ptr
Long64_t DavixReadBuffer(Davix_fd *fd, char *buf, Int_t len)
Int_t fReadCalls
Number of read calls ( not counting the cache calls )
virtual Bool_t WriteBuffer(const char *buffer, Int_t bufferLength)
Write a buffer to the file.
Davix_fd * getDavixFileInstance()
void setAwsRegion(const std::string ®ion)
const char * GetUrl(Bool_t withDeflt=kFALSE) const
Return full URL.
const char * open_mode_create
Long64_t DavixReadBuffers(Davix_fd *fd, char *buf, Long64_t *pos, Int_t *len, Int_t nbuf)
static void SetFileBytesWritten(Long64_t bytes=0)
Davix::DavPosix * davixPosix
virtual void Seek(Long64_t offset, ERelativeTo pos=kBeg)
Set position from where to start reading.
static Long64_t GetFileBytesRead()
Static function returning the total number of bytes read from all files.
static Long64_t GetFileBytesWritten()
Static function returning the total number of bytes written to all files.
void Info(const char *location, const char *msgfmt,...)
void Error(const char *location, const char *msgfmt,...)
Int_t DavixStat(const char *url, struct stat *st)
static Context * davix_context_s
Long64_t DavixWriteBuffer(Davix_fd *fd, const char *buf, Int_t len)
Long64_t fEND
Last used byte in file.
virtual Bool_t ReadBufferAsync(Long64_t offs, Int_t len)
const char * s3_token_opt
virtual TString GetNewUrl()
static Davix::Context * getDavixInstance()
std::vector< std::string > getReplicas()
virtual Int_t GetValue(const char *name, Int_t dflt)
Returns the integer value for a resource.
void removeDird(void *fd)
static auto awsRegion(TRequestParams *parameters, const char *region) -> decltype(parameters->setAwsRegion(region), void())
virtual void Error(const char *method, const char *msgfmt,...) const
Issue error message.
static Int_t GetFileReadCalls()
Static function returning the total number of read calls from all files.
void eventStop(Double_t t, Long64_t len, bool read=true)
set TFile state info
void Warning(const char *location, const char *msgfmt,...)
bool isno(const char *str)
virtual void Init(Bool_t create)
Initialize a TFile object.
virtual Bool_t ReadBuffers(char *buf, Long64_t *pos, Int_t *len, Int_t nbuf)
Read the nbuf blocks described in arrays pos and len.
const char * grid_mode_opt
static const std::string VERSION
The TTimeStamp encapsulates seconds and ns since EPOCH.
const char * s3_acckey_opt
int configure_open_flag(const std::string &str, int old_flag)
const char * s3_region_opt
void enableGridMode()
Enable the grid mode The grid Mode configure automatically all grid-CA path, VOMS authentication and ...
typedef void((*Func_t)())
void setAwsToken(const std::string &token)
const char * ca_check_opt
static int TDavixFile_http_authn_cert_X509(void *userdata, const Davix::SessionInfo &info, Davix::X509Credential *cert, Davix::DavixError **err)
Long64_t fArchiveOffset
!Offset at which file starts in archive
TDavixFile(const char *url, Option_t *option="", const char *ftitle="", Int_t compress=1)
Open function for TDavixFile.
Long64_t fOffset
!Seek offset cache
static void SetFileBytesRead(Long64_t bytes=0)
Long64_t DavixPReadBuffer(Davix_fd *fd, char *buf, Long64_t pos, Int_t len)
void setCACheck(Bool_t check)
Enable or disable certificate authority check.
void parseParams(Option_t *option)
intput params
Long64_t fBytesRead
Number of bytes read from this file.
const char * open_mode_new