Coverage for src / tracekit / inference / __init__.py: 100%
16 statements
« prev ^ index » next coverage.py v7.13.1, created at 2026-01-11 23:04 +0000
1"""Auto-inference and smart defaults for TraceKit.
3This module provides automatic parameter detection and intelligent defaults
4for common analysis tasks.
6- RE-SEQ-002: Sequence Pattern Detection
7- RE-SEQ-003: Request-Response Correlation
8- RE-STR-001: UDP Stream Reconstruction
9- RE-STR-002: TCP Stream Reassembly
10- RE-STR-003: Message Framing and Segmentation
11- RE-BIN-001: Magic Byte Detection
12- RE-BIN-002: Structure Alignment Detection
13- RE-BIN-003: Binary Parser DSL
14- RE-DSL-003: Protocol Format Library
15"""
17from tracekit.inference.adaptive_tuning import (
18 AdaptiveParameterTuner,
19 TunedParameters,
20 get_adaptive_parameters,
21)
22from tracekit.inference.alignment import (
23 AlignmentResult,
24 align_global,
25 align_local,
26 align_multiple,
27 compute_similarity,
28 find_conserved_regions,
29 find_variable_regions,
30)
31from tracekit.inference.bayesian import (
32 BayesianInference,
33 Posterior,
34 Prior,
35 SequentialBayesian,
36 infer_with_uncertainty,
37)
39# RE-BIN-001, RE-BIN-002, RE-BIN-003: Binary Format Inference
40from tracekit.inference.binary import (
41 KNOWN_MAGIC_BYTES,
42 AlignmentDetector,
43 BinaryParserGenerator,
44 MagicByteDetector,
45 MagicByteResult,
46 ParserDefinition,
47 ParserField,
48 detect_alignment,
49 detect_magic_bytes,
50 find_all_magic_bytes,
51 generate_parser,
52 parser_to_python,
53 parser_to_yaml,
54)
55from tracekit.inference.binary import (
56 AlignmentResult as BinaryAlignmentResult,
57)
58from tracekit.inference.logic import detect_logic_family
59from tracekit.inference.message_format import (
60 InferredField,
61 MessageFormatInferrer,
62 MessageSchema,
63 detect_field_types,
64 find_dependencies,
65 infer_format,
66)
67from tracekit.inference.protocol import detect_protocol
68from tracekit.inference.protocol_dsl import (
69 DecodedMessage,
70 FieldDefinition,
71 ProtocolDecoder,
72 ProtocolDefinition,
73 ProtocolEncoder,
74 decode_message,
75 load_protocol,
76)
78# RE-DSL-003: Protocol Format Library
79from tracekit.inference.protocol_library import (
80 ProtocolInfo,
81 ProtocolLibrary,
82 get_decoder,
83 get_library,
84 get_protocol,
85 list_protocols,
86)
88# RE-SEQ-002, RE-SEQ-003: Sequence Pattern Detection and Correlation
89from tracekit.inference.sequences import (
90 CommunicationFlow,
91 RequestResponseCorrelator,
92 RequestResponsePair,
93 SequencePattern,
94 SequencePatternDetector,
95 calculate_latency_stats,
96 correlate_requests,
97 detect_sequence_patterns,
98 find_message_dependencies,
99)
100from tracekit.inference.signal_intelligence import (
101 AnalysisRecommendation,
102 assess_signal_quality,
103 check_measurement_suitability,
104 classify_signal,
105 get_optimal_domain_order,
106 recommend_analyses,
107 suggest_measurements,
108)
109from tracekit.inference.spectral import auto_spectral_config
110from tracekit.inference.state_machine import (
111 FiniteAutomaton,
112 State,
113 StateMachineInferrer,
114 Transition,
115 infer_rpni,
116 minimize_dfa,
117 to_dot,
118 to_networkx,
119)
121# RE-STR-001, RE-STR-002, RE-STR-003: Stream Reassembly
122from tracekit.inference.stream import (
123 FramingResult,
124 MessageFrame,
125 MessageFramer,
126 ReassembledStream,
127 StreamSegment,
128 TCPStreamReassembler,
129 UDPStreamReassembler,
130 detect_message_framing,
131 extract_messages,
132 reassemble_tcp_stream,
133 reassemble_udp_stream,
134)
136__all__ = [
137 "KNOWN_MAGIC_BYTES",
138 "AdaptiveParameterTuner",
139 # RE-BIN-001, RE-BIN-002, RE-BIN-003: Binary Format Inference
140 "AlignmentDetector",
141 "AlignmentResult",
142 "AnalysisRecommendation",
143 "BayesianInference",
144 "BinaryAlignmentResult",
145 "BinaryParserGenerator",
146 # RE-SEQ-002, RE-SEQ-003: Sequence Patterns and Correlation
147 "CommunicationFlow",
148 "DecodedMessage",
149 "FieldDefinition",
150 "FiniteAutomaton",
151 # RE-STR-001, RE-STR-002, RE-STR-003: Stream Reassembly
152 "FramingResult",
153 "InferredField",
154 "MagicByteDetector",
155 "MagicByteResult",
156 "MessageFormatInferrer",
157 "MessageFrame",
158 "MessageFramer",
159 "MessageSchema",
160 "ParserDefinition",
161 "ParserField",
162 "Posterior",
163 "Prior",
164 "ProtocolDecoder",
165 "ProtocolDefinition",
166 "ProtocolEncoder",
167 # RE-DSL-003: Protocol Format Library
168 "ProtocolInfo",
169 "ProtocolLibrary",
170 "ReassembledStream",
171 "RequestResponseCorrelator",
172 "RequestResponsePair",
173 "SequencePattern",
174 "SequencePatternDetector",
175 "SequentialBayesian",
176 "State",
177 "StateMachineInferrer",
178 "StreamSegment",
179 "TCPStreamReassembler",
180 "Transition",
181 "TunedParameters",
182 "UDPStreamReassembler",
183 "align_global",
184 "align_local",
185 "align_multiple",
186 # Original exports
187 "assess_signal_quality",
188 "auto_spectral_config",
189 "calculate_latency_stats",
190 "check_measurement_suitability",
191 "classify_signal",
192 "compute_similarity",
193 "correlate_requests",
194 "decode_message",
195 "detect_alignment",
196 "detect_field_types",
197 "detect_logic_family",
198 "detect_magic_bytes",
199 "detect_message_framing",
200 "detect_protocol",
201 "detect_sequence_patterns",
202 "extract_messages",
203 "find_all_magic_bytes",
204 "find_conserved_regions",
205 "find_dependencies",
206 "find_message_dependencies",
207 "find_variable_regions",
208 "generate_parser",
209 "get_adaptive_parameters",
210 "get_decoder",
211 "get_library",
212 "get_optimal_domain_order",
213 "get_protocol",
214 "infer_format",
215 "infer_rpni",
216 "infer_with_uncertainty",
217 "list_protocols",
218 "load_protocol",
219 "minimize_dfa",
220 "parser_to_python",
221 "parser_to_yaml",
222 "reassemble_tcp_stream",
223 "reassemble_udp_stream",
224 "recommend_analyses",
225 "suggest_measurements",
226 "to_dot",
227 "to_networkx",
228]