91 lines
3.6 KiB
Plaintext
91 lines
3.6 KiB
Plaintext
using System;
|
|
using System.IO;
|
|
using System.Text;
|
|
using System.Collections.Generic;
|
|
|
|
namespace AniNIX.Crypto {
|
|
public class ColumnTransposition : Cipher {
|
|
|
|
public override String Description() { return "Column Transposition cipher suite\nFormat is col <command> key1 [key2...]\nThe key format is any word to use for the transposition.\nThis cipher will use an irregular columnar transposition, without padding the input string.\n"; }
|
|
public override String Command() { return "col"; }
|
|
|
|
public ColumnTransposition(Workbench w) : base (w) {}
|
|
public ColumnTransposition() {}
|
|
|
|
private int[] GetColumnOrder(String key) {
|
|
List<char> orderList = new List<char>();
|
|
for (int i = 0; i < key.Length; i++) {
|
|
orderList.Add(key[i]);
|
|
}
|
|
orderList.Sort();
|
|
char[] charArr = orderList.ToArray();
|
|
int[] returnOrderIndexes = new int[key.Length];
|
|
Console.Write("Found key order: ");
|
|
for (int i = 0; i < key.Length; i++) {
|
|
for (int j = 0; j < charArr.Length; j++) {
|
|
if (key[i] == charArr[j]) {
|
|
Console.Write(String.Format("{0} ",j+1));
|
|
returnOrderIndexes[j] = i;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
Console.WriteLine("");
|
|
return returnOrderIndexes;
|
|
}
|
|
|
|
public override String Encrypt(String workSpace, String cipher, String[] line) {
|
|
if (line.Length < 3) {
|
|
Console.Error.WriteLine("Bad formatting.");
|
|
return workSpace;
|
|
}
|
|
String workSpaceNoNewline = workSpace.Replace("\n","");
|
|
char[] changed = workSpaceNoNewline.ToCharArray();
|
|
CharGrid cg = new CharGrid(workSpaceNoNewline,line[2].Length,false);
|
|
char[][] encryptionGrid = cg.ToArray();
|
|
int[] keyOrder = GetColumnOrder(line[2]);
|
|
Console.Write(cg.ToString());
|
|
int k = 0;
|
|
for (int j = 0; j < encryptionGrid[0].Length; j++) {
|
|
for (int i = 0; i < encryptionGrid.Length; i++) {
|
|
if (i != (encryptionGrid.Length-1) || keyOrder[j] < encryptionGrid[i].Length) {
|
|
changed[k] = encryptionGrid[i][keyOrder[j]];
|
|
k++;
|
|
}
|
|
}
|
|
}
|
|
String toReturn = new String(changed);
|
|
for (k = 0; k < workSpace.Length; k++) {
|
|
if (workSpace[k] == '\n') {
|
|
toReturn = toReturn.Insert(k,"\n");
|
|
}
|
|
}
|
|
return toReturn;
|
|
}
|
|
|
|
// TODO
|
|
public override String Decrypt(String workSpace, String cipher, String[] line) {
|
|
if (line.Length < 3) {
|
|
Console.Error.WriteLine("Bad formatting.");
|
|
return workSpace;
|
|
}
|
|
String workSpaceNoNewline = workSpace.Replace("\n","");
|
|
int[] keyOrder = GetColumnOrder(line[2]);
|
|
CharGrid cg = new CharGrid(workSpaceNoNewline,line[2].Length,keyOrder);
|
|
Console.Write(cg.ToString());
|
|
char[][] cgArray = cg.ToArray();
|
|
StringBuilder sb = new StringBuilder();
|
|
for (int i=0; i < cgArray.Length; i++) {
|
|
sb.Append(new String(cgArray[i]));
|
|
}
|
|
String toReturn = sb.ToString();
|
|
for (int i=0; i < workSpace.Length; i++) {
|
|
if (workSpace[i] == '\n') {
|
|
toReturn = toReturn.Insert(i,"\n");
|
|
}
|
|
}
|
|
return toReturn;
|
|
}
|
|
}
|
|
}
|