Git-Mirrors/RetroShare
1
0
Fork 0
mirror of https://github.com/RetroShare/RetroShare.git synced 2024-10-01 02:35:48 -04:00
Code Issues Actions Packages Projects Releases Wiki Activity

forgot to commit this

Browse Source 
git-svn-id: http://svn.code.sf.net/p/retroshare/code/trunk@2969 b45a01b8-16f6-495d-af2f-9b41ad6348cc
This commit is contained in:
csoler 2010-05-21 22:05:50 +00:00
parent 0634e461fb
commit 87d96059cc
2 changed files with 180 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

123
libretroshare/src/util/rsdiscspace.cc Normal file
View File

@ -0,0 +1,123 @@
/*
* libretroshare/src/util: rsdiscspace.cc
*
* Universal Networking Header for RetroShare.
*
* Copyright 2010-2010 by Cyril Soler.
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
* License Version 2 as published by the Free Software Foundation.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Library General Public License for more details.
*
* You should have received a copy of the GNU Library General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
* USA.
*
* Please report all bugs and problems to "csoler@users.sourceforge.net".
*
*/
#include <iostream>
#include <rsiface/rsfiles.h>
#include <rsiface/rsiface.h>
#include <rsiface/rsinit.h>
#include "rsdiscspace.h"
#include <util/rsthreads.h>
#include <sys/statvfs.h>
#define DELAY_BETWEEN_CHECKS 2
/*
* #define DEBUG_RSDISCSPACE
*/
time_t RsDiscSpace::_last_check[3] = { 0,0,0 } ;
uint32_t RsDiscSpace::_size_limit_mb = 100 ;
uint32_t RsDiscSpace::_current_size[3] = { 10000,10000,10000 } ;
bool RsDiscSpace::_last_res[3] = { true,true,true };
RsMutex RsDiscSpace::_mtx ;
bool RsDiscSpace::checkForDiscSpace(RsDiscSpace::DiscLocation loc)
{
RsStackMutex m(_mtx) ; // Locked
time_t now = time(NULL) ;
if(_last_check[loc]+DELAY_BETWEEN_CHECKS < now)
{
struct statvfs v ;
int res = 1;
#ifdef DEBUG_RSDISCSPACE
std::cerr << "Size determination:" << std::endl ;
#endif
switch(loc)
{
case RS_DOWNLOAD_DIRECTORY: res = statvfs(rsFiles->getDownloadDirectory().c_str(),&v) ;
#ifdef DEBUG_RSDISCSPACE
std::cerr << " path = " << rsFiles->getDownloadDirectory() << std::endl ;
#endif
break ;
case RS_PARTIALS_DIRECTORY: res = statvfs(rsFiles->getPartialsDirectory().c_str(),&v) ;
#ifdef DEBUG_RSDISCSPACE
std::cerr << " path = " << rsFiles->getPartialsDirectory() << std::endl ;
#endif
break ;
case RS_CONFIG_DIRECTORY: res = statvfs(RsInit::RsConfigDirectory().c_str(),&v) ;
#ifdef DEBUG_RSDISCSPACE
std::cerr << " path = " << RsInit::RsConfigDirectory() << std::endl ;
#endif
break ;
}
if(res)
{
std::cerr << "Determination of free disc space failed ! Be careful !" << std::endl ;
return true ;
}
_last_check[loc] = now ;
// Now compute the size in megabytes
//
_current_size[loc] = v.f_bavail * v.f_bsize / (1024*1024) ; // on purpose integer division
bool new_result = (_current_size[loc] > _size_limit_mb) ;
#ifdef DEBUG_RSDISCSPACE
std::cerr << " f_bavail = " << v.f_bavail << std::endl ;
std::cerr << " f_bsize = " << v.f_bsize << std::endl ;
std::cerr << " free MBs = " << _current_size[loc] << std::endl ;
#endif
}
bool res = _current_size[loc] > _size_limit_mb ;
if(_last_res[loc] && !res)
rsicontrol->getNotify().notifyDiskFull(loc,_size_limit_mb) ;
_last_res[loc] = res ;
return res ;
}
void RsDiscSpace::setFreeSpaceLimit(uint32_t size_in_mb)
{
RsStackMutex m(_mtx) ; // Locked
_size_limit_mb = size_in_mb ;
}
uint32_t RsDiscSpace::freeSpaceLimit()
{
RsStackMutex m(_mtx) ; // Locked
return _size_limit_mb ;
}

57
libretroshare/src/util/rsdiscspace.h Normal file
View File

@ -0,0 +1,57 @@
/*
* libretroshare/src/util: rsdiscspace.h
*
* Universal Networking Header for RetroShare.
*
* Copyright 2010-2010 by Cyril Soler.
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
* License Version 2 as published by the Free Software Foundation.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Library General Public License for more details.
*
* You should have received a copy of the GNU Library General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
* USA.
*
* Please report all bugs and problems to "csoler@users.sourceforge.net".
*
*/
#pragma once
#include <stdlib.h>
#include <stdint.h>
#include <util/rsthreads.h>
class RsDiscSpace
{
public:
typedef uint32_t DiscLocation ;
// Returns false is the disc space is lower than the given size limit, true otherwise.
// When the size limit is reached, this class calls notify to warn the user (possibly
// with a popup window).
//
static bool checkForDiscSpace(DiscLocation loc) ;
// Allow the user to specify his own size limit. Should not be too low, especially not 0 MB ;-)
// 10MB to 100MB are sensible values.
//
static void setFreeSpaceLimit(uint32_t mega_bytes) ;
static uint32_t freeSpaceLimit() ;
private:
static RsMutex _mtx ;
static time_t _last_check[3] ;
static uint32_t _size_limit_mb ;
static uint32_t _current_size[3] ;
static bool _last_res[3] ;
};