You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

snf_engine.cpp 41KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801
  1. // snf_engine.cpp
  2. //
  3. // (C) 1985-2004 MicroNeil Research Corporation
  4. // (C) 2005-2009 ARM Research Labs, LLC
  5. // See www.armresearch.com for the copyright terms.
  6. //
  7. // Derived from original work on cellular automation for complex pattern
  8. // reflex engine 1985 Pete McNeil (Madscientist)
  9. //
  10. // Derived from rapid scripting engine (token matrix) implementation 1987
  11. //
  12. // 20040419 _M Adding Verify() method. Beginning with version 2-3 of Message Sniffer
  13. // we are embedding a Mangler digest of the rulebase file. The Verify() method reconstructs
  14. // the digest and compares it. This ensures that no part of the rulebase file can be
  15. // corrupted without the snf2check utility detecting the problem. Prior to this version
  16. // it was possible to have undetected corruption in the middle of the rulebase file. The
  17. // Mangler digest will prevent that.
  18. // 20030130 _M Added testing section in TokenMatrix to throw an exeption if the file
  19. // is too small to be a valid matrix. The value is calculated based on the idea that a
  20. // valid matrix will have been encrypted in two segments so the file must be at least
  21. // as large as these two segments. This is intended to solve the zero-length-rulebase
  22. // bug where an access violation would occur if the file was of zero length.
  23. // 20021030 _M Creation of snf_engine module by dragging the sniffer pattern matching engine out
  24. // of the sniffer.cpp file.
  25. #include <unistd.h>
  26. #include <cstdio>
  27. #include <cctype>
  28. #include <ctime>
  29. #include <cstdlib>
  30. #include <fstream>
  31. #include <iostream>
  32. #include <string>
  33. #include <vector>
  34. #include "CodeDweller/mangler.hpp"
  35. #include "SNFMulti/snf_engine.hpp"
  36. using namespace std;
  37. using namespace CodeDweller;
  38. namespace SNFMulti {
  39. ///////////////////////////////////////////////////////////////////////////////////////////
  40. // BEGIN IMPLEMENTATIONS //////////////////////////////////////////////////////////////////
  41. ///////////////////////////////////////////////////////////////////////////////////////////
  42. ///////////////////////////////////////////////////////////////////////////////////////////
  43. // Token Matrix Implementations ///////////////////////////////////////////////////////////
  44. // TokenMatrix::Load(filename)
  45. void TokenMatrix::Load(string& FileName) { // Initialize using a string for file name.
  46. Load(FileName.c_str()); // Convert the string to a null terminated
  47. } // char* and call the function below.
  48. void TokenMatrix::Load(const char* FileName) { // Initializes the token matrix by file name.
  49. ifstream MatrixFile(FileName,ios::binary); // Open the file.
  50. if(!MatrixFile.good()) // If anything is wrong with the file
  51. throw BadFile("TokenMatrix::Load()(!MatrixFile.good())"); // then throw a bad file exception.
  52. Load(MatrixFile); // Load the matrix from the file.
  53. MatrixFile.close(); // Be nice and clean up our file.
  54. }
  55. // TokenMatrix::Load(stream)
  56. const AbortCheck CompatibleIntSizeCheck("TokenMatrix::Load():CompatibleIntSizeCheck(sizeof(unsigned int)==4)");
  57. void TokenMatrix::Load(ifstream& F) { // Initializes the token matrix from a file.
  58. CompatibleIntSizeCheck(sizeof(unsigned int)==4); // Check our assumptions.
  59. MatrixSize = 0; // Clear out the old Matrix Size and array.
  60. if(Matrix) delete Matrix; // that is, if there is an array.
  61. F.seekg(0,ios::end); // Find the end of the file.
  62. MatrixSize = F.tellg() / sizeof(Token); // Calculate how many tokens.
  63. F.seekg(0); // Go back to the beginning.
  64. if(MatrixSize < MinimumValidMatrix) // If the matrix file is too small then
  65. throw BadMatrix("TokenMatrix::Load() (MatrixSize < MinimumValidMatrix)"); // we must reject it.
  66. Matrix = new Token[MatrixSize]; // Allocate an array of tokens.
  67. if(Matrix == NULL) // Check for an allocation error.
  68. throw BadAllocation("TokenMatrix::Load() Matrix == NULL)"); // and throw an exception if it happens.
  69. F.read( // Now read the file into the allocated
  70. reinterpret_cast<char*>(Matrix), // matrix by recasting it as a character
  71. (MatrixSize * sizeof(Token))); // buffer of the correct size.
  72. if(F.bad()) // If there were any problems reading the
  73. throw BadMatrix("TokenMatrix::Load() (F.bad())"); // matrix then report the bad matrix.
  74. }
  75. // TokenMatrix::Validate(key)
  76. void TokenMatrix::Validate(string& SecurityKey) { // Decrypts and validates the matrix.
  77. MANGLER ValidationChecker; // Create a mangler engine for validation.
  78. // In order to do the validation we must look at the token matrix as a sequence of bytes.
  79. // We will be decrypting the first and last SecurtySegmentSize of this sequence and then
  80. // detecting wether the appropriate security key has been properly encrypted in the end.
  81. // If we find everything as it should be then we can be sure that the two segments have
  82. // not been tampered with and that we have the correct security key.
  83. unsigned char* TokensAsBytes = reinterpret_cast<unsigned char*>(Matrix);
  84. int BytesInTokenMatrix = (MatrixSize * sizeof(Token));
  85. // Now that we have all of that stuff let's initialize our ValidationChecker.
  86. // Note that the length of our security key is always 24 bytes. The license
  87. // id is 8 bytes, the authentication code is 16 bytes. We don't bother to check
  88. // here because if it's wrong then nothing will decrypt and we'll have essentially
  89. // the same result. Note also that on the end of the rule file we pad this
  90. // encrypted security id with nulls so that we can create a string from it easily
  91. // and so that we have precisely 32 bytes which is the same size as 4 tokens.
  92. //
  93. // Note: The 32 byte value is in SecurityKeyBufferSize. This means that we can
  94. // accept security keys up to 31 bytes in length. We need the ending null to
  95. // assure our null terminated string is as expected. The security key block must
  96. // match up with the edges of tokens in the matrix so we pad the end with nulls
  97. // when encoding the security key in the encoded file.
  98. int SecurityKeyLength = SecurityKey.length(); // For the length of our key
  99. for(int a=0;a<SecurityKeyLength;a++) // feed each byte through the
  100. ValidationChecker.Encrypt(SecurityKey.at(a)); // mangler to evolve the key
  101. // state.
  102. // Now we're ready to decrypt the matrix... We start with the first segment.
  103. for(int a=0;a<SecuritySegmentSize;a++) // For the length of the segment
  104. TokensAsBytes[a] = // replace each byte with the
  105. ValidationChecker.Decrypt(TokensAsBytes[a]); // decrypted byte.
  106. // Next we decrypt the last security segment...
  107. for(int a= BytesInTokenMatrix - SecuritySegmentSize; a<BytesInTokenMatrix; a++)
  108. TokensAsBytes[a] =
  109. ValidationChecker.Decrypt(TokensAsBytes[a]);
  110. // Now that we've done this we should find that our SecurityKey is at the end
  111. // of the loaded token matrix... Let's look and find out shall we?!!!
  112. unsigned char* SecurityCheckKey = // Reference the check
  113. & TokensAsBytes[BytesInTokenMatrix-SecurityKeyBufferSize]; // space in the matrix.
  114. SecurityCheckKey[SecurityKeyBufferSize-1] = 0; // Add a safety null just in case.
  115. string SecurityCheck((char*)SecurityCheckKey); // Make a string.
  116. // By now we should have a SecurityCheck string to compare to our SecurityKey.
  117. // If they match then we know everything worked out and that our token matrix has
  118. // been decrypted properly. This is also a good indication that our token matrix
  119. // is not incomplete since if it were the decryption wouldn't work. Saddly, we
  120. // don't have the computing cycles to decrypt the entire file - so we won't be
  121. // doing that until we can load it in a server/daemon and then reuse it over and
  122. // over... Once that happens we will be able to detect tampering also.
  123. if(SecurityKey != SecurityCheck) // If the security keys don't match
  124. throw BadMatrix("TokenMatrix::Validate() (SecurityKey != SecurityCheck)"); // then we have an invalid matrix.
  125. }
  126. // TokenMatrix::Verify(key)
  127. void TokenMatrix::Verify(string& SecurityKey) { // Builds and verifies a file digest.
  128. MANGLER DigestChecker; // Create a mangler for the digest.
  129. // Gain access to our token matrix as bytes.
  130. unsigned char* TokensAsBytes = reinterpret_cast<unsigned char*>(Matrix);
  131. int BytesInTokenMatrix = (MatrixSize * sizeof(Token));
  132. // Initialize our digest engine with the security key.
  133. int SecurityKeyLength = SecurityKey.length(); // For the length of our key
  134. for(int a=0;a<SecurityKeyLength;a++) // feed each byte through the
  135. DigestChecker.Encrypt(SecurityKey.at(a)); // mangler to evolve the key
  136. // state.
  137. // Build the digest.
  138. int IndexOfDigest = // Find the index of the digest by
  139. BytesInTokenMatrix - // starting at the end of the matrix,
  140. SecurityKeyBufferSize - // backing up past the security key,
  141. RulebaseDigestSize; // then past the digest.
  142. int a=0; // Keep track of where we are.
  143. for(;a<IndexOfDigest;a++) // Loop through up to the digest and
  144. DigestChecker.Encrypt(TokensAsBytes[a]); // pump the file through the mangler.
  145. // Now that the digest is built we must test it.
  146. // The original was emitted by encrypting 0s so if we do the same thing we will match.
  147. for(int b=0;b<RulebaseDigestSize;b++) // Loop through the digest and compare
  148. if(DigestChecker.Encrypt(0)!=TokensAsBytes[a+b]) // our digest to the stored digest. If
  149. throw BadMatrix("TokenMatrix::Verify() Bad Digest"); // any byte doesn't match it's bad!
  150. // If we made it through all of that then we're valid :-)
  151. }
  152. void TokenMatrix::FlipEndian() { // Converts big/little endian tokens.
  153. unsigned int* UInts = reinterpret_cast<unsigned int*>(Matrix); // Grab the matrix as uints.
  154. int Length = ((MatrixSize * sizeof(Token)) / sizeof(unsigned int)); // Calculate it's size.
  155. for(int i = 0; i < Length; i++) { // Loop through the array of u ints
  156. unsigned int x = UInts[i]; // and re-order the bytes in each
  157. x = ((x & 0xff000000) >> 24) | // one to swap from big/little endian
  158. ((x & 0x00ff0000) >> 8) | // to little/big endian.
  159. ((x & 0x0000ff00) << 8) |
  160. ((x & 0x000000ff) << 24);
  161. UInts[i] = x; // Put the flipped int back.
  162. }
  163. }
  164. // Evaluator Implementations //////////////////////////////////////////////////////////////
  165. // 20030216 _M Optimization conversions
  166. // 20140119 _M Deprecated by jump table in evaluator
  167. // inline int Evaluator::i_lower() { return myEvaluationMatrix->i_lower; }
  168. // inline bool Evaluator::i_isDigit() { return myEvaluationMatrix->i_isDigit; }
  169. // inline bool Evaluator::i_isSpace() { return myEvaluationMatrix->i_isSpace; }
  170. // inline bool Evaluator::i_isAlpha() { return myEvaluationMatrix->i_isAlpha; }
  171. // Evaluator::Evaluator(position,evalmatrix) Constructor
  172. Evaluator::Evaluator(unsigned int s, EvaluationMatrix* m)
  173. : myEvaluationMatrix(m),
  174. JumpPoint(0),
  175. Condition(DOING_OK),
  176. NextEvaluator(NULL),
  177. StreamStartPosition(s),
  178. CurrentPosition(0),
  179. WildRunLength(0) { // Constructor...
  180. Matrix = myEvaluationMatrix->getTokens(); // Capture the token matrix I walk in.
  181. MatrixSize = myEvaluationMatrix->getMatrixSize(); // And get it's size.
  182. PositionLimit = MatrixSize - 256;
  183. }
  184. // Of course I may need to resolve some of the following
  185. // wildcard characters.
  186. int Evaluator::xLetter() { return (JumpPoint + WILD_LETTER); } // Match Any letter.
  187. int Evaluator::xDigit() { return (JumpPoint + WILD_DIGIT); } // Match Any digit.
  188. int Evaluator::xNonWhite() { return (JumpPoint + WILD_NONWHITE); } // Match Any non-whitespace.
  189. int Evaluator::xWhiteSpace() { return (JumpPoint + WILD_WHITESPACE); } // Match Any whitespace.
  190. int Evaluator::xAnyInline() { return (JumpPoint + WILD_INLINE); } // Match Any byte but new line.
  191. int Evaluator::xAnything() { return (JumpPoint + WILD_ANYTHING); } // Match Any character at all.
  192. int Evaluator::xRunGateway() { return (JumpPoint + RUN_GATEWAY); } // Match the run-loop gateway.
  193. // void Evaluator::doFollowOrMakeBuddy()
  194. void Evaluator::doFollowOrMakeBuddy(int xKey) {
  195. bool shouldFollow = (FALLEN_OFF == Condition); // What should we do?
  196. if(shouldFollow) { // This is how we follow
  197. Condition = DOING_OK;
  198. CurrentPosition = xKey +
  199. Matrix[xKey].Vector;
  200. }
  201. else { // This is how we make a buddy
  202. myEvaluationMatrix->
  203. AddEvaluator(StreamStartPosition,Matrix[xKey].Vector+xKey);
  204. }
  205. }
  206. void Evaluator::tryFollowingPrecisePath(unsigned short int i) {
  207. int xPrecise = JumpPoint + i; // Match Precise Character
  208. if(Matrix[xPrecise].Character() == i) { // If we've matched our path
  209. doFollowOrMakeBuddy(xPrecise);
  210. }
  211. if(DOING_OK == Condition) WildRunLength = 0;
  212. }
  213. void Evaluator::tryFollowingNoCasePath(unsigned short int i) {
  214. i = tolower(i);
  215. int xNoCase = JumpPoint + i; // Match caps to lower (case insensitive)
  216. if(Matrix[xNoCase].Character()==i){
  217. doFollowOrMakeBuddy(xNoCase);
  218. }
  219. if(DOING_OK == Condition) WildRunLength = 0;
  220. }
  221. void Evaluator::tryFollowingWildAlphaPath() {
  222. if(Matrix[xLetter()].Character()==WILD_LETTER){
  223. doFollowOrMakeBuddy(xLetter());
  224. }
  225. }
  226. void Evaluator::tryFollowingWildDigitPath() {
  227. if(Matrix[xDigit()].Character()==WILD_DIGIT){
  228. doFollowOrMakeBuddy(xDigit());
  229. }
  230. }
  231. void Evaluator::tryFollowingWildNonWhitePath() {
  232. if(Matrix[xNonWhite()].Character()==WILD_NONWHITE){
  233. doFollowOrMakeBuddy(xNonWhite());
  234. }
  235. }
  236. void Evaluator::tryFollowingWildWhitePath() {
  237. if(Matrix[xWhiteSpace()].Character()==WILD_WHITESPACE){
  238. doFollowOrMakeBuddy(xWhiteSpace());
  239. }
  240. }
  241. void Evaluator::tryFollowingWildInlinePath() {
  242. if(Matrix[xAnyInline()].Character()==WILD_INLINE){
  243. doFollowOrMakeBuddy(xAnyInline());
  244. }
  245. }
  246. void Evaluator::tryFollowingWildAnythingPath() {
  247. if(Matrix[xAnything()].Character()==WILD_ANYTHING){
  248. doFollowOrMakeBuddy(xAnything());
  249. }
  250. }
  251. void Evaluator::doFollowerJumpTable(unsigned short int i) {
  252. tryFollowingPrecisePath(i);
  253. // tryFollowingUppercasePath(); 0x41 - 0x5A
  254. // tryFollowingWildAlphaPath(); 0x61 - 0x7A
  255. // tryFollowingWildDigitPath(); 0x30 - 0x39
  256. // tryFollowingWildWhitePath(); 0x09 - 0x0D, 0x20
  257. // tryFollowingWildNonWhitePath(); > 0x20
  258. // tryFollowingWildInlinePath(); Not 0x0A, or 0x0D
  259. switch(i) {
  260. // These only match Precise, or WildAnything ...
  261. // NUL, SOH, STX, ETX, EOT, ENQ, ACK, BEL, BS, TAB, LF, VT, FF, CR, SO, SI
  262. case 0x00: case 0x01: case 0x02: case 0x03: case 0x04: case 0x05: case 0x06: case 0x07:
  263. case 0x08: {
  264. break;
  265. }
  266. // tab
  267. case 0x09: {
  268. tryFollowingWildWhitePath();
  269. tryFollowingWildInlinePath();
  270. break;
  271. }
  272. // LF, VT, FF, CR, SO, SI
  273. case 0x0A: case 0x0B: case 0x0C: case 0x0D: case 0x0E: case 0x0F:
  274. // DLE, DC1, DC2, DC3, DC4, NAK, SYN, ETB, CAN, EM, SUB, ESC, FS, GS, RS, US
  275. case 0x10: case 0x11: case 0x12: case 0x13: case 0x14: case 0x15: case 0x16: case 0x17:
  276. case 0x18: case 0x19: case 0x1A: case 0x1B: case 0x1C: case 0x1D: case 0x1E: case 0x1F: {
  277. tryFollowingWildWhitePath();
  278. break;
  279. }
  280. // the final fronteer
  281. case 0x20: {
  282. tryFollowingWildWhitePath();
  283. tryFollowingWildInlinePath();
  284. break;
  285. }
  286. // ! " # $ % & ' ( ) * + , - . /
  287. case 0x21: case 0x22: case 0x23: case 0x24: case 0x25: case 0x26: case 0x27:
  288. case 0x28: case 0x29: case 0x2A: case 0x2B: case 0x2C: case 0x2D: case 0x2E: case 0x2F: {
  289. tryFollowingWildNonWhitePath();
  290. tryFollowingWildInlinePath();
  291. break;
  292. }
  293. // 0 - 9
  294. case 0x30: case 0x31: case 0x32: case 0x33: case 0x34: case 0x35: case 0x36: case 0x37:
  295. case 0x38: case 0x39: {
  296. tryFollowingWildDigitPath();
  297. tryFollowingWildNonWhitePath();
  298. tryFollowingWildInlinePath();
  299. break;
  300. }
  301. // : ; < = > ? @
  302. case 0x3A: case 0x3B: case 0x3C: case 0x3D: case 0x3E: case 0x3F:
  303. case 0x40: {
  304. tryFollowingWildNonWhitePath();
  305. tryFollowingWildInlinePath();
  306. break;
  307. }
  308. // A - Z
  309. case 0x41: case 0x42: case 0x43: case 0x44: case 0x45: case 0x46: case 0x47:
  310. case 0x48: case 0x49: case 0x4A: case 0x4B: case 0x4C: case 0x4D: case 0x4E: case 0x4F:
  311. case 0x50: case 0x51: case 0x52: case 0x53: case 0x54: case 0x55: case 0x56: case 0x57:
  312. case 0x58: case 0x59: case 0x5A: {
  313. tryFollowingNoCasePath(i);
  314. tryFollowingWildAlphaPath();
  315. tryFollowingWildNonWhitePath();
  316. tryFollowingWildInlinePath();
  317. break;
  318. }
  319. // [ \ ] ^ _ `
  320. case 0x5B: case 0x5C: case 0x5D: case 0x5E: case 0x5F:
  321. case 0x60: {
  322. tryFollowingWildNonWhitePath();
  323. tryFollowingWildInlinePath();
  324. break;
  325. }
  326. // a - z
  327. case 0x61: case 0x62: case 0x63: case 0x64: case 0x65: case 0x66: case 0x67:
  328. case 0x68: case 0x69: case 0x6A: case 0x6B: case 0x6C: case 0x6D: case 0x6E: case 0x6F:
  329. case 0x70: case 0x71: case 0x72: case 0x73: case 0x74: case 0x75: case 0x76: case 0x77:
  330. case 0x78: case 0x79: case 0x7A: {
  331. tryFollowingWildAlphaPath();
  332. tryFollowingWildNonWhitePath();
  333. tryFollowingWildInlinePath();
  334. break;
  335. }
  336. // { | } ~
  337. case 0x7B: case 0x7C: case 0x7D: case 0x7E: case 0x7F: {
  338. tryFollowingWildNonWhitePath();
  339. tryFollowingWildInlinePath();
  340. }
  341. // high ascii
  342. case 0x80: case 0x81: case 0x82: case 0x83: case 0x84: case 0x85: case 0x86: case 0x87:
  343. case 0x88: case 0x89: case 0x8A: case 0x8B: case 0x8C: case 0x8D: case 0x8E: case 0x8F:
  344. case 0x90: case 0x91: case 0x92: case 0x93: case 0x94: case 0x95: case 0x96: case 0x97:
  345. case 0x98: case 0x99: case 0x9A: case 0x9B: case 0x9C: case 0x9D: case 0x9E: case 0x9F:
  346. case 0xA0: case 0xA1: case 0xA2: case 0xA3: case 0xA4: case 0xA5: case 0xA6: case 0xA7:
  347. case 0xA8: case 0xA9: case 0xAA: case 0xAB: case 0xAC: case 0xAD: case 0xAE: case 0xAF:
  348. case 0xB0: case 0xB1: case 0xB2: case 0xB3: case 0xB4: case 0xB5: case 0xB6: case 0xB7:
  349. case 0xB8: case 0xB9: case 0xBA: case 0xBB: case 0xBC: case 0xBD: case 0xBE: case 0xBF:
  350. case 0xC0: case 0xC1: case 0xC2: case 0xC3: case 0xC4: case 0xC5: case 0xC6: case 0xC7:
  351. case 0xC8: case 0xC9: case 0xCA: case 0xCB: case 0xCC: case 0xCD: case 0xCE: case 0xCF:
  352. case 0xD0: case 0xD1: case 0xD2: case 0xD3: case 0xD4: case 0xD5: case 0xD6: case 0xD7:
  353. case 0xD8: case 0xD9: case 0xDA: case 0xDB: case 0xDC: case 0xDD: case 0xDE: case 0xDF:
  354. case 0xE0: case 0xE1: case 0xE2: case 0xE3: case 0xE4: case 0xE5: case 0xE6: case 0xE7:
  355. case 0xE8: case 0xE9: case 0xEA: case 0xEB: case 0xEC: case 0xED: case 0xEE: case 0xEF:
  356. case 0xF0: case 0xF1: case 0xF2: case 0xF3: case 0xF4: case 0xF5: case 0xF6: case 0xF7:
  357. case 0xF8: case 0xF9: case 0xFA: case 0xFB: case 0xFC: case 0xFD: case 0xFE: case 0xFF: {
  358. tryFollowingWildNonWhitePath();
  359. tryFollowingWildInlinePath();
  360. break;
  361. }
  362. }
  363. tryFollowingWildAnythingPath();
  364. }
  365. // Evaluator::EvaluateThis()
  366. Evaluator::States Evaluator::EvaluateThis(unsigned short int i) { // Follow the this byte.
  367. // First upgrade will be to DOING_OK, after that we launch buddies.
  368. Condition = FALLEN_OFF; // Start off guessing we'll fall off.
  369. // In order to handle wildcard characters, this evaluation function must actually
  370. // compare the character to a number of possibilities in most-specific to least-
  371. // specific order to see if any match. In order to support overlapping rule sets,
  372. // if more than one wildcard matches at this node, an additional evaluator will be
  373. // placed in line already _AT THIS PATH POINT_ so that both possibilities will be
  374. // explored. New evaluators are always added at the TOP of the list so we are always
  375. // guaranteed not to overdrive an evaluator and end up in a recursive race condition.
  376. // 20140121_M The previous optimization with binary flags has been replaced with
  377. // a jump table implementation. Now, each byte only excites behaviors that are
  378. // possible for the current byte so only those paths will be tested.
  379. if(CurrentPosition >= PositionLimit) return Condition = OUT_OF_RANGE;
  380. // All of the positions calculated below are guaranteed to be within the ranges checked
  381. // above so we're safe if we get to this point.
  382. // So, at this point it's safe to check and see if I'm terminated. Note that if I
  383. // am at a termination point, my path has terminated and I have a symbol so I don't
  384. // need to resolve any more characters - even the current one.
  385. if(Matrix[CurrentPosition].isTermination()) return Condition = TERMINATED;
  386. // NOTE: The above is written for sudden-death termination. Eventually we will want
  387. // to support deep - filters which will show every rule match and this will need to
  388. // be rewritten.
  389. // Evaluation order, most-to-least specific with what is possible for that byte.
  390. JumpPoint = CurrentPosition;
  391. doFollowerJumpTable(i); // Excite followers based on this byte.
  392. { // Precise matches reset the wild run counter.
  393. ++WildRunLength; // Count up the run length.
  394. if(WildRunLength >= MaxWildRunLength) // If we exceed the max then
  395. return Condition = FALLEN_OFF; // we've fallen off the path
  396. } // and we do it immediately.
  397. // 20021112 _M
  398. // Beginning with version 2 of Message Sniffer we've implemented a new construct
  399. // for run-loops that prevents any interference between rules where run-loops might
  400. // appear in locations coinciding with standard match bytes. The new methodology
  401. // uses a special run-loop-gateway character to isolate any run loops from standard
  402. // nodes in the matrix. Whenever a run-loop gateway is present at a node a buddy is
  403. // inserted AFTER the current evaluator so that it will evaluate the current character
  404. // from the position of the run-loop gateway. This allows run loops to occupy the same
  405. // positional space as standard matches while maintaining isolation between their paths
  406. // in the matrix.
  407. // We don't want to launch any run loop buddies unless we matched this far. If we did
  408. // match up to this point and the next character in a pattern includes a run loop then
  409. // we will find a gateway byte at this point representing the path to any run loops.
  410. // If we made it this far launch a buddy for any run-loop gateway that's present.
  411. // Of course, the buddy must be evaluated after this evaluator during this pass because
  412. // he will have shown up late... That is, we don't detect a run gateway until we're
  413. // sitting on a new node looking for a result... The very result we may be looking for
  414. // could be behind the gateway - so we launch the buddy behind us and he will be able
  415. // to match anything in this pass that we missed when looking for a non-run match.
  416. if(Matrix[xRunGateway()].Character() == RUN_GATEWAY)
  417. myEvaluationMatrix->
  418. InsEvaluator(StreamStartPosition,Matrix[xRunGateway()].Vector+xRunGateway());
  419. // At this point, we've tried all of our rules, and created any buddies we needed.
  420. // If we got a match, we terminated long ago. If we didn't, then we either stayed
  421. // on the path or we fell off. Either way, the flag is in Condition so we can send
  422. // it on.
  423. return Condition;
  424. }
  425. ///////////////////////////////////////////////////////////////////////////////////////////
  426. // EvaluationMatrix Implementations ///////////////////////////////////////////////////////
  427. // EvaluationMatrix::AddMatchRecord(int sp, int ep, int sym)
  428. // Most of this functionality is about deep scans - which have been put on hold for now
  429. // due to the complexity and the scope of the current application. For now, although
  430. // we will use this reporting mechanism, it will generally record only one event.
  431. MatchRecord* EvaluationMatrix::AddMatchRecord(int sp, int ep, int sym) {
  432. // 20030216 _M Added range check code to watch for corruption. Some systems have
  433. // reported matches with zero length indicating an undetected corruption. This
  434. // range check will detect and report it.
  435. if(sp==ep) // Check that we're in range - no zero
  436. throw OutOfRange("sp==ep"); // length pattern matches allowed!
  437. MatchRecord* NewMatchRecord = // Then, create the new result object
  438. new MatchRecord(sp,ep,sym); // by passing it the important parts.
  439. if(NewMatchRecord==NULL) // Check for a bad allocation and throw
  440. throw BadAllocation("NewMatchRecord==NULL"); // an exception if that happens.
  441. if(ResultList == NULL) { // If this is our first result we simply
  442. ResultList = NewMatchRecord; // add the result to our list, and of course
  443. LastResultInList = NewMatchRecord; // it is the end of the list as well.
  444. } else { // If we already have some results, then
  445. LastResultInList->NextMatchRecord = // we add the new record to the result list
  446. NewMatchRecord; // and record that the new record is now the
  447. LastResultInList = NewMatchRecord; // last result in the list.
  448. }
  449. return NewMatchRecord; // Return our new match record.
  450. }
  451. // EvaluationMatrix::AddEvaluator()
  452. // 20021112 _M
  453. // This function has be modified to include a check for duplicates as well as setting
  454. // the mount point for the new evaluator. This eliminates a good deal of code elsewhere
  455. // and encapsulates the complete operation. If a duplicate evaluator is found then the
  456. // function returns NULL indicating that nothing was done. In practic, no check is made
  457. // since any serious error conditions cause errors to be thrown from within this function
  458. // call. These notes apply to some extent to InsEvaluator which is copied from this function
  459. // and which has the only difference of putting the new evaluator after the current one
  460. // in the chain in order to support branch-out operations for loop sequences in the matrix.
  461. Evaluator* EvaluationMatrix::AddEvaluator(int s, unsigned int m) { // Adds a new evaluator at top.
  462. if(!isNoDuplicate(m)) return NULL; // If there is a duplicate do nothing.
  463. if(CountOfEvaluators >= MAX_EVALS) // If we've exceeded our population size
  464. throw MaxEvalsExceeded("Add:CountOfEvaluators >= MAX_EVALS"); // then throw an exception.
  465. Evaluator* NewEvaluator = SourceEvaluator(s,this); // Make up a new evaluator.
  466. if(NewEvaluator == NULL) // Check for a bad allocation and throw
  467. throw BadAllocation("Add:NewEvaluator == NULL"); // an exception if it happens.
  468. NewEvaluator->NextEvaluator = EvaluatorList; // Point the new evaluator to the list.
  469. EvaluatorList = NewEvaluator; // Then point the list head to
  470. // the new evaluator.
  471. NewEvaluator->CurrentPosition = m; // Esablish the mount point.
  472. ++CountOfEvaluators; // Add one to our evaluator count.
  473. if(CountOfEvaluators > MaximumCountOfEvaluators) // If the count is the biggest we
  474. MaximumCountOfEvaluators = CountOfEvaluators; // have seen then keep track of it.
  475. return NewEvaluator; // Return the new evaluator.
  476. }
  477. // EvaluationMatrix::InsEvaluator()
  478. Evaluator* EvaluationMatrix::InsEvaluator(int s, unsigned int m) { // Inserts a new evaluator.
  479. if(!isNoDuplicate(m)) return NULL; // If there is a duplicate do nothing.
  480. if(CountOfEvaluators >= MAX_EVALS) // If we've exceeded our population size
  481. throw MaxEvalsExceeded("Ins:CountOfEvaluators >= MAX_EVALS"); // then throw an exception.
  482. Evaluator* NewEvaluator = SourceEvaluator(s,this); // Make up a new evaluator.
  483. if(NewEvaluator == NULL) // Check for a bad allocation and throw
  484. throw BadAllocation("Ins:NewEvaluator == NULL"); // an exception if it happens.
  485. NewEvaluator->NextEvaluator = // Point the new evaluator where the
  486. CurrentEvaluator->NextEvaluator; // current evalautor points... then point
  487. CurrentEvaluator->NextEvaluator = // the current evaluator to this one. This
  488. NewEvaluator; // accomplishes the insert operation.
  489. NewEvaluator->CurrentPosition = m; // Esablish the mount point.
  490. ++CountOfEvaluators; // Add one to our evaluator count.
  491. if(CountOfEvaluators > MaximumCountOfEvaluators) // If the count is the biggest we
  492. MaximumCountOfEvaluators = CountOfEvaluators; // have seen then keep track of it.
  493. return NewEvaluator; // Return the new evaluator.
  494. }
  495. // EvaluationMatrix::DropEvaluator()
  496. void EvaluationMatrix::DropEvaluator() { // Drops the current evaluator from the matrix.
  497. Evaluator* WhereTo = CurrentEvaluator->NextEvaluator; // Where do we go from here?
  498. // First step is to heal the list as if the current evaluator were not present.
  499. // If there is no previous evaluator - meaning this should be the first one in the
  500. // list - then we point the list head to the next evaluator on the list (WhereTo)
  501. if(PreviousEvaluator != NULL) // If we have a Previous then
  502. PreviousEvaluator->NextEvaluator = WhereTo; // our next is it's next.
  503. else // If we don't then our next
  504. EvaluatorList = WhereTo; // is the first in the list.
  505. // Now that our list is properly healed, it's time to drop the dead evaluator and
  506. // get on with our lives...
  507. CurrentEvaluator->NextEvaluator = NULL; // Disconnect from any list.
  508. CacheEvaluator(CurrentEvaluator); // Drop the current eval.
  509. CurrentEvaluator = WhereTo; // Move on.
  510. --CountOfEvaluators; // Reduce our evaluator count.
  511. }
  512. Evaluator* findEvaluatorListTail(Evaluator* head) {
  513. Evaluator* next = head;
  514. while(NULL != (next->NextEvaluator)) next = next->NextEvaluator;
  515. return next;
  516. }
  517. void EvaluationMatrix::dropAllEvaluators() {
  518. bool haveActiveEvaluators = (NULL != EvaluatorList);
  519. if(haveActiveEvaluators) {
  520. Evaluator* tail = findEvaluatorListTail(EvaluatorList);
  521. tail->NextEvaluator = EvaluatorCache;
  522. EvaluatorCache = EvaluatorList;
  523. }
  524. PreviousEvaluator = NULL;
  525. CurrentEvaluator = NULL;
  526. EvaluatorList = NULL;
  527. CountOfEvaluators = 0;
  528. }
  529. void EvaluationMatrix::restartEngineAt(int newCharacterCount) {
  530. dropAllEvaluators();
  531. CountOfCharacters = newCharacterCount;
  532. }
  533. // EvaluationMatrix::EvaluateThis()
  534. //
  535. // This function returns the number of matches that were found. It is possible for more
  536. // than one evaluator to match on a single character.
  537. //
  538. // 0 indicates no matches were found.
  539. // >0 indicates some matches were found.
  540. // If there is a problem then an exception will be thrown.
  541. int EvaluationMatrix::EvaluateThis(unsigned short int i) {
  542. AddEvaluator(CountOfCharacters,0); // First, add a new Evaluator at the root of the
  543. // matrix for the current position in the scan
  544. // stream.
  545. // The new evaluator is now at the top of our list.
  546. // If there was a problem then an exception will have been thrown.
  547. // If our allocation worked ok, then we'll be here and ready to start scanning
  548. // the rule set with our current character.
  549. PassResult = 0; // Start by assuming we won't match.
  550. CurrentEvaluator = EvaluatorList; // Start at the top of the list.
  551. PreviousEvaluator = NULL; // NULL means previous is the top.
  552. // 20030216 _M
  553. // Next do some basic conversions and evaluations so they don't need to be done
  554. // again within the evaluators. From now on the evaluators will look here for basic
  555. // conversions and boolean check values rather than performing the checks themselves.
  556. // 20140119 _M deprecated by jump table in evaluator
  557. // i_lower = tolower(i); // Convert i to lower case.
  558. // i_isDigit = isdigit(i); // Check for a digit.
  559. // i_isSpace = isspace(i); // Check for whitespace.
  560. // i_isAlpha = isalpha(i); // Check for letters.
  561. // Next, loop through the list and pass the incoming character to
  562. // each evaluator. Drop those that fall off, and record those that terminate. The
  563. // rest of them stick around to walk their paths until they meet their fate.
  564. while(CurrentEvaluator != NULL) { // While there are more evaluators...
  565. // go through the list and evaluate
  566. switch(CurrentEvaluator->EvaluateThis(i)) { // the current character against each.
  567. case Evaluator::FALLEN_OFF: { // If we've fallen off the path
  568. DropEvaluator(); // drop the current evaluator and
  569. break; // move on with our lives.
  570. }
  571. case Evaluator::DOING_OK: { // If we're still going then...
  572. PreviousEvaluator = CurrentEvaluator; // keep track of where we've been and
  573. CurrentEvaluator = // move forward to the next evaluator
  574. CurrentEvaluator->NextEvaluator; // in the list.
  575. break;
  576. }
  577. case Evaluator::TERMINATED: { // If we've terminated a path...
  578. ++PassResult; // Record our PassResult.
  579. // Create a new match result using the data in the current evaluator.
  580. // If there is a problem adding the match an exception will be thrown.
  581. AddMatchRecord(
  582. CurrentEvaluator->StreamStartPosition,
  583. CountOfCharacters - 1,
  584. myTokenMatrix->Symbol(CurrentEvaluator->CurrentPosition)
  585. );
  586. // From Version 2 onward we're always doing deep scans...
  587. // Having successfully recorded the result of this critter we can kill them off.
  588. DropEvaluator(); // He's dead.
  589. break; // Now let's keep looking.
  590. }
  591. case Evaluator::OUT_OF_RANGE: { // This result is really bad and
  592. throw OutOfRange("case Evaluator::OUT_OF_RANGE:"); // probably means we have a bad matrix.
  593. break;
  594. // The reason we don't throw OutOfRange from within the evaluator is that we
  595. // may want to take some other action in the future... So, we allow the evaluator
  596. // to tell us we sent it out of range and then we decide what to do about it.
  597. }
  598. }
  599. }
  600. // At the end of this function our PassResult is either an error (which is
  601. // reported immediately), or it is a match condition. We start out by assuming
  602. // there will be no match. If we find one, then we reset that result... so at
  603. // this point, all we need do is report our findings.
  604. ++CountOfCharacters; // Add one to our Character Count statistic.
  605. // Note that from this point on, the index in the stream is one less than the
  606. // CountOfCharacters... for example, if I've evaluated (am evaluating) one character
  607. // the it's index is 0. This will be important when we create any match records.
  608. return PassResult; // When we're finished, return the last known result.
  609. }
  610. void EvaluationMatrix::evaluateSegment(vector<unsigned char>& data, unsigned int start, unsigned int finish) {
  611. restartEngineAt(start);
  612. finish = (finish < data.size()) ? finish : data.size();
  613. for(unsigned int a = start; a < finish; a++) EvaluateThis(data[a]);
  614. }
  615. }