void fetchAllBuckets(RTS3 hS3)
{
/* Fetch all available buckets */
RTTestIPrintf(RTTESTLVL_ALWAYS, " List all buckets...\n");
char pszTitle[] = "RTS3GetBuckets";
RTS3SetProgressCallback(hS3, progress, pszTitle);
PCRTS3BUCKETENTRY pBuckets = NULL;
int rc = RTS3GetBuckets(hS3, &pBuckets);
if (RT_SUCCESS(rc))
{
if (pBuckets)
{
PCRTS3BUCKETENTRY pTmpBuckets = pBuckets;
while (pBuckets)
{
RTTestIPrintf(RTTESTLVL_ALWAYS, " > %s, %s\n", pBuckets->pszName, pBuckets->pszCreationDate);
pBuckets = pBuckets->pNext;
}
RTS3BucketsDestroy(pTmpBuckets);
}
else
RTTestIPrintf(RTTESTLVL_ALWAYS, " > empty\n");
}
else
RTTestIFailed("RTS3GetBuckets -> %Rrc", rc);
}
void fetchAllKeys(RTS3 hS3, const char *pszBucketName)
{
/* Fetch all available keys of a specific bucket */
RTTestIPrintf(RTTESTLVL_ALWAYS, " List all keys of bucket '%s'...\n", pszBucketName);
PCRTS3KEYENTRY pKeys = NULL;
char pszTitle[] = "RTS3GetBucketKeys";
RTS3SetProgressCallback(hS3, progress, pszTitle);
int rc = RTS3GetBucketKeys(hS3, pszBucketName, &pKeys);
if (RT_SUCCESS(rc))
{
if (pKeys)
{
PCRTS3KEYENTRY pTmpKeys = pKeys;
while (pKeys)
{
RTTestIPrintf(RTTESTLVL_ALWAYS, " > %s, %s, %lu\n", pKeys->pszName, pKeys->pszLastModified, pKeys->cbFile);
pKeys = pKeys->pNext;
}
RTS3KeysDestroy(pTmpKeys);
}
else
RTTestIPrintf(RTTESTLVL_ALWAYS, " > empty\n");
}
else
RTTestIFailed("RTS3GetBucketKeys -> %Rrc", rc);
}
void createBucket(RTS3 hS3, const char *pszBucketName)
{
/* Create the bucket */
RTTestIPrintf(RTTESTLVL_ALWAYS, " Create bucket '%s'...\n", pszBucketName);
char pszTitle[] = "RTS3CreateBucket";
RTS3SetProgressCallback(hS3, progress, pszTitle);
int rc = RTS3CreateBucket(hS3, pszBucketName);
if (RT_FAILURE(rc))
RTTestIFailed("RTS3CreateBucket -> %Rrc", rc);
}
void putKey(RTS3 hS3, const char *pszBucketName, const char *pszKeyName, const char *pszFilename)
{
/* Fetch the content of a key */
RTTestIPrintf(RTTESTLVL_ALWAYS, " Put '%s' into key '%s' in bucket '%s' ...\n", pszFilename, pszKeyName, pszBucketName);
char pszTitle[] = "RTS3PutKey";
RTS3SetProgressCallback(hS3, progress, pszTitle);
int rc = RTS3PutKey(hS3, pszBucketName, pszKeyName, pszFilename);
if (RT_FAILURE(rc))
RTTestIFailed("RTS3PutKey -> %Rrc", rc);
}
void deleteKey(RTS3 hS3, const char *pszBucketName, const char *pszKeyName)
{
/* Delete the key */
RTTestIPrintf(RTTESTLVL_ALWAYS, " Delete key '%s' in bucket '%s'...\n", pszKeyName, pszBucketName);
char pszTitle[] = "RTS3DeleteKey";
RTS3SetProgressCallback(hS3, progress, pszTitle);
int rc = RTS3DeleteKey(hS3, pszBucketName, pszKeyName);
if (RT_FAILURE(rc))
RTTestIFailed("RTS3DeleteKey -> %Rrc", rc);
}
HRESULT VFSExplorer::i_deleteS3(TaskVFSExplorer *aTask)
{
LogFlowFuncEnter();
AutoCaller autoCaller(this);
if (FAILED(autoCaller.rc())) return autoCaller.rc();
AutoWriteLock appLock(this COMMA_LOCKVAL_SRC_POS);
HRESULT rc = S_OK;
RTS3 hS3 = NULL;
float fPercentStep = 100.0f / aTask->filenames.size();
try
{
int vrc = RTS3Create(&hS3, m->strUsername.c_str(), m->strPassword.c_str(),
m->strHostname.c_str(), "virtualbox-agent/" VBOX_VERSION_STRING);
if (RT_FAILURE(vrc))
throw setError(E_FAIL, tr ("Can't open S3 storage service (%Rrc)"), vrc);
RTS3SetProgressCallback(hS3, VFSExplorer::TaskVFSExplorer::uploadProgress, &aTask);
std::list<Utf8Str>::const_iterator it;
size_t i = 0;
for (it = aTask->filenames.begin();
it != aTask->filenames.end();
++it, ++i)
{
vrc = RTS3DeleteKey(hS3, m->strBucket.c_str(), (*it).c_str());
if (RT_FAILURE(vrc))
throw setError(VBOX_E_FILE_ERROR, tr ("Can't delete file '%s' (%Rrc)"), (*it).c_str(), vrc);
if (aTask->progress)
aTask->progress->SetCurrentOperationProgress((ULONG)(fPercentStep * i));
}
}
catch(HRESULT aRC)
{
rc = aRC;
}
aTask->rc = rc;
if (hS3 != NULL)
RTS3Destroy(hS3);
if (!aTask->progress.isNull())
aTask->progress->i_notifyComplete(rc);
LogFlowFunc(("rc=%Rhrc\n", rc));
LogFlowFuncLeave();
return VINF_SUCCESS;
}
HRESULT VFSExplorer::i_updateS3(TaskVFSExplorer *aTask)
{
LogFlowFuncEnter();
AutoCaller autoCaller(this);
if (FAILED(autoCaller.rc())) return autoCaller.rc();
AutoWriteLock appLock(this COMMA_LOCKVAL_SRC_POS);
HRESULT rc = S_OK;
RTS3 hS3 = NULL;
std::list<VFSExplorer::Data::DirEntry> fileList;
try
{
int vrc = RTS3Create(&hS3, m->strUsername.c_str(), m->strPassword.c_str(),
m->strHostname.c_str(), "virtualbox-agent/" VBOX_VERSION_STRING);
if (RT_FAILURE(vrc))
throw setError(E_FAIL, tr ("Can't open S3 storage service (%Rrc)"), vrc);
RTS3SetProgressCallback(hS3, VFSExplorer::TaskVFSExplorer::uploadProgress, &aTask);
/* Do we need the list of buckets or keys? */
if (m->strBucket.isEmpty())
{
PCRTS3BUCKETENTRY pBuckets = NULL;
vrc = RTS3GetBuckets(hS3, &pBuckets);
if (RT_FAILURE(vrc))
throw setError(E_FAIL, tr ("Can't get buckets (%Rrc)"), vrc);
PCRTS3BUCKETENTRY pTmpBuckets = pBuckets;
while (pBuckets)
{
/* Set always read/write permissions of the current logged in user. */
fileList.push_back(VFSExplorer::Data::DirEntry(pBuckets->pszName, VFSFileType_Directory,
0, RTFS_UNIX_IRUSR | RTFS_UNIX_IWUSR));
pBuckets = pBuckets->pNext;
}
RTS3BucketsDestroy(pTmpBuckets);
}
else
{
PCRTS3KEYENTRY pKeys = NULL;
vrc = RTS3GetBucketKeys(hS3, m->strBucket.c_str(), &pKeys);
if (RT_FAILURE(vrc))
throw setError(E_FAIL, tr ("Can't get keys for bucket (%Rrc)"), vrc);
PCRTS3KEYENTRY pTmpKeys = pKeys;
while (pKeys)
{
Utf8Str name(pKeys->pszName);
/* Set always read/write permissions of the current logged in user. */
fileList.push_back(VFSExplorer::Data::DirEntry(pKeys->pszName, VFSFileType_File, pKeys->cbFile,
RTFS_UNIX_IRUSR | RTFS_UNIX_IWUSR));
pKeys = pKeys->pNext;
}
RTS3KeysDestroy(pTmpKeys);
}
}
catch(HRESULT aRC)
{
rc = aRC;
}
if (hS3 != NULL)
RTS3Destroy(hS3);
/* Assign the result on success (this clears the old list) */
if (rc == S_OK)
m->entryList.assign(fileList.begin(), fileList.end());
aTask->rc = rc;
if (!aTask->progress.isNull())
aTask->progress->i_notifyComplete(rc);
LogFlowFunc(("rc=%Rhrc\n", rc));
LogFlowFuncLeave();
return VINF_SUCCESS;
}
