コード例 #1
0
    /// <summary>
    /// We have NOT seen this IP before, and we want to make
    /// a new one
    /// Author: Ben Hoffman
    /// </summary>
    public void NewComputer(Source jsonSourceData)
    {
        // Get a new computer device from the object pooler
        Computer newDevice = computerPooler.GetPooledObject().GetComponent <Computer>();

        // Set the DATA on this gameobject to the data from the JSON data
        newDevice.SourceInt = jsonSourceData.sourceIpInt;

        // Set this object as active in the hierachy so that you can actually see it
        newDevice.gameObject.SetActive(true);

        // Set the snort manager if we have one
        if (snortManager != null)
        {
            newDevice.snortManager = snortManager;
        }

        // Add the object to the dictionary
        computersDict.Add(jsonSourceData.sourceIpInt, newDevice);

        // Check the connections to this, if there are connections then add them to it's list
        CheckConnection(jsonSourceData);

        // Check if we can add it to a group
        IPGroupManager.currentIpGroups.CheckGroups(jsonSourceData.sourceIpInt);

        // If we have a device count text...
        if (deviceCountText != null)
        {
            // Update the UI that tells us how many devices there are
            deviceCountText.text = Computer.ComputerCount.ToString();
        }

        // ============== Sending the necessary info to draw lines between objects ======================= //

        // If there is a service runnign on this, then send it to the netflow controller to visualize it
        if (jsonSourceData.service != null)
        {
            connectionController.CheckPacketbeatData(jsonSourceData.sourceIpInt, jsonSourceData.destIpInt, jsonSourceData.service);
        }
        // Otherwise if it is UDP / TCP traffic...
        else if (jsonSourceData.protocol != null)
        {
            // Send the protocol
            connectionController.CheckPacketbeatData(jsonSourceData.sourceIpInt, jsonSourceData.destIpInt, jsonSourceData.protocol);
        }
    }
コード例 #2
0
    /// <summary>
    /// Loop through the data that we have and send it to the netflow
    /// controller if we should
    /// </summary>
    /// <param name="packetDataObj"></param>
    private IEnumerator CheckData(Packetbeat_Json_Data packetDataObj)
    {
        // ================= Check and make sure that our data is valid =====================
        // Make sure that our data is not null
        if (packetDataObj.hits.hits.Length == 0)
        {
            _UseLastSuccess = true;

            // Tell this to use the last successful query
            yield break;
        }

        // Let this know that we no longer need to bank on the last success
        if (_UseLastSuccess)
        {
            _UseLastSuccess = false;
        }

        // ============= Keep track of stuff to prevent duplicates =======================
        packetPerQuery = 0;
        // Set our latest packetbeat time to the most recent one
        _latest_time  = packetDataObj.hits.hits[packetDataObj.hits.hits.Length - 1]._source.timestamp;
        checkingState = CheckDataStates.Running;
        // ============== Actually loop through our hits data  =========================
        for (int i = 0; i < packetDataObj.hits.hits.Length; i++)
        {
            // Set the integer IP values of this object
            SetIntegerValues(packetDataObj.hits.hits[i]._source);

            // As long as what we got from those IP's is valid:
            if (packetDataObj.hits.hits[i]._source.destIpInt != 0 && packetDataObj.hits.hits[i]._source.sourceIpInt != 0)
            {
                // Change the protocol to HTTP if we want to, this is optional because
                // sometimes it is techincally incorrect
                if (assumeHttp && packetDataObj.hits.hits[i]._source.dest.port == 80 ||
                    packetDataObj.hits.hits[i]._source.dest.port == 8080)
                {
                    // This traffic is HTTP
                    packetDataObj.hits.hits[i]._source.transport = "http";
                }

                // Send the data to the netflow controller
                connectionController.CheckPacketbeatData(
                    packetDataObj.hits.hits[i]._source.sourceIpInt,
                    packetDataObj.hits.hits[i]._source.destIpInt,
                    packetDataObj.hits.hits[i]._source.transport);
                packetPerQuery++;
            }

            // Get them frames
            yield return(null);
        }
        checkingState = CheckDataStates.Done;
    }