#ifdef HAVE_CONFIG_H
#include "config.h"
#endif /* HAVE_CONFIG_H */

#include <arpa/inet.h>
#include <stdlib.h>
#include <string.h>

#include <atalk/adouble.h>
#include <atalk/errchk.h>
#include <atalk/logger.h>
#include <atalk/util.h>

#define FILEIOFF_ATTR 14
#define AFPFILEIOFF_ATTR 2

/*
   Note:
   the "shared" and "invisible" attributes are opaque and stored and
   retrieved from the FinderFlags. This fixes Bug #2802236:
   <https://sourceforge.net/tracker/?func=detail&aid=2802236&group_id=8642&atid=108642>
 */
int ad_getattr(const struct adouble *ad, uint16_t *attr)
{
    uint16_t fflags;
    char *adp = NULL;
    *attr = 0;

    if (ad_getentryoff(ad, ADEID_AFPFILEI)
            && (adp = ad_entry(ad, ADEID_AFPFILEI)) != NULL) {
        memcpy(attr, adp + AFPFILEIOFF_ATTR, 2);

        /* Now get opaque flags from FinderInfo */
        if ((adp = ad_entry(ad, ADEID_FINDERI)) != NULL) {
            memcpy(&fflags, adp + FINDERINFO_FRFLAGOFF, 2);
        } else {
            LOG(log_debug, logtype_default, "ad_getattr(%s): invalid FinderInfo",
                ad->ad_name);
            memset(&fflags, 0, 2);
        }

        if (fflags & htons(FINDERINFO_INVISIBLE)) {
            *attr |= htons(ATTRBIT_INVISIBLE);
        } else {
            *attr &= htons(~ATTRBIT_INVISIBLE);
        }

        /*
         * This one is tricky, I actually got it wrong the first time:
         * for directories bit 1<<1 is ATTRBIT_EXPFLDR and is NOT opaque !
         */
        if (!(ad->ad_adflags & ADFLAGS_DIR)) {
            if (fflags & htons(FINDERINFO_ISHARED)) {
                *attr |= htons(ATTRBIT_MULTIUSER);
            } else {
                *attr &= htons(~ATTRBIT_MULTIUSER);
            }
        }
    }

    *attr |= htons(ad->ad_open_forks);
    return 0;
}

/* ----------------- */
int ad_setattr(const struct adouble *ad, const uint16_t attribute)
{
    uint16_t fflags;
    char *ade = NULL, *adp = NULL;
    /* we don't save open forks indicator */
    uint16_t attr = attribute & ~htons(ATTRBIT_DOPEN | ATTRBIT_ROPEN);

    /* Proactively (10.4 does indeed try to set ATTRBIT_MULTIUSER (=ATTRBIT_EXPFLDR)
       for dirs with SetFile -a M <dir> ) disable all flags not defined for dirs. */
    if (ad->ad_adflags & ADFLAGS_DIR) {
        attr &= ~(ATTRBIT_MULTIUSER | ATTRBIT_NOWRITE | ATTRBIT_NOCOPY);
    }

    if (ad_getentryoff(ad, ADEID_AFPFILEI)
            && (ade = ad_entry(ad, ADEID_AFPFILEI)) != NULL
            && ad_getentryoff(ad, ADEID_FINDERI)
            && (adp = ad_entry(ad, ADEID_FINDERI)) != NULL) {
        memcpy(ade + AFPFILEIOFF_ATTR, &attr, sizeof(attr));
        /* Now set opaque flags in FinderInfo too */
        memcpy(&fflags, adp + FINDERINFO_FRFLAGOFF, 2);

        if (attr & htons(ATTRBIT_INVISIBLE)) {
            fflags |= htons(FINDERINFO_INVISIBLE);
        } else {
            fflags &= htons(~FINDERINFO_INVISIBLE);
        }

        /* See above comment in ad_getattr() */
        if (attr & htons(ATTRBIT_MULTIUSER)) {
            if (!(ad->ad_adflags & ADFLAGS_DIR)) {
                fflags |= htons(FINDERINFO_ISHARED);
            }
        } else {
            fflags &= htons(~FINDERINFO_ISHARED);
        }

        memcpy(adp + FINDERINFO_FRFLAGOFF, &fflags, 2);
    }

    return 0;
}

