| -rw-r--r-- | lib/Makefile.am | 1 | ||||
| -rw-r--r-- | lib/util.cc | 127 |
2 files changed, 128 insertions, 0 deletions
diff --git a/lib/Makefile.am b/lib/Makefile.am index a904569..7d44856 100644 --- a/lib/Makefile.am +++ b/lib/Makefile.am @@ -1,16 +1,17 @@ lib_LTLIBRARIES = libkonforka.la INCLUDES = -I${top_srcdir}/include AM_CXXFLAGS = ${PQXX_CFLAGS} LDADD = ${PQXX_LIBS} EXTRA_DIST = pqxx_pile.cc EXTRA_SOURCES= if HAVE_PQXX EXTRA_SOURCES += pqxx_pile.cc endif libkonforka_la_SOURCES = \ exception.cc \ + util.cc \ pointer_map.cc \ ${EXTRA_SOURCES} diff --git a/lib/util.cc b/lib/util.cc new file mode 100644 index 0000000..74039c6 --- a/dev/null +++ b/lib/util.cc @@ -0,0 +1,127 @@ +#include <sys/types.h> +#include <sys/stat.h> +#include <konforka/util.h> + +namespace konforka { + + /* + * XXX: this code is borrowed from sitecing as is, although it should be optimized. + */ + + string normalize_path(const string& p,int o) { + const char *s = p.c_str(); + if( s[0]=='.' && s[1]=='/' ) + s += 2; // skip leading './' + if(o&strip_leading_slash) + for(;(*s)=='/';s++); + string rv; + string::size_type notslash = 0; + for(;*s;s++) { + if(s[0]=='/') { + if(s[1]=='/') + continue; // skip duplicate slash + if(s[1]=='.' && s[2]=='/') { + // '/./' sequence encountered + s += 2; + continue; + } + } + if( + (o&restrict_dotdot) && ( + ( rv.empty() && s[0]=='.' && s[1]=='.' && s[2]=='/' ) // '^../' + || + ( s[0]=='/' && s[1]=='.' && s[2]=='.' && (s[3]=='/' || s[3]==0) ) // '/../' or '/..$' + ) + ) + throw restricted_sequence_error(CODEPOINT,"restricted updir (..) sequence encountered"); + rv += *s; + if( (*s) !='/' ) + notslash = rv.length(); + } + if(!(o&strip_trailing_slash)) + notslash++; + if(notslash<rv.length()) + rv.erase(notslash); // XXX: does this operation have enough sense to be performed? + return rv; + } + + string dir_name(const string& p) { + string::size_type sl = p.find_last_of('/'); + if(sl==string::npos) + return ""; // no slashes -- no dir. + string::size_type nosl = p.find_last_not_of('/',sl); + if(nosl==string::npos) + return ""; // only slashes -- no dir. + return p.substr(0,nosl+1); + } + + string combine_path(const string& orig,const string& rel,int o) { + string r = normalize_path(rel,0); + if(r.empty()) { + // XXX: this behaviour is questionable. + return normalize_path( (o&origin_is_file)?dir_name(orig):orig, strip_leading_slash|restrict_dotdot|strip_trailing_slash); + } + string rv; + if(r[0]=='/') { + r.erase(0,1); + }else{ + rv = normalize_path( (o&origin_is_file)?dir_name(orig):orig, restrict_dotdot|strip_trailing_slash); + } + string::size_type lsl = rv.rfind('/'); + // iterate through slashes in relative path + for(string::size_type sl=r.find('/');sl!=string::npos;sl=r.find('/')) { + assert(sl!=0); // sure we don't start with '/' at this point + if(sl==1 && r[0]=='.') { // './' + r.erase(0,2); + }else if(sl==2 && r[0]=='.' && r[1]=='.') { // '../' + if(lsl==string::npos) { + if(rv.empty() && (o&fail_beyond_root)) + throw beyond_root_error(CODEPOINT,"went beyond root while combining path"); + rv.clear(); + }else{ + rv.erase(lsl); + lsl = rv.rfind('/'); + } + r.erase(0,3); + }else{ // 'something/' + lsl = rv.length(); + rv += '/'; + rv += r.substr(0,sl); + r.erase(0,sl+1); + } + } + if(r.empty()) + return rv+'/'; + if(r.length()==2 && r[0]=='.' && r[1]=='.') { + if(lsl==string::npos) { + if(rv.empty() && (o&fail_beyond_root)) + throw beyond_root_error(CODEPOINT,"went beyond root while combining path"); + return "/"; + }else{ + rv.erase(lsl+1); + return rv; + } + } + rv += '/'; + rv += r; + return rv; + } + + void make_path(const string& p,mode_t m) { + struct stat st; + for(string::size_type sl=0;sl!=string::npos;sl=p.find('/',sl+1)) { + if(!sl) + continue; + string pp = p.substr(0,sl); + if(stat(pp.c_str(),&st) || !S_ISDIR(st.st_mode)) { + if(mkdir(pp.c_str(),m)) + throw konforka::system_error(CODEPOINT,"failed to mkdir()"); + } + } + if(stat(p.c_str(),&st) || !S_ISDIR(st.st_mode)) { + if(mkdir(p.c_str(),m)) + throw konforka::system_error(CODEPOINT,"failed to mkdir()"); + } + } + +} |