/* release.c - the opkg package management system

   Copyright (C) 2010,2011 Javier Palacios

   This program is free software; you can redistribute it and/or
   modify it under the terms of the GNU General Public License as
   published by the Free Software Foundation; either version 2, or (at
   your option) any later version.

   This program is distributed in the hope that it will be useful, but
   WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
   General Public License for more details.
*/

#include <unistd.h>
#include <ctype.h>

#include "release.h"
#include "opkg_utils.h"
#include "libbb/libbb.h"

#include "opkg_download.h"
#include "sprintf_alloc.h"

#include "release_parse.h"

#include "parse_util.h"
#include "file_util.h"

static void
release_init(release_t *release)
{
     release->name = NULL;
     release->datestring = NULL;
     release->architectures = NULL;
     release->architectures_count = 0;
     release->components = NULL;
     release->components_count = 0;
     release->complist = NULL;
     release->complist_count = 0;
}

release_t *
release_new(void)
{
     release_t *release;

     release = xcalloc(1, sizeof(release_t));
     release_init(release);

     return release;
}

void
release_deinit(release_t *release)
{
    int i;

    free(release->name);
    free(release->datestring);

    for(i = 0; i < release->architectures_count; i++){
	free(release->architectures[i]);
    }
    free(release->architectures);

    for(i = 0; i < release->components_count; i++){
	free(release->components[i]);
    }
    free(release->components);

    for(i = 0; i < release->complist_count; i++){
	free(release->complist[i]);
    }
    free(release->complist);

}

int
release_init_from_file(release_t *release, const char *filename)
{
	int err = 0;
	FILE *release_file;

	release_file = fopen(filename, "r");
	if (release_file == NULL) {
		opkg_perror(ERROR, "Failed to open %s", filename);
		return -1;
	}

	err=release_parse_from_stream(release, release_file);
	if (!err) {
		if (!release_arch_supported(release)) {
			opkg_msg(ERROR, "No valid architecture found on Release file.\n");
			err = -1;
		}
	}

	return err;
}

const char *
item_in_list(const char *comp, char **complist, const unsigned int count)
{
     int i;

     if (!complist)
	  return comp;

     for(i = 0; i < count; i++){
	  if (strcmp(comp, complist[i]) == 0)
		    return complist[i];
     }

     return NULL;
}

int
release_arch_supported(release_t *release)
{
     nv_pair_list_elt_t *l;

     list_for_each_entry(l , &conf->arch_list.head, node) {
	  nv_pair_t *nv = (nv_pair_t *)l->data;
	  if (item_in_list(nv->name, release->architectures, release->architectures_count)) {
	       opkg_msg(DEBUG, "Arch %s (priority %s) supported for dist %s.\n",
			       nv->name, nv->value, release->name);
	       return 1;
	  }
     }

     return 0;
}

int
release_comps_supported(release_t *release, const char *complist)
{
     int ret = 1;
     int i;

     if (complist) {
	  release->complist = parse_list(complist, &release->complist_count, ' ', 1);
	  for(i = 0; i < release->complist_count; i++){
	       if (!item_in_list(release->complist[i], release->components, release->components_count)) {
		    opkg_msg(ERROR, "Component %s not supported for dist %s.\n",
				    release->complist[i], release->name);
		    ret = 0;
	       }
	  }
     }

     return ret;
}

const char **
release_comps(release_t *release, unsigned int *count)
{
     char **comps = release->complist;

     if (!comps) {
	  comps = release->components;
	  *count = release->components_count;
     } else {
	  *count = release->complist_count;
     }

     return (const char **)comps;
}

