--- gforth/extend.fs	1995/10/16 18:33:07	1.11
+++ gforth/extend.fs	1995/12/04 16:38:52	1.14
@@ -1,5 +1,24 @@
 \ EXTEND.FS    CORE-EXT Word not fully tested!         12may93jaw
 
+\ Copyright (C) 1995 Free Software Foundation, Inc.
+
+\ This file is part of Gforth.
+
+\ Gforth 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., 675 Mass Ave, Cambridge, MA 02139, USA.
+
+
 \ May be cross-compiled
 
 decimal
@@ -110,25 +129,59 @@ decimal
 \ SOURCE-ID SAVE-INPUT RESTORE-INPUT                    11jun93jaw
 
 : source-id ( -- 0 | -1 | fileid ) \ core-ext source-i-d
-  loadfile @ dup 0= IF  drop loadline @ 0 min  THEN ;
+  loadfile @ dup 0= IF  drop sourceline# 0 min  THEN ;
 
 : save-input ( -- x1 .. xn n ) \ core-ext
-  >in @
-  loadfile @ ?dup
-  IF    dup file-position throw loadline @ >tib @ 6
-        #tib @ >tib +!
-  ELSE  loadline @ blk @ linestart @ >tib @ 5 THEN
-;
+    >in @
+    loadfile @
+    if
+	loadfile @ file-position throw
+    else
+	blk @
+	linestart @
+    then
+    sourceline#
+    >tib @
+    source-id
+    6 ;
 
 : restore-input ( x1 .. xn n -- flag ) \ core-ext
-  swap >tib !
-  6 = IF   loadline ! rot dup loadfile !
-           reposition-file IF drop true EXIT THEN
-      ELSE linestart ! blk !
-           dup loadline @ <> IF 2drop true EXIT THEN
-           loadline !
-      THEN
-  >in ! false ;
+    6 <> -12 and throw
+    source-id <> -12 and throw
+    >tib !
+    >r ( line# )
+    loadfile @ 0<>
+    if
+	loadfile @ reposition-file throw
+    else
+	linestart !
+	blk !
+	sourceline# r@ <> blk @ 0= and loadfile @ 0= and
+	if
+	    drop rdrop true EXIT
+	then
+    then
+    r> loadline !
+    >in !
+    false ;
+
+\ : save-input ( -- x1 .. xn n ) \ core-ext
+\   >in @
+\   loadfile @ ?dup
+\   IF    dup file-position throw sourceline# >tib @ 6
+\         #tib @ >tib +!
+\   ELSE  sourceline# blk @ linestart @ >tib @ 5 THEN
+\ ;
+
+\ : restore-input ( x1 .. xn n -- flag ) \ core-ext
+\   swap >tib !
+\   6 = IF   loadline ! rot dup loadfile !
+\            reposition-file IF drop true EXIT THEN
+\       ELSE linestart ! blk !
+\            dup sourceline# <> IF 2drop true EXIT THEN
+\            loadline !
+\       THEN
+\   >in ! false ;