diff options
| -rw-r--r-- | scripts/aurblup/aurblup.c | 38 |
1 files changed, 27 insertions, 11 deletions
diff --git a/scripts/aurblup/aurblup.c b/scripts/aurblup/aurblup.c index 8cf49e0..1d73fe3 100644 --- a/scripts/aurblup/aurblup.c +++ b/scripts/aurblup/aurblup.c @@ -15,6 +15,7 @@ #define mysql_die(...) die(__VA_ARGS__, mysql_error(c)); void die(const char *, ...); +void blacklist_add(const char *); void blacklist_sync(alpm_list_t *); alpm_list_t *get_package_list(alpm_list_t *); alpm_list_t *create_db_list(void); @@ -45,12 +46,26 @@ die(const char *format, ...) } void -blacklist_sync(alpm_list_t *pkgs) +blacklist_add(const char *name) { - alpm_list_t *r; - char *se; + char *esc = malloc(strlen(name) * 2 + 1); char query[1024]; + mysql_real_escape_string(c, esc, name, strlen(name)); + *(esc + strcspn(esc, "<=>")) = 0; + snprintf(query, 1024, "INSERT IGNORE INTO PackageBlacklist (Name) " + "VALUES ('%s');", esc); + free(esc); + + if (mysql_query(c, query)) + mysql_die("failed to query MySQL database (\"%s\"): %s\n", query); +} + +void +blacklist_sync(alpm_list_t *pkgs) +{ + alpm_list_t *r, *p; + if (mysql_query(c, "LOCK TABLES PackageBlacklist WRITE;")) mysql_die("failed to lock MySQL table: %s\n"); @@ -58,16 +73,17 @@ blacklist_sync(alpm_list_t *pkgs) mysql_die("failed to clear MySQL table: %s\n"); for (r = pkgs; r; r = alpm_list_next(r)) { - const char *s = alpm_pkg_get_name(alpm_list_getdata(r)); + pmpkg_t *pkg = alpm_list_getdata(r); - se = malloc(strlen(s) * 2 + 1); - mysql_real_escape_string(c, se, s, strlen(s)); - snprintf(query, 1024, "INSERT INTO PackageBlacklist (Name) " - "VALUES ('%s');", se); - free(se); + blacklist_add(alpm_pkg_get_name(pkg)); - if (mysql_query(c, query)) - mysql_die("failed to query MySQL database (\"%s\"): %s\n", query); + for (p = alpm_pkg_get_provides(pkg); p; p = alpm_list_next(p)) { + blacklist_add(alpm_list_getdata(p)); + } + + for (p = alpm_pkg_get_replaces(pkg); p; p = alpm_list_next(p)) { + blacklist_add(alpm_list_getdata(p)); + } } if (mysql_query(c, "UNLOCK TABLES;")) |