int
release_download(release_t *release, pkg_src_t *dist, char *lists_dir, char *tmpdir)
{
     int ret = 0;
     unsigned int ncomp;
     const char **comps = release_comps(release, &ncomp);
     nv_pair_list_elt_t *l;
     int i;

     for(i = 0; i < ncomp; i++){
	  int err = 0;
	  char *prefix;

	  sprintf_alloc(&prefix, "%s/dists/%s/%s/binary", dist->value, dist->name,
			comps[i]);

	  list_for_each_entry(l , &conf->arch_list.head, node) {
	       char *url;
	       char *tmp_file_name, *list_file_name;
	       char *subpath = NULL;

	       nv_pair_t *nv = (nv_pair_t *)l->data;

	       sprintf_alloc(&list_file_name, "%s/%s-%s-%s", lists_dir, dist->name, comps[i], nv->name);

	       sprintf_alloc(&tmp_file_name, "%s/%s-%s-%s%s", tmpdir, dist->name, comps[i], nv->name, ".gz");

	       sprintf_alloc(&subpath, "%s/binary-%s/%s", comps[i], nv->name, dist->gzip ? "Packages.gz" : "Packages");

	       if (dist->gzip) {
	       sprintf_alloc(&url, "%s-%s/Packages.gz", prefix, nv->name);
	       err = opkg_download(url, tmp_file_name, NULL, NULL, 1);
	       if (!err) {
		    err = release_verify_file(release, tmp_file_name, subpath);
		    if (err) {
			 unlink (tmp_file_name);
			 unlink (list_file_name);
		    }
	       }
	       if (!err) {
		    FILE *in, *out;
		    opkg_msg(NOTICE, "Inflating %s.\n", url);
		    in = fopen (tmp_file_name, "r");
		    out = fopen (list_file_name, "w");
		    if (in && out) {
			 err = unzip (in, out);
			 if (err)
			      opkg_msg(INFO, "Corrumpt file at %s.\n", url);
		    } else
			 err = 1;
		    if (in)
			 fclose (in);
		    if (out)
			 fclose (out);
		    unlink (tmp_file_name);
	       }
	       free(url);
	       }

	       if (err) {
		    sprintf_alloc(&url, "%s-%s/Packages", prefix, nv->name);
		    err = opkg_download(url, list_file_name, NULL, NULL, 1);
		    if (!err) {
			 err = release_verify_file(release, tmp_file_name, subpath);
			 if (err)
			      unlink (list_file_name);
		    }
		    free(url);
	       }

	       free(tmp_file_name);
	       free(list_file_name);
	  }

	  if(err)
	       ret = 1;

	  free(prefix);
     }

     return ret;
}

int
release_get_size(release_t *release, const char *pathname)
{
     const cksum_t *cksum;

     if (release->md5sums) {
	  cksum = cksum_list_find(release->md5sums, pathname);
	  return cksum->size;
     }

#ifdef HAVE_SHA256
     if (release->sha256sums) {
	  cksum = cksum_list_find(release->sha256sums, pathname);
	  return cksum->size;
     }
#endif

     return -1;
}

const char *
release_get_md5(release_t *release, const char *pathname)
{
     const cksum_t *cksum;

     if (release->md5sums) {
	  cksum = cksum_list_find(release->md5sums, pathname);
	  return cksum->value;
     }

     return '\0';
}

#ifdef HAVE_SHA256
const char *
release_get_sha256(release_t *release, const char *pathname)
{
     const cksum_t *cksum;

     if (release->sha256sums) {
	  cksum = cksum_list_find(release->sha256sums, pathname);
	  return cksum->value;
     }

     return '\0';
}
#endif

int
release_verify_file(release_t *release, const char* file_name, const char *pathname)
{
     struct stat f_info;
     char *f_md5 = NULL;
     const char *md5 = release_get_md5(release, pathname);
#ifdef HAVE_SHA256
     char *f_sha256 = NULL;
     const char *sha256 = release_get_sha256(release, pathname);
#endif
     int ret = 0;

     if (stat(file_name, &f_info) || (f_info.st_size!=release_get_size(release, pathname))) {
	  opkg_msg(ERROR, "Size verification failed for %s - %s.\n", release->name, pathname);
	  ret = 1;
     } else {

     f_md5 = file_md5sum_alloc(file_name);
#ifdef HAVE_SHA256
     f_sha256 = file_sha256sum_alloc(file_name);
#endif

     if (md5 && strcmp(md5, f_md5)) {
	  opkg_msg(ERROR, "MD5 verification failed for %s - %s.\n", release->name, pathname);
	  ret = 1;
#ifdef HAVE_SHA256
     } else if (sha256 && strcmp(sha256, f_sha256)) {
	  opkg_msg(ERROR, "SHA256 verification failed for %s - %s.\n", release->name, pathname);
	  ret = 1;
#endif
     }

     }

     free(f_md5);
#ifdef HAVE_SHA256
     free(f_sha256);
#endif

     return ret;
}