public ContraCoreClient(string hostname) { Task.Run(() => Read(hostname)); OnConnected += Setup; OnConnected += () => { Connected = true; Common.Logger.Info("Connected to ContraCore"); ConnectionComplete.Set(); OnStatusChange?.Invoke(); }; OnDisconnected += () => { Connected = false; GeneratingRules = false; GeneratingOUI = false; ReloadingBlacklist = false; ReloadingWhitelist = false; OnGenRulesChange?.Invoke(); OnGenOUIChange?.Invoke(); OnReloadBlacklistChange?.Invoke(); OnReloadWhitelistChange?.Invoke(); Common.Logger.Info("Disconnected from ContraCore"); OnStatusChange?.Invoke(); }; }
public async Task GenOUI() { GeneratingOUI = true; OnGenOUIChange?.Invoke(); await Send("gen_oui"); }
private async void Read(string hostname) { while (true) { try { _client = new TcpClient(hostname, 64417); _stream = _client.GetStream(); _reader = new StreamReader(_stream); if (!_connected) { OnConnected?.Invoke(); } _connected = true; } catch (Exception e) { Common.Logger.Warning("Failed to connect to server; retrying...", e: e, meta: new Fields { { "Hostname", hostname } }); Thread.Sleep(1000); if (_connected) { OnDisconnected?.Invoke(); } _connected = false; continue; } try { string line; while ((line = (await _reader.ReadLineAsync()) !) != null) { string cmd = line.Contains(" ") ? line.Substring(0, line.IndexOf(" ", StringComparison.Ordinal)) : line; string val = line.Substring(line.IndexOf(" ", StringComparison.Ordinal) + 1); if (cmd != "query" && cmd != "gen_oui.gen_progress") { Common.Logger.Debug($"NetMgr <- {cmd}"); } switch (cmd) { case "ping": await Send("ping.pong"); break; case "ping.pong": _pongEvent.Release(); break; case "get_cache.cache": List <Log> logs = JsonConvert.DeserializeObject <List <Log> >(val); await _cacheChannel.Writer.WriteAsync(logs); break; case "query": Log log = JsonConvert.DeserializeObject <Log>(val); Common.Logger.Debug($"Incoming: {log}"); _data.Enqueue(log); OnNewLog?.Invoke(log); break; case "gen_rules.sources": case "gen_rules.gen_progress": case "gen_rules.save_progress": case "gen_rules.saved_in": OnGenRulesCallback?.Invoke(val); break; case "gen_rules.complete": GeneratingRules = false; OnGenRulesChange?.Invoke(); break; case "gen_oui.gen_progress": OnGenOUICallback?.Invoke(val); break; case "gen_oui.generated_in": GeneratingOUI = false; OnGenOUIChange?.Invoke(); break; case "reload_config.complete": _reloadConfigCompleteEvent.Release(); break; case "reload_blacklist.reload_progress": case "reload_blacklist.reloaded_in": OnReloadBlacklistStatusCallback?.Invoke(val); break; case "reload_blacklist.contradb_offline": ReloadingBlacklist = false; OnReloadBlacklistErrorCallback?.Invoke(val); OnReloadBlacklistChange?.Invoke(); break; case "reload_blacklist.complete": ReloadingBlacklist = false; OnReloadBlacklistChange?.Invoke(); break; case "reload_whitelist.complete": ReloadingWhitelist = false; OnReloadWhitelistChange?.Invoke(); break; case "reload_whitelist.contradb_offline": case "reload_whitelist.error": ReloadingWhitelist = false; OnReloadWhitelistErrorCallback?.Invoke(val); OnReloadWhitelistChange?.Invoke(); break; default: Common.Logger.Error( $"Unmatched command received from NetManagerClient: {cmd}", new InvalidOperationException() ); break; } } } catch (Exception e) { Common.Logger.Warning("Failed to read from server; reconnecting...", e, new Fields { { "Hostname", hostname } }); Thread.Sleep(1000); OnDisconnected?.Invoke(); } } }