1 files changed, 367 insertions, 0 deletions

diff --git a/src/sh/file.cc b/src/sh/file.cc
new file mode 100644
index 0000000..3d885c5
--- /dev/null
+++ b/src/sh/file.cc
@@ -0,0 +1,367 @@
+/*************************************************************************
+ *
+ * HTCd - Copyright (C) 1998-2006 Henrik Rydberg
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program 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 General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+ */
+
+#include <sh/htc.h>
+#include <hdb/buffer.h>
+#include <mt/dates.h>
+#include <dirent.h>
+#include <sys/stat.h>
+
+const mstring sdot=".",sdots="..";
+
+///////////////////////////////////////////////////////////////////
+
+inline void stat(struct stat& fs,Env& env,const sref& path) throw(merror_t) {
+  if(::stat(DocPath(env,path).c_str(),&fs)<0)
+    THROW("htc: could not stat "<<path<<" - not found");
+}
+
+///////////////////////////////////////////////////////////////////
+
+TOK_IMPL(remfile) {
+  mstring path=DocPath(env,env[0]);
+  if(remove(path.c_str())<0) THROW("htc: Could not remove "<<path);
+}
+
+TOK_IMPL(renfile) {
+  mstring oldpath=DocPath(env,env[0]),newpath=DocPath(env,env[1]);
+  if(rename(oldpath.c_str(),newpath.c_str())<0)
+    THROW("htc: Could not rename "<<oldpath<<" to "<<newpath);
+}
+
+TOK_IMPL(filesafe) {
+  static const charset fsafe=cset_alphareal|'_';
+  mstring file=env[0];
+  int pos=file.rfind('/'); if(pos>=0) file=file.past(pos,1);
+  pos=file.rfind('\\'); if(pos>=0) file=file.past(pos,1);
+  while((pos=file.find(sdots))>=0) file.erase(pos,2);
+  mstring acc;
+  for(int i=0;i<file.size();i++) if(fsafe.test(file[i])) acc.append(1,file[i]);
+  out<<acc;
+}
+
+///////////////////////////////////////////////////////////////////
+
+TOK_IMPL(seturi) {
+  Req(env)->uri=env[0];
+}
+
+TOK_IMPL(uri) {
+  out<<Req(env)->uri;
+}
+
+TOK_IMPL(pwd) {
+  out<<Req(env)->uri.left_last('/');
+}
+
+TOK_IMPL(npwd) {
+  mstring s=Req(env)->uri.left_last('/');
+  for(int i=0;i<MAXLANG;i++) {
+    if(s.past_first('/').left_first('/')==LANG[i]) {
+      s=s.past_first('/').right_first('/');
+      break;
+    }
+  }
+  out<<s;
+}
+
+TOK_IMPL(home) {
+  if(Req(env)->htc()->docroot.nempty())
+    THROW("htc: home only works in absolute file mode");
+  out<<htchome();
+}
+
+///////////////////////////////////////////////////////////////////
+
+TOK_IMPL(load) {
+  mstring name=env.parg(-1),uri=env.URI(env[0]),path=DocPath(env,uri);
+  if(name.nempty()) LoadBuffer(Reference(*env.toks,name,uri,1)->data,path);
+  else { mstring buffer; LoadBuffer(buffer,path); out<<buffer; }
+}
+
+TOK_IMPL(open) {
+  mstring name=env[0],uri=env.URI(env[1]),path=DocPath(env,uri);
+  dbref_t* ref=DBRef(*env.toks,env[0],ms_empty,tok_t::PRIVATE,1);
+  ref->db.Open(path,atoi(env.parg(-1)));
+}
+
+TOK_IMPL(save) {
+  mstring path=DocPath(env,env[0]),var=env[1];
+  int prot=atoi(env.parg(2)); if(!prot) prot=0600;
+  tok_t* p=env.toks->get(var);
+  if(!p) THROW("htc: "<<var<<": no such variable");
+  if(p->is_reference()) SaveBuffer(((reference_t*)p)->data,path,prot);
+  else if(p->is_dbref()) ((dbref_t*)p)->db.Saveas(path,prot);
+  else SaveBuffer(p->body,path,prot);
+}
+
+TOK_IMPL(dbmod) {
+  dbref_t* p=(dbref_t*)env.toks->get(env[0]);
+  if(!p&&!p->is_dbref()) THROW("htc: "<<env[0]<<": no such database");
+  out<<p->db.Lastmod();
+}
+
+///////////////////////////////////////////////////////////////////
+
+static void Run(mstream& out,Env& env,const mstring& path) throw(merror_t) {
+  mstring buffer;
+  LoadBuffer(buffer,path);
+  env.parse(out,buffer);
+}
+
+TOK_IMPL(include) {
+  mstring cwd=env.URI(env[0]),path=DocPath(env,cwd);
+  if(!env.glob_find(cwd)) {
+    env.toks->add(new tok_t(cwd,ms_empty,tok_t::CONSTANT|tok_t::PRIVATE),0,1);
+    mstream m;
+    Env e(env,env.mod,cwd);
+    Run(m,e,path);
+  }
+}
+
+TOK_IMPL(parse) {
+  mstring cwd=env.URI(env[0]),path=DocPath(env,cwd);
+  Env e(env,env.mod,cwd);
+  Run(out,e,path);
+}
+
+TOK_IMPL(parse_private) {
+  mstring cwd=env.URI(env[0]),path=DocPath(env,cwd);
+  Env e(env,XPRIVATE,cwd);
+  Run(out,e,path);
+}
+
+TOK_IMPL(parse_public) {
+  mstring cwd=env.URI(env[0]),path=DocPath(env,cwd);
+  Env e(env,XPUBLIC,cwd);
+  Run(out,e,path);
+}
+
+///////////////////////////////////////////////////////////////////
+
+TOK_IMPL(dirent) {
+  mstring uri=env.URI(env[0]);
+  sref uribase=uri.leftand_last('/'),ext=uri.adv(uribase);  
+  mstring dirpath=DocPath(env,uribase);
+  DIR* dir=opendir(dirpath.c_str());
+  if(dir) {
+    dirent* p;
+    while((p=readdir(dir))!=0) {
+      sref name(p->d_name);
+      if(name.empty()||name==sdot||name==sdots) continue;
+      if(ext.empty()||name.right(-ext.size()-1)==ext) out<<uribase<<name<<"\n";
+    }
+    closedir(dir);
+  }
+}
+
+TOK_IMPL(dirmod) {
+  struct stat fs;
+  if(::stat(DocPath(env,env[0]).c_str(),&fs)>=0) out<<fs.st_mode;
+  else out<<(int)0;
+}
+
+TOK_IMPL(dirmtime) { struct stat fs; stat(fs,env,env[0]); out<<fs.st_mtime; }
+TOK_IMPL(diruid) { struct stat fs; stat(fs,env,env[0]); out<<fs.st_uid; }
+TOK_IMPL(dirgid) { struct stat fs; stat(fs,env,env[0]); out<<fs.st_gid; }
+TOK_IMPL(filesize) { struct stat fs; stat(fs,env,env[0]); out<<fs.st_size; }
+
+TOK_IMPL(setperm) {
+  mstring path=DocPath(env,env[0]);
+  if(chmod(path.c_str(),atoi(env[1]))<0)
+    THROW("htc: could not set file permission for "<<path);
+}
+
+///////////////////////////////////////////////////////////////////
+
+void envAddFile(tokmap& T)
+{
+  TOK_ADD(remfile,"x",tok_t::SYSONLY,
+          "%Remove file\n"
+          "%Syntax: \\remfile{uri}\n"
+          );
+  TOK_ADD(renfile,"xx",tok_t::SYSONLY,
+          "%Rename file\n"
+          "%Syntax: \\renfile{olduri}{newuri}\n"
+          );
+  TOK_ADD(filesafe,"x",tok_t::PUBLIC,
+          "%Transform uri to something safe to save\n"
+          "%Syntax: \\filesafe{uri}\n"
+          "%\n"
+          "%Basically removes eventual path information and\n"
+          "%odd characters.\n"
+          );
+
+  ////////////////////////////////////////////////////////
+
+  TOK_ADD(seturi,"x",tok_t::SYSONLY,
+          "%Set current uri of current request\n"
+          "%Syntax: \\seturi\n"
+          "%\n"
+          "%Always points to a (virtual) uri.\n"
+          );
+  TOK_ADD(uri,"",tok_t::PUBLIC,
+          "%uri of current request\n"
+          "%Syntax: \\uri\n"
+          "%\n"
+          "%Always points to a (virtual) uri.\n"
+          );
+  TOK_ADD(pwd,"",tok_t::PUBLIC,
+          "%uri/directory of current request\n"
+          "%Syntax: \\pwd\n"
+          "%\n"
+          "%Always points to a (virtual) directory, and never contain any\n"
+          "%trailing slashes.\n"
+          );
+  TOK_ADD(npwd,"",tok_t::PUBLIC,
+          "%uri/directory of current request, without language reference\n"
+          "%Syntax: \\npwd\n"
+          "%\n"
+          "%Always points to a (virtual) directory, and never contain any\n"
+          "%trailing slashes.\n"
+          );
+  TOK_ADD(home,"",tok_t::PRIVATE,
+          "%Home directory\n"
+          "%Syntax: \\home\n"
+          "%\n"
+          "%Only works in absolute path mode, i.e. when no docroot is set.\n"
+          );
+  
+  ////////////////////////////////////////////////////////
+  
+  TOK_ADD(load,"?x",tok_t::PRIVATE,
+          "%Load file directly to output or to variable\n"
+          "%Syntax: \\load<varname>{uri}\n"
+          );
+  TOK_ADD(open,"xx?",tok_t::SYSONLY,
+          "%Open database to variable\n"
+          "%Syntax: \\load{varname}{uri}<update>\n"
+          "%\n"
+          "%If update is true, the database is opened for reading\n"
+          "%and writing. NOTE that all updates are syncronous and\n"
+          "%instantly - saves or locks are not needed.\n"
+          );
+  TOK_ADD(save,"xx?",tok_t::SYSONLY,
+          "%Save variable to file\n"
+          "%Syntax: \\save{uri}{varname}<prot>\n"
+          "%\n"
+          "%If the variable points to a reference object, the\n"
+          "%reference is saved. If it points to a database,\n"
+          "%the database is saved. For globals and functions,\n"
+          "%the body is saved.\n"
+          "%\n"
+          "%The optional prot argument is a (base 10) integer\n"
+          "%defining the creation mode mask (unix).\n"
+          );
+  TOK_ADD(dbmod,"x",tok_t::PUBLIC,
+          "%Get modification time of open database\n"
+          "%Syntax: \\dbmod{var}\n"
+          );
+
+  ////////////////////////////////////////////////////////
+
+  TOK_ADD(include,"x",tok_t::PRIVATE,
+          "%Include file in current running mode\n"
+          "%Syntax: \\include{uri}\n"
+          "%\n"
+          "%The file is loaded and evaluated, but produces no output\n"
+          "%Any uri can only be included once, so it is safe to use\n"
+          "%include as a precondition.\n"
+          "%\n"
+          "%NOTE that files parsed this way must be known to\n"
+          "%the super-user in order not to create a security hole -\n"
+          "%especially since this command is open in private mode also.\n"
+          );
+  
+  TOK_ADD(parse,"x",tok_t::SYSONLY,
+          "%Parse file in current running mode\n"
+          "%Syntax: \\parse{uri}\n"
+          "%\n"
+          "%The file is loaded and evaluated to output.\n"
+          "%\n"
+          "%NOTE that files parsed this way must be known to\n"
+          "%the super-user in order not to create a security hole.\n"
+          );
+  
+  TOK_ADD(parse_private,"x",tok_t::PRIVATE,
+          "%Parse file in private mode\n"
+          "%Syntax: \\parse.private{uri}\n"
+          "%\n"
+          "%The file is loaded and evaluated to output.\n"
+          "%\n"
+          "%System-mode commands are silently ignored.\n"
+          );
+
+  TOK_ADD(parse_public,"x",tok_t::PRIVATE,
+          "%Parse file in public mode\n"
+          "%Syntax: \\parse.public{uri}\n"
+          "%\n"
+          "%The file is loaded and evaluated to output.\n"
+          "%\n"
+          "%System/Private mode commands are silently ignored.\n"
+          );
+
+  ////////////////////////////////////////////////////////
+
+  TOK_ADD(dirent,"x",tok_t::SYSONLY,
+          "%Collect directory entries\n"
+          "%Syntax: \\dirent{uri/match}\n"
+          "%\n"
+          "%An argument of anydir/ collects all files in anydir.\n"
+          "%An argument of anydir/.ext collects all files ending with .ext\n"
+          "%\n"
+          "%Note that an argument of anydir collects files in the parent\n"
+          "%directory of anydir which end whith anydir, which probably is\n"
+          "%not what you want. Trailing slashes are important.\n"
+          );
+  TOK_ADD(dirmod,"x",tok_t::SYSONLY,
+          "%Return file mode of uri\n"
+          "%Syntax: \\dirmod{uri}\n"
+          "%\n"
+          "%dirmod uses the unix stat function.\n"
+          "%On failure to read, returns zero.\n"
+          );
+  TOK_ADD(dirmtime,"x",tok_t::SYSONLY,
+          "%Return modification time of uri\n"
+          "%Syntax: \\dirmtime{uri}\n"
+          );
+  TOK_ADD(diruid,"x",tok_t::SYSONLY,
+          "%Return UNIX uid of uri\n"
+          "%Syntax: \\diruid{uri}\n"
+          "%\n"
+          "%Note that the UNIX uid is different from the HTC one.\n"
+          );
+  TOK_ADD(dirgid,"x",tok_t::SYSONLY,
+          "%Return UNIX gid of uri\n"
+          "%Syntax: \\dirgid{uri}\n"
+          "%\n"
+          "%Note that the UNIX gid is different from the HTC one.\n"
+          );
+  TOK_ADD(filesize,"x",tok_t::SYSONLY,
+          "%Return filesize of uri\n"
+          "%Syntax: \\filesize{uri}\n"
+          );
+  TOK_ADD(setperm,"xx",tok_t::SYSONLY,
+          "%Set file permissions of uri\n"
+          "%Syntax: \\setperm{uri}{mode}\n"
+          "%\n"
+          "%Mode is an integer bitmask compliant with fcntl.\n"
+          );
+  
+}