File:  [LON-CAPA] / loncom / LONCAPA.pm
Revision 1.13: download - view: text, annotated - select for diffs
Fri Aug 11 20:07:18 2006 UTC (17 years, 8 months ago) by albertel
Branches: MAIN
CVS tags: version_2_2_X, version_2_2_2, version_2_2_1, version_2_2_0, version_2_1_99_3, auto_convert_32_to_64, HEAD
- need to cleanup the sym if result is undefined

    1: # The LearningOnline Network
    2: # Base routines
    3: #
    4: # $Id: LONCAPA.pm,v 1.13 2006/08/11 20:07:18 albertel Exp $
    5: #
    6: # Copyright Michigan State University Board of Trustees
    7: #
    8: # This file is part of the LearningOnline Network with CAPA (LON-CAPA).
    9: #
   10: # LON-CAPA is free software; you can redistribute it and/or modify
   11: # it under the terms of the GNU General Public License as published by
   12: # the Free Software Foundation; either version 2 of the License, or
   13: # (at your option) any later version.
   14: #
   15: # LON-CAPA is distributed in the hope that it will be useful,
   16: # but WITHOUT ANY WARRANTY; without even the implied warranty of
   17: # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   18: # GNU General Public License for more details.
   19: #
   20: # You should have received a copy of the GNU General Public License
   21: # along with LON-CAPA; if not, write to the Free Software
   22: # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
   23: #
   24: # /home/httpd/html/adm/gpl.txt
   25: #
   26: # http://www.lon-capa.org/
   27: #
   28: ###
   29: 
   30: package LONCAPA;
   31: 
   32: use strict;
   33: use lib '/home/httpd/lib/perl/';
   34: use LONCAPA::Configuration;
   35: use Fcntl qw(:flock);
   36: use GDBM_File;
   37: use POSIX;
   38: 
   39: my $loncapa_max_wait_time = 13;
   40: 
   41: require Exporter;
   42: our @ISA = qw (Exporter);
   43: our @EXPORT = qw(&add_get_param &escape &unescape &tie_domain_hash &untie_domain_hash &tie_user_hash &untie_user_hash &propath);
   44: my %perlvar;
   45: 
   46: 
   47: 
   48: # Inputs are a url, and a hash ref of
   49: # form name => value pairs
   50: # takes care of properly adding the form name elements and values to the 
   51: # the url doing proper escaping of the values and joining with ? or & as 
   52: # needed
   53: 
   54: sub add_get_param {
   55:     my ($url,$form_data) = @_;
   56:     my $needs_question_mark = ($url !~ /\?/);
   57: 
   58:     while (my ($name,$value) = each(%$form_data)) {
   59: 	if ($needs_question_mark) {
   60: 	    $url.='?';
   61: 	    $needs_question_mark = 0;
   62: 	} else { 
   63: 	    $url.='&';
   64: 	}
   65: 	$url.=$name.'='.&escape($form_data->{$name});
   66:     }
   67:     return $url;
   68: }
   69: 
   70: # -------------------------------------------------------- Escape Special Chars
   71: 
   72: sub escape {
   73:     my $str=shift;
   74:     $str =~ s/(\W)/"%".unpack('H2',$1)/eg;
   75:     return $str;
   76: }
   77: 
   78: # ----------------------------------------------------- Un-Escape Special Chars
   79: 
   80: sub unescape {
   81:     my $str=shift;
   82:     $str =~ s/%([a-fA-F0-9][a-fA-F0-9])/pack("C",hex($1))/eg;
   83:     return $str;
   84: }
   85: 
   86: # -------------------------------------------- Return path to profile directory
   87: 
   88: sub propath {
   89:     my ($udom,$uname)=@_;
   90:     $udom=~s/\W//g;
   91:     $uname=~s/\W//g;
   92:     my $subdir=$uname.'__';
   93:     $subdir =~ s/(.)(.)(.).*/$1\/$2\/$3/;
   94:     my $proname="$perlvar{'lonUsersDir'}/$udom/$subdir/$uname";
   95:     return $proname;
   96: } 
   97: 
   98: 
   99: #---------------------------------------------------------------
  100: #
  101: # Manipulation of hash based databases (factoring out common code
  102: # for later use as we refactor.
  103: #
  104: #  Ties a domain level resource file to a hash.
  105: #  If requested a history entry is created in the associated hist file.
  106: #
  107: #  Parameters:
  108: #     domain    - Name of the domain in which the resource file lives.
  109: #     namespace - Name of the hash within that domain.
  110: #     how       - How to tie the hash (e.g. GDBM_WRCREAT()).
  111: #     loghead   - Optional parameter, if present a log entry is created
  112: #                 in the associated history file and this is the first part
  113: #                  of that entry.
  114: #     logtail   - Goes along with loghead,  The actual logentry is of the
  115: #                 form $loghead:<timestamp>:logtail.
  116: # Returns:
  117: #    Reference to a hash bound to the db file or alternatively undef
  118: #    if the tie failed.
  119: #
  120: sub tie_domain_hash {
  121:     my ($domain,$namespace,$how,$loghead,$logtail) = @_;
  122:     
  123:     # Filter out any whitespace in the domain name:
  124:     
  125:     $domain =~ s/\W//g;
  126:     
  127:     # We have enough to go on to tie the hash:
  128:     
  129:     my $user_top_dir   = $perlvar{'lonUsersDir'};
  130:     my $domain_dir     = $user_top_dir."/$domain";
  131:     my $resource_file  = $domain_dir."/$namespace";
  132:     return &_locking_hash_tie($resource_file,$namespace,$how,$loghead,$logtail);
  133: }
  134: 
  135: sub untie_domain_hash {
  136:     return &_locking_hash_untie(@_);
  137: }
  138: #
  139: #   Ties a user's resource file to a hash.  
  140: #   If necessary, an appropriate history
  141: #   log file entry is made as well.
  142: #   This sub factors out common code from the subs that manipulate
  143: #   the various gdbm files that keep keyword value pairs.
  144: # Parameters:
  145: #   domain       - Name of the domain the user is in.
  146: #   user         - Name of the 'current user'.
  147: #   namespace    - Namespace representing the file to tie.
  148: #   how          - What the tie is done to (e.g. GDBM_WRCREAT().
  149: #   loghead      - Optional first part of log entry if there may be a
  150: #                  history file.
  151: #   what         - Optional tail of log entry if there may be a history
  152: #                  file.
  153: # Returns:
  154: #   hash to which the database is tied.  It's up to the caller to untie.
  155: #   undef if the has could not be tied.
  156: #
  157: sub tie_user_hash {
  158:     my ($domain,$user,$namespace,$how,$loghead,$what) = @_;
  159: 
  160:     $namespace=~s/\//\_/g;	# / -> _
  161:     $namespace=~s/\W//g;		# whitespace eliminated.
  162:     my $proname     = &propath($domain, $user);
  163: 
  164:     my $file_prefix="$proname/$namespace";
  165:     return &_locking_hash_tie($file_prefix,$namespace,$how,$loghead,$what);
  166: }
  167: 
  168: sub untie_user_hash {
  169:     return &_locking_hash_untie(@_);
  170: }
  171: 
  172: # routines if you just have a filename
  173: # return tied hashref or undef
  174: 
  175: sub locking_hash_tie {
  176:     my ($filename,$how)=@_;
  177:     my ($file_prefix,$namespace)=&db_filename_parts($filename);
  178:     if ($namespace eq '') { return undef; }
  179:     return &_locking_hash_tie($file_prefix,$namespace,$how);
  180: }
  181: 
  182: sub locking_hash_untie {
  183:     return &_locking_hash_untie(@_);
  184: }
  185: 
  186: sub db_filename_parts {
  187:     my ($filename)=@_;
  188:     my ($file_path,$namespace)=($filename=~/^(.*)\/([^\/]+)\.db$/);
  189:     if ($namespace eq '') { return undef; }
  190:     return ($file_path.'/'.$namespace,$namespace);
  191: }
  192: 
  193: # internal routines that handle the actual tieing and untieing process
  194: 
  195: sub _do_hash_tie {
  196:     my ($file_prefix,$namespace,$how,$loghead,$what) = @_;
  197:     my %hash;
  198:     if(tie(%hash, 'GDBM_File', "$file_prefix.db", $how, 0640)) {
  199: 	# If this is a namespace for which a history is kept,
  200: 	# make the history log entry:    
  201: 	if (($namespace !~/^nohist\_/) && (defined($loghead))) {
  202: 	    my $hfh = IO::File->new(">>$file_prefix.hist"); 
  203: 	    if($hfh) {
  204: 		my $now = time();
  205: 		print $hfh ("$loghead:$now:$what\n");
  206: 	    }
  207: 	    $hfh->close;
  208: 	}
  209: 	return \%hash;
  210:     } else {
  211: 	return undef;
  212:     }
  213: }
  214: 
  215: sub _do_hash_untie {
  216:     my ($hashref) = @_;
  217:     my $result = untie(%$hashref);
  218:     return $result;
  219: }
  220: 
  221: {
  222:     my $sym;
  223:     my @pushed_syms;
  224: 
  225:     sub clean_sym {
  226: 	undef($sym);
  227:     }
  228:     sub push_locking_hash_tie {
  229: 	if (!defined($sym)) {
  230: 	    die("Invalid used of push_locking_hash_tie, should only be called after a lock has occurred and before and unlock.");
  231: 	}
  232: 	push(@pushed_syms,$sym);
  233: 	undef($sym);
  234:     }
  235: 
  236:     sub pop_locking_hash_tie {
  237: 	if (defined($sym)) {
  238: 	    die("Invalid nested used of pop_locking_hash_tie, should only be called after a unlock has occurred.");
  239: 	}
  240: 	$sym = pop(@pushed_syms);
  241:     }
  242: 
  243:     sub _locking_hash_tie {
  244: 	my ($file_prefix,$namespace,$how,$loghead,$what) = @_;
  245: 	if (defined($sym)) {
  246: 	    die('Nested locking attempted without proper use of push_locking_hash_tie, this is unsupported');
  247: 	}
  248: 
  249:         my $lock_type=LOCK_SH;
  250: # Are we reading or writing?
  251:         if ($how eq &GDBM_READER()) {
  252: # We are reading
  253:            if (!open($sym,"$file_prefix.db.lock")) {
  254: # We don't have a lock file. This could mean
  255: # - that there is no such db-file
  256: # - that it does not have a lock file yet
  257:                if ((! -e "$file_prefix.db") && (! -e "$file_prefix.db.gz")) {
  258: # No such file. Forget it.                
  259:                    $! = 2;
  260: 		   &clean_sym();
  261:                    return undef;
  262:                }
  263: # Apparently just no lock file yet. Make one
  264:                open($sym,">>$file_prefix.db.lock");
  265:            }
  266: # Do a shared lock
  267:            if (!&flock_sym(LOCK_SH)) { 
  268: 	       &clean_sym();
  269: 	       return undef; 
  270: 	   } 
  271: # If this is compressed, we will actually need an exclusive lock
  272: 	   if (-e "$file_prefix.db.gz") {
  273: 	       if (!&flock_sym(LOCK_EX)) {
  274: 		   &clean_sym();
  275: 		   return undef;
  276: 	       }
  277: 	   }
  278:         } elsif ($how eq &GDBM_WRCREAT()) {
  279: # We are writing
  280:            open($sym,">>$file_prefix.db.lock");
  281: # Writing needs exclusive lock
  282:            if (!&flock_sym(LOCK_EX)) {
  283: 	       &clean_sym();
  284: 	       return undef;
  285: 	   }
  286:         } else {
  287:            die("Unknown method $how for $file_prefix");
  288:         }
  289: # The file is ours!
  290: # If it is archived, un-archive it now
  291:        if (-e "$file_prefix.db.gz") {
  292:            system("gunzip $file_prefix.db.gz");
  293: 	   if (-e "$file_prefix.hist.gz") {
  294: 	       system("gunzip $file_prefix.hist.gz");
  295: 	   }
  296:        }
  297: # Change access mode to non-blocking
  298:        $how=$how|&GDBM_NOLOCK();
  299: # Go ahead and tie the hash
  300:       	my $result = 
  301: 	    &_do_hash_tie($file_prefix,$namespace,$how,$loghead,$what);
  302: 	if (!$result) {
  303: 	    &clean_sym();
  304: 	}
  305: 	return $result;
  306:     }
  307: 
  308:     sub flock_sym {
  309:         my ($lock_type)=@_;
  310: 	my $failed=0;
  311: 	eval {
  312: 	    local $SIG{__DIE__}='DEFAULT';
  313: 	    local $SIG{ALRM}=sub {
  314: 		$failed=1;
  315: 		die("failed lock");
  316: 	    };
  317: 	    alarm($loncapa_max_wait_time);
  318: 	    flock($sym,$lock_type);
  319: 	    alarm(0);
  320: 	};
  321: 	if ($failed) {
  322: 	    $! = 100; # throwing error # 100
  323: 	    return undef;
  324: 	} else {
  325: 	    return 1;
  326: 	}
  327:     }
  328: 
  329:     sub _locking_hash_untie {
  330: 	my ($hashref) = @_;
  331: 	my $result = untie(%$hashref);
  332: 	flock($sym,LOCK_UN);
  333: 	close($sym);
  334: 	&clean_sym();
  335: 	return $result;
  336:     }
  337: }
  338: 
  339: BEGIN {
  340:     %perlvar=%{&LONCAPA::Configuration::read_conf('loncapa.conf')};
  341: }
  342: 
  343: 1;
  344: 
  345: __END__
  346: 
  347: =pod
  348: 
  349: =head1 NAME
  350: 
  351: LONCAPA - Basic routines
  352: 
  353: =head1 SYNOPSIS
  354: 
  355: Generally useful routines
  356: 
  357: =head1 EXPORTED SUBROUTINES
  358: 
  359: =over 4
  360: 
  361: =item *
  362: 
  363: escape() : unpack non-word characters into CGI-compatible hex codes
  364: 
  365: =item *
  366: 
  367: unescape() : pack CGI-compatible hex codes into actual non-word ASCII character
  368: 
  369: =item *
  370: 
  371: add_get_param() :
  372:  Inputs:  url (with or without exit GET from parameters), hash ref of
  373:               form name => value pairs
  374: 
  375:  Return: url with properly added the form name elements and values to the 
  376:          the url doing proper escaping of the values and joining with ? or &
  377:          as needed
  378: 
  379: =back

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>