1 /*
   2  * Copyright (c) 2016, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.
   8  *
   9  * This code is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * version 2 for more details (a copy is included in the LICENSE file that
  13  * accompanied this code).
  14  *
  15  * You should have received a copy of the GNU General Public License version
  16  * 2 along with this work; if not, write to the Free Software Foundation,
  17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  20  * or visit www.oracle.com if you need additional information or have any
  21  * questions.
  22  */
  23 
  24 import java.util.ArrayList;
  25 import java.util.List;
  26 import java.util.regex.Matcher;
  27 import java.util.regex.Pattern;
  28 
  29 /**
  30  * Matches the krb5 debug output:
  31  * >>> KDCCommunication: kdc=host UDP:11555, timeout=100,Attempt =1, #bytes=138
  32  *
  33  * Example:
  34  *   CommMatcher cm = new CommMatcher();
  35  *   cm.addPort(12345).addPort(23456);
  36  *   for (String line : debugOutput) {
  37  *       if (cm.match(line)) {
  38  *           println("KDC: %c, %s, Timeout: %d\n",
  39  *              cm.kdc(), cm.protocol(), cm.timeout());
  40  *       }
  41  *   }
  42  */
  43 public class CommMatcher {
  44 
  45     static final Pattern re = Pattern.compile(
  46             ">>> KDCCommunication: kdc=\\S+ (TCP|UDP):(\\d+), " +
  47                     "timeout=(\\d+),Attempt\\s*=(\\d+)");
  48 
  49     List<Integer> kdcPorts = new ArrayList<>();
  50     Matcher matcher;
  51 
  52     /**
  53      * Add KDC ports one by one. The 1st KDC will be 'a' in {@link #kdc()},
  54      * 2nd is 'b', etc, etc.
  55      */
  56     public CommMatcher addPort(int port) {
  57         if (port > 0) {
  58             kdcPorts.add(port);
  59         } else {
  60             kdcPorts.clear();
  61         }
  62         return this;
  63     }
  64 
  65     public boolean match(String line) {
  66         matcher = re.matcher(line);
  67         return matcher.find();
  68     }
  69 
  70     public String protocol() {
  71         return matcher.group(1);
  72     }
  73 
  74     public char kdc() {
  75         int port = Integer.parseInt(matcher.group(2));
  76         return (char)(kdcPorts.indexOf(port) + 'a');
  77     }
  78 
  79     public int timeout() {
  80         return BadKdc.toSymbolicSec(Integer.parseInt(matcher.group(3)));
  81     }
  82 
  83     public int attempt() {
  84         return Integer.parseInt(matcher.group(4));
  85     }
  86 }