#!/usr/bin/python3
#
# Copyright 2020 Centreon
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
# For more information : contact@centreon.com
#
# This script is a little tcp server working on port 5669. It can simulate
# a cbd instance. It is useful to test the validity of BBDO packets sent by
# centengine.

import socket
import sys
import time
from datetime import datetime
import sys


def welcome(bbdo_version: str):
    if (bbdo_version >= "3.0.0"):
        s = "\x65\xfe\x00\x0e\x00\x02\x00\x01\x00\x00\x00\x01\x00\x00\x00\x01\x00\x03\x00\x00\x00\x00\x54\x4c\x53\x00\x00\x00\x00\x00"
    else:
        s = "\x65\xfe\x00\x0e\x00\x02\x00\x01\x00\x00\x00\x01\x00\x00\x00\x01\x00\x02\x00\x00\x00\x00\x54\x4c\x53\x00\x00\x00\x00\x00"
    retval = bytearray()
    retval.extend(map(ord, s))
    return retval


def get_header(header):
    h = int.from_bytes(header[0:2], byteorder='big')
    s = int.from_bytes(header[2:4], byteorder='big')
    t = int.from_bytes(header[4:8], byteorder='big')
    src = int.from_bytes(header[8:12], byteorder='big')
    dst = int.from_bytes(header[12:16], byteorder='big')
    print("chksum: ", hex(h))
    print("size: ", s)
    print("type: ", t)
    print("src: ", src)
    print("dst: ", dst)
    assert (0 <= dst <= 3)
    assert (0 <= src <= 3)
    return h, s, t, src, dst


host_addr = 'localhost'
host_port = 5671

s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect((host_addr, host_port))

bbdo_version = "2.0.0"
if len(sys.argv) > 2:
    bbdo_version = sys.argv[2]
w = welcome(bbdo_version)
s.send(w)
header = s.recv(16)
chksum, size, typ, src, dst = get_header(header)
content = s.recv(size)

while True:
    time.sleep(float(sys.argv[1]))
    l = size - len(content)
    content = s.recv(l)