/* --------------
 * save file/folder ID in AppleDoubleV2 netatalk private parameters
 * return 1 if resource fork has been modified
 * return -1 on error.
 */
int ad_setid(struct adouble *adp, const dev_t dev, const ino_t ino,
             const uint32_t id, const cnid_t did, const void *stamp)
{
    EC_INIT;
    uint32_t tmp;
    char *ade = NULL;
    ssize_t id_len = -1, dev_len = -1, ino_len = -1, did_len = -1, syn_len = -1;
    id_len = ad_getentrylen(adp, ADEID_PRIVID);
    ad_setentrylen(adp, ADEID_PRIVID, sizeof(id));
    tmp = id;

    if (adp->ad_vers == AD_VERSION_EA) {
        tmp = htonl(tmp);
    }

    ade = ad_entry(adp, ADEID_PRIVID);

    if (ade == NULL) {
        LOG(log_warning, logtype_ad, "ad_setid(%s): failed to set ADEID_PRIVID",
            adp->ad_name);
        EC_FAIL;
    }

    memcpy(ade, &tmp, sizeof(tmp));
    dev_len = ad_getentrylen(adp, ADEID_PRIVDEV);
    ad_setentrylen(adp, ADEID_PRIVDEV, sizeof(dev_t));
    ade = ad_entry(adp, ADEID_PRIVDEV);

    if (ade == NULL) {
        LOG(log_warning, logtype_ad, "ad_setid(%s): failed to set ADEID_PRIVDEV",
            adp->ad_name);
        EC_FAIL;
    }

    if (adp->ad_options & ADVOL_NODEV) {
        memset(ade, 0, sizeof(dev_t));
    } else {
        memcpy(ade, &dev, sizeof(dev_t));
    }

    ino_len = ad_getentrylen(adp, ADEID_PRIVINO);
    ad_setentrylen(adp, ADEID_PRIVINO, sizeof(ino_t));
    ade = ad_entry(adp, ADEID_PRIVINO);

    if (ade == NULL) {
        LOG(log_warning, logtype_ad, "ad_setid(%s): failed to set ADEID_PRIVINO",
            adp->ad_name);
        EC_FAIL;
    }

    memcpy(ade, &ino, sizeof(ino_t));

    if (adp->ad_vers != AD_VERSION_EA) {
        did_len = ad_getentrylen(adp, ADEID_DID);
        ad_setentrylen(adp, ADEID_DID, sizeof(did));
        ade = ad_entry(adp, ADEID_DID);

        if (ade == NULL) {
            LOG(log_warning, logtype_ad, "ad_setid(%s): failed to set ADEID_DID",
                adp->ad_name);
            EC_FAIL;
        }

        memcpy(ade, &did, sizeof(did));
    }

    syn_len = ad_getentrylen(adp, ADEID_PRIVSYN);
    ad_setentrylen(adp, ADEID_PRIVSYN, ADEDLEN_PRIVSYN);
    ade = ad_entry(adp, ADEID_PRIVSYN);

    if (ade == NULL) {
        LOG(log_warning, logtype_ad, "ad_setid(%s): failed to set ADEID_PRIVSYN",
            adp->ad_name);
        EC_FAIL;
    }

    memcpy(ade, stamp, ADEDLEN_PRIVSYN);
EC_CLEANUP:

    if (ret != 0) {
        if (id_len != -1) {
            ad_setentrylen(adp, ADEID_PRIVID, id_len);
        }

        if (dev_len != -1) {
            ad_setentrylen(adp, ADEID_PRIVDEV, dev_len);
        }

        if (ino_len != -1) {
            ad_setentrylen(adp, ADEID_PRIVINO, ino_len);
        }

        if (did_len != -1) {
            ad_setentrylen(adp, ADEID_DID, did_len);
        }

        if (syn_len != -1) {
            ad_setentrylen(adp, ADEID_PRIVSYN, syn_len);
        }

        return 0;
    }

    return 1;
}

