--- loncom/auth/migrateuser.pm	2022/06/18 01:01:42	1.25.2.5.2.5
+++ loncom/auth/migrateuser.pm	2022/07/08 15:51:15	1.25.2.5.2.6
@@ -1,7 +1,7 @@
 # The LearningOnline Network
 # Starts a user off based of an existing token.
 #
-# $Id: migrateuser.pm,v 1.25.2.5.2.5 2022/06/18 01:01:42 raeburn Exp $
+# $Id: migrateuser.pm,v 1.25.2.5.2.6 2022/07/08 15:51:15 raeburn Exp $
 #
 # Copyright Michigan State University Board of Trustees
 #
@@ -193,7 +193,7 @@ sub ip_changed {
                     'username'        => $dataref->{'username'},
                     'sessionserver'   => $lonhost,
                   );
-        my @names = ('origurl','symb','role','linkprot','linkkey');
+        my @names = ('origurl','symb','role','linkprotuser','linkprotexit','linkprot','linkkey');
         foreach my $name (@names) {
             if ($dataref->{$name} ne '') {
                 $info{$name} = $dataref->{$name};
@@ -216,7 +216,7 @@ sub ip_changed {
 }
 
 sub logout {
-    my ($r,$ip,$handle,$data,$lti_env) = @_;
+    my ($r,$ip,$handle,$data,$lti_env,$linkprot_info) = @_;
     my $lonidsdir=$r->dir_config('lonIDsDir');
     if (unlink("$lonidsdir/$handle.id")) {
         if (($env{'user.linkedenv'} =~ /^[a-f0-9]+_linked$/) &&
@@ -243,15 +243,8 @@ sub logout {
                                 -expires => '-10y',);
         $r->headers_out->add('Set-cookie' => $c);
     }
-    my (%info,%user_info,%lti_info);
-    if (ref($lti_env) eq 'HASH') {
-        foreach my $key (sort(keys(%{$lti_env}))) {
-            if ($key =~ /^request\.(.+)$/) {
-                $lti_info{$1} = $lti_env->{$key};
-            }
-        }
-    }
     my $lonhost = $r->dir_config('lonHostID');
+    my ($url,$token,$delay,%info,%user_info);
     if (ref($data) eq 'HASH') {
         %user_info=('ip'       => $ip,
                     'domain'   => $data->{'domain'},
@@ -262,13 +255,39 @@ sub logout {
                     'symb'     => $data->{'symb'},
                     'server'   => $lonhost);
     }
-    %info = (%user_info,%lti_info);
-    my $token = &Apache::lonnet::tmpput(\%info,$lonhost);
-    my $url = '/adm/migrateuser?token='.$token;
+    $delay = 0.1;
+    if (ref($linkprot_info) eq 'HASH') {
+        if ($linkprot_info->{'linkprotuser'} ne '') {
+            $token = &Apache::lonnet::tmpput($linkprot_info,$lonhost);
+            $url = '/adm/login?ltoken='.$token;
+        } else {
+            foreach my $key (keys(%user_info)) {
+                if ($user_info{$key} eq '') {
+                    delete($user_info{$key});
+                }
+            }
+            %info = (%user_info,%{$linkprot_info});
+            $token = &Apache::lonnet::tmpput(\%info,$lonhost);
+            $url = '/adm/migrateuser?token='.$token;
+        }
+        $delay = 0;
+    } else {
+        my %lti_info;
+        if (ref($lti_env) eq 'HASH') {
+            foreach my $key (sort(keys(%{$lti_env}))) {
+                if ($key =~ /^request\.(.+)$/) {
+                    $lti_info{$1} = $lti_env->{$key};
+                }
+            }
+        }
+        %info = (%user_info,%lti_info);
+        $token = &Apache::lonnet::tmpput(\%info,$lonhost);
+        $url = '/adm/migrateuser?token='.$token;
+    }
     $r->send_http_header;
     $r->print(
         &Apache::loncommon::start_page('Updating Session ...',undef,
-                                       {'redirect'       => [0.1,$url],
+                                       {'redirect'       => [$delay,$url],
                                         'only_body'      => 1,}).
         &Apache::loncommon::end_page());
     if ($env{'request.balancercookie'}) {
@@ -383,6 +402,9 @@ sub handler {
                 unless ($env{'request.linkprot'} eq $data{'linkprot'}) {
                     $checklaunch = 1;
                 }
+                unless ($env{'request.linkprotexit'} eq $data{'linkprotexit'}) {
+                    $checklaunch = 1;
+                }
                 unless ($env{'request.linkkey'} eq $data{'linkkey'}) {
                     $checklaunch = 1;
                 }
@@ -394,15 +416,18 @@ sub handler {
                 if (($env{'user.name'} ne $data{'username'}) ||
                     ($env{'user.domain'} ne $data{'domain'})) {
                     my %linkprot_env;
-                    foreach my $item ('linkprot','deeplink.login') {
+                    foreach my $item ('linkprot','linkprotexit','deeplink.login') {
                         if ($data{$item}) {
-                            $linkprot_env{'request.'.$item} = $data{$item};
+                            $linkprot_env{$item} = $data{$item};
                         }
                     }
-                    &logout($r,$ip,$handle,\%data,\%linkprot_env);
+                    &logout($r,$ip,$handle,\%data,undef,\%linkprot_env);
                     return OK;
                 }
                 &Apache::lonnet::appenv({'request.linkprot' => $data{'linkprot'}});
+                if ($data{'linkprotexit'}) {
+                    &Apache::lonnet::appenv({'request.linkprotexit' => $data{'linkprotexit'}});
+                }
                 if ($env{'request.linkkey'}) {
                     &Apache::lonnet::delenv('request.linkkey');
                 }
@@ -468,6 +493,11 @@ sub handler {
                     my %info;
                     if ($env{'request.linkprot'}) {
                         $info{'linkprot'} = $env{'request.linkprot'};
+                        foreach my $item ('linkprotuser','linkprotexit') {
+                            if ($data{$item}) {
+                                $info{$item} = $data{$item};
+                            }
+                        }
                     } elsif ($env{'request.linkkey'} ne '') {
                         $info{'linkkey'} = $env{'request.linkkey'};
                     }
@@ -476,7 +506,7 @@ sub handler {
                         $info{'checklaunch'} = 1;
                     }
                     my $token = &Apache::lonnet::tmpput(\%info,$r->dir_config('lonHostID'),'link');
-                    unless (($token eq 'con_lost') || ($token eq 'refused') ||
+                    unless (($token eq 'con_lost') || ($token eq 'refused') || ($token =~ /^error:/) ||
                             ($token eq 'unknown_cmd') || ($token eq 'no_such_host')) {
                         $dest .= (($dest =~ /\?/) ? '&' : '?') . 'ttoken='.$token;
                     }
@@ -503,6 +533,14 @@ sub handler {
                                           'user.linkproturi' => $linkuri,
                                           'request.linkprot' => $data{'linkprot'}};
                         }
+                        if ($data{'linkprotexit'}) {
+                            $extra_env->{'request.linkprotexit'} = $data{'linkprotexit'};
+                        }
+                    }
+                    foreach my $item ('linkprotuser','linkprotexit') {
+                        if ($data{$item} ne '') {
+                            $form{$item} = $data{$item};
+                        }
                     }
                 } elsif ($data{'linkkey'}) {
                     if (ref($extra_env) eq 'HASH') {
@@ -542,6 +580,14 @@ sub handler {
         }
         if ($data{'linkprot'}) {
             $extra_env->{'request.linkprot'} = $data{'linkprot'};
+            if ($data{'linkprotexit'}) {
+                $extra_env->{'request.linkprotexit'} = $data{'linkprotexit'};
+            }
+            foreach my $item ('linkprotuser','linkprotexit') {
+                if ($data{$item}) {
+                    $form{'request.'.$item} = $data{$item};
+                }
+            }
         } elsif ($data{'linkkey'} ne '') {
             $extra_env->{'request.linkkey'} = $data{'linkkey'};
         }