--- loncom/lonenc.pm	2004/03/31 04:38:06	1.1
+++ loncom/lonenc.pm	2005/01/19 18:16:35	1.10
@@ -1,7 +1,7 @@
 # The LearningOnline Network
 # URL translation for encrypted filenames
 #
-# $Id: lonenc.pm,v 1.1 2004/03/31 04:38:06 www Exp $
+# $Id: lonenc.pm,v 1.10 2005/01/19 18:16:35 albertel Exp $
 #
 # Copyright Michigan State University Board of Trustees
 #
@@ -34,6 +34,7 @@ use Apache::lonnet();
 use Apache::File();
 use Apache::loncommon;
 use Crypt::IDEA;
+use Time::HiRes qw(gettimeofday);
 
 sub handler {
     my $r = shift;
@@ -44,51 +45,123 @@ sub handler {
 	my $handle=$lonid->value;
         $handle=~s/\W//g;
         my $lonidsdir=$r->dir_config('lonIDsDir');
+	$ENV{'request.enc'}=1;
         if ((-e "$lonidsdir/$handle.id") && ($handle ne '')) {
 # Initialize Environment
             &Apache::lonnet::transfer_profile_to_env($lonidsdir,$handle);
 # Decrypt URL and redirect
-	    $r->internal_redirect(&unencrypted($r->uri));
+	    my $redirect=&unencrypted($r->uri);
+	    if ($r->args) { $redirect.='?'.$r->args; }
+	    $r->internal_redirect($redirect);
 	    return OK;
 	} 
     }
     return FORBIDDEN;
 }
 
+sub encryptseed {
+    my $seed=$ENV{'course.'.$ENV{'request.course.id'}.'.internal.encseed'};
+    $seed=~s/[^0-9a-f]/0/g;
+    $seed.='0123456789abcdef';
+    $seed=substr($seed.$seed,0,32);
+    return pack("H32",$seed);
+}
+
 sub unencrypted {
     my $uri=shift;
     $uri=~s/^\/enc\/(\d+)\///;
-    &Apache::lonnet::logthis($uri);
     my $cmdlength=$1;
-    unless ($ENV{'course.'.$ENV{'request.course.id'}.'.internal.encseed'}) {
+    my $seed=&encryptseed();
+    unless ($seed) {
 	return '/'.$uri;
     }
     $uri=&Apache::lonnet::unescape($uri);
-    my $cipher=
-	new IDEA $ENV{'course.'.$ENV{'request.course.id'}.'.internal.encseed'};
-    &Apache::lonnet::logthis($ENV{'course.'.$ENV{'request.course.id'}.'.internal.encseed'});
+    my $cipher=new IDEA $seed;
     my $decuri='';
     for (my $encidx=0;$encidx<length($uri);$encidx+=16) {
 	$decuri.=$cipher->decrypt(
 				  pack("H16",substr($uri,$encidx,16))
 				  );
     }
+    $ENV{'request.enc'}=1;
+    $decuri=&remove_noise($decuri);
     return substr($decuri,0,$cmdlength);
 }
 
+# add a randomish character after every 4th caharacter
+sub add_noise {
+    my ($uri)=@_;
+    my @noise=split(/(.)/,(&gettimeofday())[1]);
+    my $noisy;
+    my $i;
+    foreach my $chunk (split(/(....)/,$uri)) {
+	$noisy.=$chunk;
+	$noisy.=$noise[($i++)%(scalar@noise)];
+    }
+    return $noisy;
+}
+
+# remove every fifth character
+sub remove_noise {
+    my ($uri)=@_;
+    my $clean;
+    foreach my $chunk (split(/(....)./,$uri)) { $clean.=$chunk; }
+    return $clean;
+}
+
 sub encrypted {
     my $uri=shift;
+    if ($ENV{'request.role.adv'}) { return($uri); }
+    my $seed=&encryptseed();
+    unless ($seed) {
+	return $uri;
+    }
     my $cmdlength=length($uri);
+    # add noise before enc so that that same url's look different
+    $uri=&add_noise($uri);
+    my $noiselength=length($uri);
+    $uri.=time;
     my $encuri='';
-    my $cipher=
-	new IDEA $ENV{'course.'.$ENV{'request.course.id'}.'.internal.encseed'};
-    for (my $encidx=0;$encidx<=$cmdlength;$encidx+=8) {
+    my $cipher=new IDEA $seed;
+    for (my $encidx=0;$encidx<=$noiselength;$encidx+=8) {
 	$encuri.=unpack("H16",
 			$cipher->encrypt(substr($uri,$encidx,8)));
     }
     return '/enc/'.$cmdlength.'/'.&Apache::lonnet::escape($encuri);
 }
 
+sub check_encrypt {
+    my $str=shift;
+    if ($ENV{'request.enc'}) { return &Apache::lonenc::encrypted($str); }
+    return $str;
+}
+
+sub check_decrypt {
+    my ($str)=@_;
+    if (ref($str)) {
+	if ($$str=~m|^/enc/|) { $$str=&Apache::lonenc::unencrypted($$str); }
+	return;
+    }
+    if ($str=~m|^/enc/|) { return &Apache::lonenc::unencrypted($str); }
+    return $str;
+}
+
+sub encrypt_ref {
+    my ($token,$elements)=@_;
+    my $html;
+    if ($ENV{'request.enc'}) {
+	while (my ($name,$value)= each(%{ $elements })) {
+	    if (!$value) { next; }
+	    my $href=&Apache::lonnet::hreflocation($Apache::lonxml::pwd[-1],$value);
+	    if ($href !~ /^http:/) { $href=&Apache::lonenc::encrypted($href); }
+	    $token->[2]->{$name}=$href;
+	}
+	$html = &Apache::edit::rebuild_tag($token);
+    } else {
+	$html = $token->[4];
+    }
+    return $html;
+}
 1;
 __END__