/*
 * Retrieve stored file / folder. Callers should treat a return of CNID_INVALID (0) as an invalid value.
 */
uint32_t ad_getid(struct adouble *adp, const dev_t st_dev, const ino_t st_ino,
                  const cnid_t did, const void *stamp _U_)
{
    uint32_t aint = 0;
    dev_t  dev;
    ino_t  ino;
    cnid_t a_did = 0;

    if (adp) {
        if (sizeof(dev_t) == ad_getentrylen(adp, ADEID_PRIVDEV)) {
            char *ade = NULL;
            ade = ad_entry(adp, ADEID_PRIVDEV);

            if (ade == NULL) {
                LOG(log_warning, logtype_ad, "ad_getid: failed to retrieve ADEID_PRIVDEV\n");
                return CNID_INVALID;
            }

            memcpy(&dev, ade, sizeof(dev_t));
            ade = ad_entry(adp, ADEID_PRIVINO);

            if (ade == NULL) {
                LOG(log_warning, logtype_ad, "ad_getid: failed to retrieve ADEID_PRIVINO\n");
                return CNID_INVALID;
            }

            memcpy(&ino, ade, sizeof(ino_t));

            if (adp->ad_vers != AD_VERSION_EA) {
                /* ADEID_DID is not stored for AD_VERSION_EA */
                ade = ad_entry(adp, ADEID_DID);

                if (ade == NULL) {
                    LOG(log_warning, logtype_ad, "ad_getid: failed to retrieve ADEID_DID\n");
                    return CNID_INVALID;
                }

                memcpy(&a_did, ade, sizeof(cnid_t));
            }

            if (((adp->ad_options & ADVOL_NODEV) || (dev == st_dev))
                    && ino == st_ino
                    && (!did || a_did == 0 || a_did == did)) {
                ade = ad_entry(adp, ADEID_PRIVID);

                if (ade == NULL) {
                    LOG(log_warning, logtype_ad, "ad_getid: failed to retrieve ADEID_PRIVID\n");
                    return CNID_INVALID;
                }

                memcpy(&aint, ade, sizeof(aint));

                if (adp->ad_vers == AD_VERSION2) {
                    return aint;
                } else {
                    return ntohl(aint);
                }
            }
        }
    }

    return CNID_INVALID;
}

/* ----------------------------- */
uint32_t ad_forcegetid(struct adouble *adp)
{
    uint32_t aint = 0;

    if (adp) {
        char *ade = NULL;
        ade = ad_entry(adp, ADEID_PRIVID);

        if (ade == NULL) {
            return CNID_INVALID;
        }

        memcpy(&aint, ade, sizeof(aint));

        if (adp->ad_vers == AD_VERSION2) {
            return aint;
        } else {
            return ntohl(aint);
        }
    }

    return CNID_INVALID;
}

/* -----------------
 * set resource fork filename attribute.
 */
int ad_setname(struct adouble *ad, const char *path)
{
    int len;

    if ((len = strlen(path)) > ADEDLEN_NAME) {
        len = ADEDLEN_NAME;
    }

    if (path && ad_getentryoff(ad, ADEID_NAME)) {
        char *ade = NULL;
        ad_setentrylen(ad, ADEID_NAME, len);
        ade = ad_entry(ad, ADEID_NAME);

        if (ade == NULL) {
            return -1;
        }

        memcpy(ade, path, len);
        return 1;
    }

    return 0;
}
