opkg: implement new opkg_upgrade_package and opkg_upgrade_all functions
[oweals/opkg-lede.git] / libopkg / opkg.c
index 3a662d2b24dff2806b53b2b6e43f7756fdc0095e..e72693c4f097edce1f0e86737f863be981a10a83 100644 (file)
@@ -39,6 +39,12 @@ struct _opkg_t
   opkg_option_t *options;
 };
 
+#define opkg_assert(expr) if (!(expr)) { \
+   printf ("opkg: file %s: line %d (%s): Assertation '%s' failed",\
+       __FILE__, __LINE__, __PRETTY_FUNCTION__, #expr); abort(); }
+
+#define progress(percent) if (progress_callback) progress_callback (opkg, percent, user_data);
+
 /** Private Functions ***/
 
 
@@ -81,6 +87,39 @@ opkg_configure_packages(opkg_conf_t *conf, char *pkg_name)
   return err;
 }
 
+struct _curl_cb_data
+{
+  opkg_progress_callback_t cb;
+  opkg_t *opkg;
+  void *user_data;
+  int start_range;
+  int finish_range;
+};
+
+int
+curl_progress_cb (struct _curl_cb_data *cb_data,
+                   double t, /* dltotal */
+                   double d, /* dlnow */
+                   double ultotal,
+                   double ulnow)
+{
+  int p = (t) ? d*100/t : 0;
+  static int prev = -1;
+
+  /* prevent the same value being sent twice (can occur due to rounding) */
+  if (p == prev)
+    return 0;
+  prev = p;
+
+  if (t < 1)
+    return 0;
+
+  (cb_data->cb) (cb_data->opkg,
+      cb_data->start_range + (d/t * ((cb_data->finish_range - cb_data->start_range))),
+      cb_data->user_data);
+
+  return 0;
+}
 
 
 /*** Public API ***/
@@ -104,6 +143,8 @@ opkg_new ()
 void
 opkg_free (opkg_t *opkg)
 {
+  opkg_assert (opkg != NULL);
+
   opkg_conf_deinit (opkg->conf);
   args_deinit (opkg->args);
 }
