--- loncom/homework/lonmaxima.pm	2007/07/11 18:50:33	1.17
+++ loncom/homework/lonmaxima.pm	2017/03/06 17:01:59	1.33
@@ -1,7 +1,7 @@
 # The LearningOnline Network with CAPA
 # Interface routines to MAXIMA CAS
 #
-# $Id: lonmaxima.pm,v 1.17 2007/07/11 18:50:33 albertel Exp $
+# $Id: lonmaxima.pm,v 1.33 2017/03/06 17:01:59 raeburn Exp $
 #
 # Copyright Michigan State University Board of Trustees
 #
@@ -51,7 +51,7 @@ sub maximareply {
 	print $socket &escape($cmd)."\n";
         my $reply=<$socket>;
         chomp($reply);
-        if ($reply=~/^Incorrect/) { $reply='Error: '.$reply; }
+        if ($reply=~/^Incorrect/i) { $reply='Error: '.$reply; }
         return &unescape($reply);
     } else {
         return 'Error: no connection.';
@@ -60,19 +60,51 @@ sub maximareply {
 
 sub blacklisted {
     my ($cmd)=@_;
-    foreach my $forbidden ('save','load','plot','lisp','includ','compil',
-			   'file','batch','stringout','translat','stout',
-			   'stin','block','system','concat') {
+    foreach my $forbidden (
+        '\? ','\?','%i\d+','%o','batch','block'
+       ,'compil','concat','describe','display2d','file','inchar'
+       ,'includ','lisp','load','outchar','plot','quit'
+       ,'read','reset','save','stin','stout','stringout'
+       ,'system','translat','ttyoff','with_stdout','writefile'
+     ) {
 	if ($cmd=~/$forbidden/s) { return 1; }
     } 
     return 0;
 }
 
+sub maxima_allowed_libraries {
+   return (
+      "absimp","affine","atensor","atrig1","augmented_lagrangian","contrib_ode","ctensor","descriptive","diag",
+      "eigen","facexp","fft","fourie","functs","ggf","grobner","impdiff","ineq","interpol","itensor","lapack",
+      "lbfgs","lindstedt","linearalgebra","lsquares","makeOrders","mnewton","mchrpl","ntrig","orthopoly",
+      "quadpack","rducon","romberg","scifac","simplex","solve_rec","sqdnst","stats","sterling","sym","units",
+      "vect","zeilberger");
+}
+
+sub maxima_is_allowed_library {
+    my ($library)=@_;
+    foreach my $allowed_library (&maxima_allowed_libraries()) {
+       if ($library eq $allowed_library) { return 1; }
+    }
+    return 0;
+}
+
 sub runscript {
-    my ($socket,$fullscript)=@_;
+    my ($socket,$fullscript,$libraries)=@_;
     if (&blacklisted($fullscript)) { return 'Error: blacklisted'; }
     my $reply;
     $fullscript=~s/[\n\r\l]//gs;
+    if ($libraries) {
+       foreach my $library (split(/\s*\,\s*/,$libraries)) {
+          unless ($library=~/\w/) { next; }
+          if (&maxima_is_allowed_library($library)) {
+              $reply=&maximareply($socket,'load('.$library.')$'."\n");
+              if ($reply=~/^Error\:/) { return $reply; }
+          } else { 
+             return 'Error: blacklisted'; 
+          }
+       }
+    }
     foreach my $line (split(/\;/s,$fullscript)) {
 	if ($line=~/\w/) { $reply=&maximareply($socket,$line.";\n"); }
 	if ($reply=~/^Error\:/) { return $reply; }
@@ -85,11 +117,13 @@ sub runscript {
 
 sub maxima_cas_formula_fix {
    my ($expression)=@_;
-   return &Apache::response::implicit_multiplication($expression);
+   $expression=&Apache::response::implicit_multiplication($expression);
+   $expression=~s/([^\w\%])pi/$1\%pi/gs;
+   return $expression;
 }
 
 sub maxima_run {
-    my ($script,$submission,$argument) = @_;
+    my ($script,$submission,$argument,$libraries) = @_;
     my $socket=&connect();
     my @submissionarray=split(/\s*\,\s*/,$submission);
     for (my $i=0;$i<=$#submissionarray;$i++) {
@@ -103,7 +137,7 @@ sub maxima_run {
         my $fixedargument=&maxima_cas_formula_fix($argumentarray[$i]);
         $script=~s/LONCAPALIST\[$n\]/$fixedargument/gs;
     }
-    my $reply=&runscript($socket,$script);
+    my $reply=&runscript($socket,$script,$libraries);
     &disconnect($socket);
     if ($reply=~/^\s*true\s*$/i) { return 'EXACT_ANS'; }
     if ($reply=~/^\s*false\s*$/i) { return 'INCORRECT'; } 
@@ -111,9 +145,9 @@ sub maxima_run {
 }
 
 sub maxima_eval {
-    my ($script) = @_;
+    my ($script,$libraries) = @_;
     my $socket=&connect();
-    my $reply=&runscript($socket,$script);
+    my $reply=&runscript($socket,$script,$libraries);
     &disconnect($socket);
     return $reply;
 }
@@ -121,6 +155,9 @@ sub maxima_eval {
 
 sub compareterms {
     my ($socket,$terma,$termb)=@_;
+    if ($terma =~ m{[.*^/\s]+(|\(\s*\((|[^(]+))$}) {
+        return 'Error: blacklisted';
+    }
     my $difference=$terma.'-('.$termb.')';
     if (&blacklisted($difference)) { return 'Error: blacklisted'; }
     my $reply=&maximareply($socket,'trigsimp(trigreduce('.$difference.'));');
@@ -137,6 +174,8 @@ sub maxima_check {
     # integer to string mappings come from capaParser.h
     # 1 maps to 'EXACT_ANS'
     if ($reply eq 'true') { return 1; }
+    # 11 maps to 'BAD_FORMULA'
+    if ($reply=~/^Error\:/) { return 11; }
     # 7 maps to 'INCORRECT'
     return 7;
 }