#include <stdbool.h>
#include <assert.h>
#include "protm.h"

size_t protm_size(const struct protm *pm) {
	switch (pm->type) {
		case PROTM_TYPE_ESTAB:
			return 2;

		case PROTM_TYPE_TERM:
			return 2;

		case PROTM_TYPE_DATA:
			return 7 + pm->data.size;

		case PROTM_TYPE_RR:
			return 3 + pm->rr.nrej;

		case PROTM_TYPE_RR_POLL:
			return 1;

		default:
			assert(false);
	}
}