93 lines
2.2 KiB
C
93 lines
2.2 KiB
C
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
|
|
#include "adb.h"
|
|
#include "apk_applet.h"
|
|
#include "apk_print.h"
|
|
|
|
struct sign_ctx {
|
|
struct apk_ctx *ac;
|
|
struct adb_xfrm xfrm;
|
|
int reset_signatures : 1;
|
|
int signatures_written : 1;
|
|
};
|
|
|
|
#define ADBSIGN_OPTIONS(OPT) \
|
|
OPT(OPT_ADBSIGN_reset_signatures, "reset-signatures")
|
|
|
|
APK_OPT_APPLET(option_desc, ADBSIGN_OPTIONS);
|
|
|
|
static int option_parse_applet(void *pctx, struct apk_ctx *ac, int optch, const char *optarg)
|
|
{
|
|
struct sign_ctx *ctx = (struct sign_ctx *) pctx;
|
|
|
|
switch (optch) {
|
|
case OPT_ADBSIGN_reset_signatures:
|
|
ctx->reset_signatures = 1;
|
|
break;
|
|
default:
|
|
return -ENOTSUP;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static const struct apk_option_group optgroup_applet = {
|
|
.desc = option_desc,
|
|
.parse = option_parse_applet,
|
|
};
|
|
|
|
static int update_signatures(struct adb_xfrm *xfrm, struct adb_block *blk, struct apk_istream *is)
|
|
{
|
|
struct sign_ctx *ctx = container_of(xfrm, struct sign_ctx, xfrm);
|
|
struct apk_trust *trust = apk_ctx_get_trust(ctx->ac);
|
|
int r;
|
|
|
|
switch (blk ? adb_block_type(blk) : -1) {
|
|
case ADB_BLOCK_ADB:
|
|
return adb_c_block_copy(xfrm->os, blk, is, &xfrm->vfy);
|
|
case ADB_BLOCK_SIG:
|
|
if (ctx->reset_signatures)
|
|
break;
|
|
return adb_c_block_copy(xfrm->os, blk, is, NULL);
|
|
default:
|
|
if (!ctx->signatures_written) {
|
|
ctx->signatures_written = 1;
|
|
r = adb_trust_write_signatures(trust, &xfrm->db, &xfrm->vfy, xfrm->os);
|
|
if (r) return r;
|
|
}
|
|
if (!blk) break;
|
|
return adb_c_block_copy(xfrm->os, blk, is, NULL);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int adbsign_main(void *pctx, struct apk_ctx *ac, struct apk_string_array *args)
|
|
{
|
|
struct apk_out *out = &ac->out;
|
|
struct sign_ctx *ctx = pctx;
|
|
char **arg;
|
|
int r;
|
|
|
|
ctx->ac = ac;
|
|
foreach_array_item(arg, args) {
|
|
ctx->xfrm.is = apk_istream_from_file(AT_FDCWD, *arg);
|
|
ctx->xfrm.os = apk_ostream_to_file(AT_FDCWD, *arg, 0644);
|
|
adb_c_xfrm(&ctx->xfrm, update_signatures);
|
|
apk_istream_close(ctx->xfrm.is);
|
|
r = apk_ostream_close(ctx->xfrm.os);
|
|
if (r) apk_err(out, "%s: %s", *arg, apk_error_str(r));
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static struct apk_applet apk_adbsign = {
|
|
.name = "adbsign",
|
|
.context_size = sizeof(struct sign_ctx),
|
|
.optgroups = { &optgroup_global, &optgroup_signing, &optgroup_applet },
|
|
.main = adbsign_main,
|
|
};
|
|
|
|
APK_DEFINE_APPLET(apk_adbsign);
|