Исправьте задачу на шифровку и расшифровку

[Анна997]

Исправьте пожалуйста, нужно , что бы в задачи выбиралось 3 режима :
1) шифрование
2)расшифровка
3)выход
(программа работает, и кроме как режимов , ничего исправлять не нужно)
const n=30;
var a,b:array[1..n] of integer;
s,s1,s2,r:string;
i, j, p, key:byte;
m:set of byte;
begin
writeln('Введите текст для шифрования: ');
readln(s);
writeln('-------------------------------------------------');
write('Введите количество цифр в ключе от 2 до ',n,' key= ');
readln(key);
while length(s) mod key<>0 do s:=s+' ';
writeln('Введите ключ из ',key,' цифр: ');
m:=[];
for i:=1 to key do
begin
repeat
readln(p);
until(p in [1..key])and not(p in m);
a[i]:=p;
m:=m+[p];
end;
r:='';
i:=1;
while i<=length(s)-key+1 do
begin
s1:=copy(s,i,key);
s2:='';
for j:=1 to key do
s2:=s2+s1[a[j]];
r:=r+s2;
i:=i+key;
end;
for i:=length(r) downto 1 do
if r[i]=' ' then delete(r,i,1);
writeln('--------------------------------------------------');
writeln('Зашифрованный текст: ');
writeln(r);
//создадим ключ для дешифровки
for i:=1 to key do
begin
for j:=1 to key do
if a[j]=i then b[i]:=j
end;
s:=r;
while length(s) mod key<>0 do s:=s+' ';
r:='';
i:=1;
while i<=length(s)-key+1 do
begin
s1:=copy(s,i,key);
s2:='';
for j:=1 to key do
s2:=s2+s1[b[j]];
r:=r+s2;
i:=i+key;
end;
for i:=length(r) downto 1 do
if r[i]=' ' then delete(r,i,1);
writeln('Расшифрованный текст');
writeln(r);
End.

[evg_m]

Сообщение от Анна997

Исправьте пожалуйста, нужно , что бы в задачи выбиралось 3 режима :
1) шифрование
2)расшифровка
3)выход

const 
  n=30;

procedure DoShifr;
var 
  a,b:array[1..n] of integer;
  s,s1,s2,r:string;
  i, j, p, key:byte;
  m:set of byte;
begin
// ЗДЕСЬ код шифрования
end;

procedure DoRashifr;
var 
  a,b:array[1..n] of integer;
  s,s1,s2,r:string;
  i, j, p, key:byte;
  m:set of byte;
begin
// А ЗДЕСЬ код расшифрования
end;

var
  vkey:  integer;
begin
  repeat
    writeln('укажите режим работы:1-шифр. 2-расш. 0-закончить работу:');
    readln(vkey);
    case vkey of
    1: DoShifr;
    2: DoRashifr;
    0: ;//ничего не делаем и выходим по until
    else writeln('выбран несуществующий режим работы');
    end;
  until vkey<>0;
end.

(программа работает, и кроме как режимов , ничего исправлять не нужно)

// ВЕСЬ этот(последующий) код НАДО "распихать" по ДВУМ процедурам 
//DoShifr --код для шифрования
//DOrashifr --код для расшифрования

  writeln('Введите текст для шифрования: ');
  readln(s);
  writeln('-------------------------------------------------');
  write('Введите количество цифр в ключе от 2 до ',n,' key= ');
  readln(key);
  while length(s) mod key<>0 do s:=s+' ';
  writeln('Введите ключ из ',key,' цифр: ');
  m:=[];
  for i:=1 to key do begin
    repeat
      readln(p);
    until(p in [1..key])and not(p in m);
    a[i]:=p;
   m:=m+[p];
  end;

  r:='';
  i:=1;
  while i<=length(s)-key+1 do begin
    s1:=copy(s,i,key);
    s2:='';
    for j:=1 to key do
      s2:=s2+s1[a[j]];
    r:=r+s2;
    i:=i+key;
  end;
  for i:=length(r) downto 1 do
    if r[i]=' ' then delete(r,i,1);
  
  writeln('--------------------------------------------------');
  writeln('Зашифрованный текст: ');
  writeln(r);

  //создадим ключ для дешифровки
  for i:=1 to key do begin
    for j:=1 to key do
      if a[j]=i then b[i]:=j
  end;
  
  s:=r;
  while length(s) mod key<>0 do s:=s+' ';
  r:='';
  i:=1;
  while i<=length(s)-key+1 do begin
    s1:=copy(s,i,key);
    s2:='';
    for j:=1 to key do
      s2:=s2+s1[b[j]];
    r:=r+s2;
    i:=i+key;
  end;
  for i:=length(r) downto 1 do
    if r[i]=' ' then delete(r,i,1);

  writeln('Расшифрованный текст');
  writeln(r);