/* SPDX-License-Identifier: LGPL-2.1-or-later */

#include "sd-event.h"

#include "alloc-util.h"

#include "journal-remote-parse.h"

#include "log.h"

RemoteSource* source_free(RemoteSource *source) {

        if (!source)

                return NULL;

        journal_importer_cleanup(&source->importer);

        log_trace("Writer ref count %u", source->writer->n_ref);

        writer_unref(source->writer);

        sd_event_source_unref(source->event);

        sd_event_source_unref(source->buffer_event);

        free(source->encoding);

        return mfree(source);

}

/**

 * Initialize zero-filled source with given values. On success, takes

 * ownership of fd and writer, otherwise does not touch them. Always takes

 * ownership of name, even on failure.

 */

RemoteSource* source_new(int fd, bool passive_fd, char *name, Writer *writer) {

        RemoteSource *source;

        log_debug("Creating source for %sfd:%d (%s)",

                  passive_fd ? "passive " : "", fd, name);

        assert(fd >= 0);

        source = new0(RemoteSource, 1);

        if (!source) {

                free(name);

                return NULL;

        }

        source->importer = JOURNAL_IMPORTER_MAKE(fd);

        source->importer.passive_fd = passive_fd;

        source->importer.name = name;

        source->writer = writer;

        return source;

}

int process_source(RemoteSource *source, JournalFileFlags file_flags) {

        int r;

        assert(source);

        assert(source->writer);

        r = journal_importer_process_data(&source->importer);

        if (r <= 0)

                return r;

        /* We have a full event */

        log_trace("Received full event from source@%p fd:%d (%s)",

                  source, source->importer.fd, source->importer.name);

        if (source->importer.iovw.count == 0) {

                log_warning("Entry with no payload, skipping");

                goto freeing;

        }

        assert(source->importer.iovw.iovec);

        r = writer_write(source->writer,

                         &source->importer.iovw,

                         &source->importer.ts,

                         &source->importer.boot_id,

                         file_flags);

        if (IN_SET(r, -EBADMSG, -EADDRNOTAVAIL)) {

                log_warning_errno(r, "Entry is invalid, ignoring.");

                r = 0;

        } else if (r < 0)

                log_error_errno(r, "Failed to write entry of %zu bytes: %m",

                                iovw_size(&source->importer.iovw));

        else

                r = 1;

 freeing:

        journal_importer_drop_iovw(&source->importer);

        return r;

}