@@ -111,8 +152,13 @@ opkg_free (opkg_t *opkg)
 int
 opkg_read_config_files (opkg_t *opkg)
 {
-  args_t *a = opkg->args;
-  opkg_conf_t *c = opkg->conf;
+  args_t *a;
+  opkg_conf_t *c;
+
+  opkg_assert (opkg != NULL);
+
+  a = opkg->args;
+  c = opkg->conf;
 
   /* Unfortunatly, the easiest way to re-read the config files right now is to
    * throw away opkg->conf and start again */
@@ -164,11 +210,13 @@ void
 opkg_get_option (opkg_t *opkg, char *option, void **value)
 {
   int i = 0;
-  opkg_option_t *options = opkg->options;
+  opkg_option_t *options;
 
-  /* can't store a value in a NULL pointer! */
-  if (!value)
-    return;
+  opkg_assert (opkg != NULL);
+  opkg_assert (option != NULL);
+  opkg_assert (value != NULL);
+
+  options = opkg->options;
 
   /* look up the option
    * TODO: this would be much better as a hash table
@@ -204,11 +252,13 @@ void
 opkg_set_option (opkg_t *opkg, char *option, void *value)
 {
   int i = 0;
-  opkg_option_t *options = opkg->options;
+  opkg_option_t *options;
 
-  /* NULL values are not defined */
-  if (!value)
-    return;
+  opkg_assert (opkg != NULL);
+  opkg_assert (option != NULL);
+  opkg_assert (value != NULL);
+
+  options = opkg->options;
 
   /* look up the option
    * TODO: this would be much better as a hash table
@@ -249,11 +299,15 @@ opkg_install_package (opkg_t *opkg, const char *package_name, opkg_progress_call
   int err;
   char *package_id = NULL;
 
-  progress_callback (opkg, 0, user_data);
+  opkg_assert (opkg != NULL);
+  opkg_assert (package_name != NULL);
+
+  progress (0);
 
   /* download the package */
   opkg_prepare_url_for_install (opkg->conf, package_name, &package_id);
-  progress_callback (opkg, 50, user_data);
+
+  progress (50);
 
   /* ... */
   pkg_info_preinstall_check (opkg->conf);
@@ -274,7 +328,7 @@ opkg_install_package (opkg_t *opkg, const char *package_name, opkg_progress_call
   if (err)
     return err;
 
-  progress_callback (opkg, 75, user_data);
+  progress (75);
 
   /* run configure scripts, etc. */
   err = opkg_configure_packages (opkg->conf, NULL);
@@ -285,7 +339,7 @@ opkg_install_package (opkg_t *opkg, const char *package_name, opkg_progress_call
   opkg_conf_write_status_files (opkg->conf);
   pkg_write_changed_filelists (opkg->conf);
 
-  progress_callback (opkg, 100, user_data);
+  progress (100);
   return 0;
 }
 
@@ -295,13 +349,10 @@ opkg_remove_package (opkg_t *opkg, const char *package_name, opkg_progress_callb
   pkg_t *pkg = NULL;
   pkg_t *pkg_to_remove;
 
-  if (!opkg)
-    return 1;
-
-  if (!package_name)
-    return 1;
+  opkg_assert (opkg != NULL);
+  opkg_assert (package_name != NULL);
 
-  progress_callback (opkg, 0, user_data);
+  progress (0);
 
   pkg_info_preinstall_check (opkg->conf);
 
@@ -309,7 +360,7 @@ opkg_remove_package (opkg_t *opkg, const char *package_name, opkg_progress_callb
 
   pkg_hash_fetch_all_installed (&opkg->conf->pkg_hash, installed_pkgs);
 
-  progress_callback (opkg, 25, user_data);
+  progress (25);
 
   pkg = pkg_hash_fetch_installed_by_name (&opkg->conf->pkg_hash, package_name);
 
@@ -325,7 +376,7 @@ opkg_remove_package (opkg_t *opkg, const char *package_name, opkg_progress_callb
     return 1;
   }
 
-  progress_callback (opkg, 75, user_data);
+  progress (75);
 
   if (opkg->conf->restrict_to_default_dest)
   {
@@ -339,7 +390,7 @@ opkg_remove_package (opkg_t *opkg, const char *package_name, opkg_progress_callb
   }
 
 
-  progress_callback (opkg, 75, user_data);
+  progress (75);
 
   opkg_remove_pkg (opkg->conf, pkg_to_remove, 0);
 
@@ -348,20 +399,87 @@ opkg_remove_package (opkg_t *opkg, const char *package_name, opkg_progress_callb
   pkg_write_changed_filelists (opkg->conf);
 
 
-  progress_callback (opkg, 100, user_data);
+  progress (100);
   return 0;
 }
 
 int
 opkg_upgrade_package (opkg_t *opkg, const char *package_name, opkg_progress_callback_t progress_callback, void *user_data)
 {
-  return 1;
+  pkg_t *pkg;
+
+  opkg_assert (opkg != NULL);
+  opkg_assert (package_name != NULL);
+
+  progress (0);
+
+  pkg_info_preinstall_check (opkg->conf);
+
+  if (opkg->conf->restrict_to_default_dest)
+  {
+    pkg = pkg_hash_fetch_installed_by_name_dest (&opkg->conf->pkg_hash,
+                                                 package_name,
+                                                 opkg->conf->default_dest);
+    if (pkg == NULL)
+    {
+      /* XXX: Error: Package not installed in default_dest */
+      return 1;
+    }
+  }
+  else
+  {
+    pkg = pkg_hash_fetch_installed_by_name (&opkg->conf->pkg_hash,
+                                            package_name);
+  }
+
+  if (!pkg)
+  {
+    /* XXX: Error: Package not installed */
+    return 1;
+  }
+
+  progress (25);
+
+  opkg_upgrade_pkg (opkg->conf, pkg);
+  progress (75);
+
+  opkg_configure_packages (opkg->conf, NULL);
+  progress (100);
+  return 0;
 }
 
 int
 opkg_upgrade_all (opkg_t *opkg, opkg_progress_callback_t progress_callback, void *user_data)
 {
-  return 1;
+  pkg_vec_t *installed;
+  int err = 0;
+  int i;
+  pkg_t *pkg;
+
+  opkg_assert (opkg != NULL);
+  progress (0);
+
+  installed = pkg_vec_alloc ();
+  pkg_info_preinstall_check (opkg->conf);
+
+  pkg_hash_fetch_all_installed (&opkg->conf->pkg_hash, installed);
+  for (i = 0; i < installed->len; i++)
+  {
+    pkg = installed->pkgs[i];
+    err += opkg_upgrade_pkg (opkg->conf, pkg);
+    progress (100 * i / installed->len);
+  }
+  pkg_vec_free (installed);
+
+  if (err)
+    return 1;
+
+  err = opkg_configure_packages (opkg->conf, NULL);
+  if (err)
+    return 1;
+
+  progress (100);
+  return 0;
 }
 
 int
@@ -374,7 +492,9 @@ opkg_update_package_lists (opkg_t *opkg, opkg_progress_callback_t progress_callb
   pkg_src_t *src;
   int sources_list_count, sources_done;
 
-  progress_callback (opkg, 0, user_data);
+  opkg_assert (opkg != NULL);
+
+  progress (0);
 
   sprintf_alloc (&lists_dir, "%s",
                  (opkg->conf->restrict_to_default_dest)
@@ -436,11 +556,19 @@ opkg_update_package_lists (opkg_t *opkg, opkg_progress_callback_t progress_callb
     {
       char *tmp_file_name;
       FILE *in, *out;
+      struct _curl_cb_data cb_data;
 
       sprintf_alloc (&tmp_file_name, "%s/%s.gz", tmp, src->name);
 
       /* XXX: Note: downloading url */
-      err = opkg_download (opkg->conf, url, tmp_file_name);
+
+      cb_data.cb = progress_callback;
+      cb_data.opkg = opkg;
+      cb_data.user_data = user_data;
+      cb_data.start_range = 100 * sources_done / sources_list_count;
+      cb_data.finish_range = 100 * (sources_done + 1) / sources_list_count;
+
+      err = opkg_download (opkg->conf, url, tmp_file_name, (curl_progress_func) curl_progress_cb, &cb_data);
 
       if (err == 0)
       {
@@ -459,7 +587,7 @@ opkg_update_package_lists (opkg_t *opkg, opkg_progress_callback_t progress_callb
       }
     }
     else
-      err = opkg_download (opkg->conf, url, list_file_name);
+      err = opkg_download (opkg->conf, url, list_file_name, NULL, NULL);
 
     if (err)
     {
@@ -481,7 +609,7 @@ opkg_update_package_lists (opkg_t *opkg, opkg_progress_callback_t progress_callb
 
     sprintf_alloc (&tmp_file_name, "%s/%s", tmp, "Packages.sig");
 
-    err = opkg_download (opkg->conf, url, tmp_file_name);
+    err = opkg_download (opkg->conf, url, tmp_file_name, NULL, NULL);
     if (err)
     {
       /* XXX: Warning: Download failed */
@@ -510,8 +638,9 @@ opkg_update_package_lists (opkg_t *opkg, opkg_progress_callback_t progress_callb
     free (list_file_name);
 
     sources_done++;
-    progress_callback (opkg, 100 * sources_done / sources_list_count, user_data);
+    progress (100 * sources_done / sources_list_count);
   }
+
   rmdir (tmp);
   free (tmp);
   free (lists_dir);