diff options
Diffstat (limited to 'tex/context/third')
-rw-r--r-- | tex/context/third/enigma/enigma.lua | 196 |
1 files changed, 133 insertions, 63 deletions
diff --git a/tex/context/third/enigma/enigma.lua b/tex/context/third/enigma/enigma.lua index 79aadc6..371ddd9 100644 --- a/tex/context/third/enigma/enigma.lua +++ b/tex/context/third/enigma/enigma.lua @@ -67,6 +67,7 @@ libraries. \stopparagraph --ichd]]-- +local ioread = io.read local iowrite = io.write local mathfloor = math.floor local mathrandom = math.random @@ -272,6 +273,8 @@ local pprint_new_machine local pprint_rotor local pprint_rotor_scheme local pprint_step +local polite_key_request +local key_invalid do local eol = "\n" @@ -463,9 +466,47 @@ and only then pushes the output. end return 0 end +--[[ichd-- +\startparagraph +The \luafunction{polite_key_request} will be called in case the +\identifier{day_key} field of the machine setup is empty at the time of +initialization. +\stopparagraph +--ichd]]-- + local s_request = "\n\n " + .. underline"This is an encrypted document." .. [[ + + + Please enter the document key for enigma machine + “%s”. + + Key Format: + +Ref R1 R2 R3 I1 I2 I3 [P1 ..] Ref: reflector A/B/C + Rn: rotor, I through V + In: ring position, 01 through 26 + Pn: optional plugboard wiring, upto 32 + +>]] + polite_key_request = function (name) + return stringformat(s_request, colorize(name, 33)) + end + + local s_invalid_key = colorize"Warning!" + .. " The specified key is invalid." + key_invalid = function () + return s_invalid_key + end end +--[[ichd-- +\startparagraph +The functions \luafunction{new} and \luafunction{ask_for_day_key} are +used outside their scope, so we declare them beforehand. +\stopparagraph +--ichd]]-- local new +local ask_for_day_key do --[[ichd-- \stopdocsection @@ -756,7 +797,6 @@ extraction of successive characters from the sequence. result[#result+1] = tmp end end - print(str) machine:processed_chars() return tableconcat(result) end @@ -812,25 +852,25 @@ extraction of successive characters from the sequence. local p_init = P{ "init", - init = Ct(V"do_init"), + init = V"whitespace"^-1 * Ct(V"do_init"), do_init = V"reflector" * V"whitespace" - * V"rotors" * V"whitespace" - * V"ring" - * (V"whitespace" * V"plugboard")^-1 - , + * V"rotors" * V"whitespace" + * V"ring" + * (V"whitespace" * V"plugboard")^-1 + , reflector = Cg(C(R("ac","AC")) / stringlower, "reflector"), rotors = Cg(Ct(V"rotor" * V"whitespace" - * V"rotor" * V"whitespace" - * V"rotor"), - "rotors") - , + * V"rotor" * V"whitespace" + * V"rotor"), + "rotors") + , rotor = Cs(V"roman_five" / roman_digits + V"roman_four" / roman_digits + V"roman_three" / roman_digits + V"roman_two" / roman_digits + V"roman_one" / roman_digits) - , + , roman_one = P"I" + P"i", roman_two = P"II" + P"ii", roman_three = P"III" + P"iii", @@ -838,10 +878,10 @@ extraction of successive characters from the sequence. roman_five = P"V" + P"v", ring = Cg(Ct(V"double_digit" * V"whitespace" - * V"double_digit" * V"whitespace" - * V"double_digit"), + * V"double_digit" * V"whitespace" + * V"double_digit"), "ring") - , + , double_digit = C(R"02" * R"09"), plugboard = Cg(V"do_plugboard", "plugboard"), @@ -858,7 +898,7 @@ extraction of successive characters from the sequence. -- * V"letter_combination") do_plugboard = Ct(V"letter_combination" * (V"whitespace" * V"letter_combination")^0) - , + , letter_combination = C(R("az", "AZ") * R("az", "AZ")), whitespace = S" \n\t\v"^1, @@ -1027,14 +1067,27 @@ consists of three elements: local processed_chars = function (machine) emit(1, pprint_machine_step, machine.step, machine.name) end - new = function (setup_string, pattern) - local raw_settings = lpegmatch(p_init, setup_string) + + local handle_day_key handle_day_key = function (dk, name, old) + local result + if not dk or dk == "" then + dk = ask_for_day_key(name, old) + end + result = lpegmatch(p_init, dk) + -- If we don’t like the key we’re going to ask again. And again.... + return result or handle_day_key(nil, name, dk) + end + + new = function (name, setup_string, pattern) + --local raw_settings = lpegmatch(p_init, setup_string) + local raw_settings = handle_day_key(setup_string, name) local rotors, ring = get_rotors(raw_settings.rotors, raw_settings.ring) local plugboard = raw_settings.plugboard and get_plugboard_substitution(raw_settings.plugboard) or get_plugboard_substitution{ } local machine = { + name = name, step = 0, -- n characters encoded init = { rotors = raw_settings.rotors, @@ -1068,6 +1121,7 @@ consists of three elements: emit(1, pprint_new_machine, machine) return machine end + end --[[ichd-- \stopdocsection @@ -1078,7 +1132,9 @@ end %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% \startdocsection[title=Setup Argument Handling] %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% - +--ichd]]-- +do +--[[ichd-- \startparagraph As the module is intended to work both with the Plain and \LATEX\ formats as well as \CONTEXT, we can’t rely on format dependent setups. @@ -1087,47 +1143,47 @@ as all the functionality resides in Lua. \stopparagraph --ichd]]-- -local p_args = P{ - "args", - args = Cf(Ct"" * (V"kv_pair" + V"emptyline")^0, rawset), - kv_pair = Cg(V"key" - * V"separator" - * (V"value" * V"final" - + V"empty")) - * V"rest_of_line"^-1 - , - key = V"whitespace"^0 * C(V"key_char"^1), - key_char = (1 - V"whitespace" - V"eol" - V"equals")^1, - separator = V"whitespace"^0 * V"equals" * V"whitespace"^0, - empty = V"whitespace"^0 * V"comma" * V"rest_of_line"^-1 - * Cc(false) - , - value = C((V"balanced" + (1 - V"final"))^1), - final = V"whitespace"^0 * V"comma" + V"rest_of_string", - rest_of_string = V"whitespace"^0 * V"eol_comment"^-1 * V"eol"^0 * V"eof", - rest_of_line = V"whitespace"^0 * V"eol_comment"^-1 * V"eol", - eol_comment = V"comment_string" * (1 - (V"eol" + V"eof"))^0, - comment_string = V"lua_comment" + V"TeX_comment", - TeX_comment = V"percent", - lua_comment = V"double_dash", - emptyline = V"rest_of_line", - - balanced = V"balanced_brk" + V"balanced_brc", - balanced_brk = V"lbrk" * (V"balanced" + (1 - V"rbrk"))^0 * V"rbrk", - balanced_brc = V"lbrc" * (V"balanced" + (1 - V"rbrc"))^0 * V"rbrc", - - -- Terminals - eol = P"\n\r" + P"\r\n" + P"\n" + P"\r", -- users do strange things - eof = -P(1), - whitespace = S" \t\v", - equals = P"=", - dot = P".", - comma = P",", - dash = P"-", double_dash = V"dash" * V"dash", - percent = P"%", - lbrk = P"[", rbrk = P"]", - lbrc = P"{", rbrc = P"}", -} + local p_args = P{ + "args", + args = Cf(Ct"" * (V"kv_pair" + V"emptyline")^0, rawset), + kv_pair = Cg(V"key" + * V"separator" + * (V"value" * V"final" + + V"empty")) + * V"rest_of_line"^-1 + , + key = V"whitespace"^0 * C(V"key_char"^1), + key_char = (1 - V"whitespace" - V"eol" - V"equals")^1, + separator = V"whitespace"^0 * V"equals" * V"whitespace"^0, + empty = V"whitespace"^0 * V"comma" * V"rest_of_line"^-1 + * Cc(false) + , + value = C((V"balanced" + (1 - V"final"))^1), + final = V"whitespace"^0 * V"comma" + V"rest_of_string", + rest_of_string = V"whitespace"^0 * V"eol_comment"^-1 * V"eol"^0 * V"eof", + rest_of_line = V"whitespace"^0 * V"eol_comment"^-1 * V"eol", + eol_comment = V"comment_string" * (1 - (V"eol" + V"eof"))^0, + comment_string = V"lua_comment" + V"TeX_comment", + TeX_comment = V"percent", + lua_comment = V"double_dash", + emptyline = V"rest_of_line", + + balanced = V"balanced_brk" + V"balanced_brc", + balanced_brk = V"lbrk" * (V"balanced" + (1 - V"rbrk"))^0 * V"rbrk", + balanced_brc = V"lbrc" * (V"balanced" + (1 - V"rbrc"))^0 * V"rbrc", + + -- Terminals + eol = P"\n\r" + P"\r\n" + P"\n" + P"\r", -- users do strange things + eof = -P(1), + whitespace = S" \t\v", + equals = P"=", + dot = P".", + comma = P",", + dash = P"-", double_dash = V"dash" * V"dash", + percent = P"%", + lbrk = P"[", rbrk = P"]", + lbrc = P"{", rbrc = P"}", + } --[[ichd-- @@ -1138,7 +1194,6 @@ a sanitizer routine and, if so, apply it to its value. \stopparagraph --ichd]]-- -do local boolean_synonyms = { ["1"] = true, doit = true, @@ -1155,7 +1210,7 @@ do local ans = alpha + digit + space local p_ans = Cs((ans + (1 - ans / ""))^1) local alphanum_or_space = function (str) - if type(str) ~= "string" then return "" end + if type(str) ~= "string" then return nil end return lpegmatch(p_ans, str) end local ensure_int = function (n) @@ -1188,6 +1243,21 @@ do end return args end +--[[ichd-- +\startparagraph +If the machine setting lacks key settings then we’ll go ahead and ask +the user directly, hence the function \luafunction{ask_for_day_key}. +\stopparagraph +--ichd]]-- + ask_for_day_key = function (name, old) + if old then + emit(0, key_invalid) + end + emit(0, polite_key_request, name) + local result = ioread() + iowrite("\n") + return alphanum_or_space(result) or ask_for_day_key(name) + end end --[[ichd-- @@ -1279,10 +1349,10 @@ end enigma.save_raw_args = save_raw_args enigma.retrieve_raw_args = retrieve_raw_args + local new_machine = function (args, name) verbose_level = args.verbose - local machine = new(args.day_key, args.rotor_setting) - machine.name = name + local machine = new(name, args.day_key, args.rotor_setting) return machine end |