static int _sceNpDrmGetVersionKey(u8 * key, u8 * act, u8 * rif, u32 flags)
{
char keypath[128];
int ret, result;
result = (*sceNpDrmGetVersionKey)(key, act, rif, flags);
if (g_is_custom_ps1) {
printk("%s: -> 0x%08X\n", __func__, result);
result = 0;
if (g_keys_bin_found) {
memcpy(key, g_keys, sizeof(g_keys));
}
printk("%s:[FAKE] -> 0x%08X\n", __func__, result);
} else {
get_keypath(keypath, sizeof(keypath));
if (result == 0) {
memcpy(g_keys, key, sizeof(g_keys));
ret = save_key(keypath, g_keys, sizeof(g_keys));
printk("%s: save_key -> %d\n", __func__, ret);
} else {
if (g_keys_bin_found) {
memcpy(key, g_keys, sizeof(g_keys));
result = 0;
}
}
}
return result;
}
int define_key(void)
{
struct _select_def group_conf;
struct key_struct key;
POINT *pts;
int i;
skip_key = 1;
if (keymem_total==0) {
key.key = 'A';
key.mapped[0] = 'A';
key.mapped[1] = 0;
key.status[0] = 0;
add_key(&key);
ask_define();
}
bzero(&group_conf, sizeof(struct _select_def));
group_conf.item_count = keymem_total;
pts = (POINT *) malloc(sizeof(POINT) * BBS_PAGESIZE);
for (i = 0; i < BBS_PAGESIZE; i++) {
pts[i].x = 2;
pts[i].y = i + 3;
}
group_conf.item_per_page = BBS_PAGESIZE;
/*
* 加上 LF_VSCROLL 才能用 LEFT 键退出
*/
group_conf.flag = LF_VSCROLL | LF_BELL | LF_LOOP | LF_MULTIPAGE;
group_conf.prompt = "◆";
group_conf.item_pos = pts;
// group_conf.arg = &arg;
group_conf.title_pos.x = 0;
group_conf.title_pos.y = 0;
group_conf.pos = 1; /* initialize cursor on the first mailgroup */
group_conf.page_pos = 1; /* initialize page to the first one */
group_conf.on_select = set_keydefine_select;
group_conf.show_data = set_keydefine_show;
group_conf.pre_key_command = set_keydefine_prekey;
group_conf.key_command = set_keydefine_key;
group_conf.show_title = set_keydefine_refresh;
group_conf.get_data = set_keydefine_getdata;
list_select_loop(&group_conf);
save_key();
free(pts);
skip_key = 0;
return 0;
}
/**
* Program main function.
*
* We leave the meat of the processing to the above functions. Here, we
* simply process the command line arguments.
*/
int
main(int argc, char *argv[])
{
int opt;
char *out_name;
FILE *keyfd, *infd, *outfd;
/* Parse command options */
while ((opt = getopt_long(argc, argv, opts_str, long_opts, NULL)) != -1) {
switch (opt) {
//case 'f': flags.force = true; break;
//case 'b': flags.burn_file = true; break;
/* Verbose option */
case 'v': flags.verbose = true; break;
/* Key size option */
case 's': flags.key_size = atoi(optarg);
if (flags.key_size != key_16_bytes
&& flags.key_size != key_24_bytes
&& flags.key_size != key_32_bytes) {
exit_error(PROGRAM_NAME ": Error: Invalid key size specified.\n");
}
break;
/* Key file name option */
case 'k': flags.key_file_name = optarg; break;
/* Specified output directory */
case 'd': flags.out_directory = optarg; break;
default:
exit_error(PROGRAM_NAME ": Error: Invalid option indicated.\n");
break;
}
}
/* Generate Encryption Key */
VERBOSE("Generating encryption key.\n");
key_init(&encrypt_key, flags.key_size);
/* Open file to save key */
/* Use default file name if not specified by user */
if (flags.key_file_name == NULL) {
VERBOSE("Key file not specified. Creating file '" DEFAULT_KEY_FILE "'.\n");
flags.key_file_name = DEFAULT_KEY_FILE;
}
/* Save key to file */
VERBOSE("Saving encryption key to file '%s'.\n", flags.key_file_name);
keyfd = fopen(flags.key_file_name, "w");
if (keyfd == NULL) {
exit_error(PROGRAM_NAME ": Error: Could not create encryption key" \
" file '%s'.\n",\
flags.key_file_name);
}
save_key(&encrypt_key, keyfd);
fclose(keyfd); /* Close file */
/* Encrypt specified files using newly generated key */
if ((optind == argc) || (argv[optind][0] == '-')) {
/* Encrypt standard input if no file specified. */
VERBOSE("Reading from Standard Input.\n");
infd = stdin;
outfd = fopen(DEFAULT_OUT_FILE OUTPUT_EXTENSION, "wb");
encrypt_file(infd, outfd, &encrypt_key);
optind++;
}
for (; optind < argc; optind++) {
infd = fopen(argv[optind], "rb"); /* Open file for reading */
if (infd == NULL) {
fprintf(stderr, PROGRAM_NAME ": Error: Failed to open file '%s'.\n",
argv[optind]);
}
else {
out_name = create_out_file_name(argv[optind]);
outfd = fopen(out_name, "wb");
if (outfd == NULL) {
fprintf(stderr, PROGRAM_NAME ": Error: Failed to create file '%s'.\n",
out_name);
fclose(infd);
}
else {
VERBOSE("Encrypting file '%s'...\n", argv[optind]);
if (encrypt_file(infd, outfd, &encrypt_key)) {
printf(PROGRAM_NAME ": Cipher '%s' created from file '%s'.\n",
out_name, argv[optind]);
}
fclose(infd);
fclose(outfd);
}
free(out_name);
}
}
printf(PROGRAM_NAME ": Encryption complete. Key stored at file '%s'.\n",
flags.key_file_name);
exit(EXIT_SUCCESS);
}
/*
* Message receiver handler.
*/
static pj_bool_t on_rx_message(pjsip_rx_data *rdata)
{
pjsip_msg *msg = rdata->msg_info.msg;
pj_str_t branch_param = rdata->msg_info.via->branch_param;
pj_status_t status;
if (pj_stricmp2(&branch_param, TEST1_BRANCH_ID) == 0 ||
pj_stricmp2(&branch_param, TEST2_BRANCH_ID) == 0)
{
/*
* TEST1_BRANCH_ID tests that non-INVITE transaction transmits 2xx
* final response using correct transport and terminates transaction
* after 32 seconds.
*
* TEST2_BRANCH_ID performs similar test for non-2xx final response.
*/
int status_code = (pj_stricmp2(&branch_param, TEST1_BRANCH_ID) == 0) ?
TEST1_STATUS_CODE : TEST2_STATUS_CODE;
if (msg->type == PJSIP_REQUEST_MSG) {
/* On received request, create UAS and respond with final
* response.
*/
pjsip_transaction *tsx;
status = pjsip_tsx_create_uas(&tsx_user, rdata, &tsx);
if (status != PJ_SUCCESS) {
app_perror(" error: unable to create transaction", status);
test_complete = -110;
return PJ_TRUE;
}
pjsip_tsx_recv_msg(tsx, rdata);
save_key(tsx);
send_response(rdata, tsx, status_code);
} else {
/* Verify the response received. */
++recv_count;
/* Verify status code. */
if (msg->line.status.code != status_code) {
PJ_LOG(3,(THIS_FILE, " error: incorrect status code"));
test_complete = -113;
}
/* Verify that no retransmissions is received. */
if (recv_count > 1) {
PJ_LOG(3,(THIS_FILE, " error: retransmission received"));
test_complete = -114;
}
}
return PJ_TRUE;
} else if (pj_stricmp2(&branch_param, TEST3_BRANCH_ID) == 0) {
/* TEST3_BRANCH_ID tests provisional response. */
if (msg->type == PJSIP_REQUEST_MSG) {
/* On received request, create UAS and respond with provisional
* response, then schedule timer to send final response.
*/
pjsip_transaction *tsx;
status = pjsip_tsx_create_uas(&tsx_user, rdata, &tsx);
if (status != PJ_SUCCESS) {
app_perror(" error: unable to create transaction", status);
test_complete = -116;
return PJ_TRUE;
}
pjsip_tsx_recv_msg(tsx, rdata);
save_key(tsx);
send_response(rdata, tsx, TEST3_PROVISIONAL_CODE);
schedule_send_response(rdata, &tsx->transaction_key,
TEST3_STATUS_CODE, 2000);
} else {
/* Verify the response received. */
++recv_count;
if (recv_count == 1) {
/* Verify status code. */
if (msg->line.status.code != TEST3_PROVISIONAL_CODE) {
PJ_LOG(3,(THIS_FILE, " error: incorrect status code"));
test_complete = -123;
}
} else if (recv_count == 2) {
/* Verify status code. */
if (msg->line.status.code != TEST3_STATUS_CODE) {
PJ_LOG(3,(THIS_FILE, " error: incorrect status code"));
test_complete = -124;
}
} else {
PJ_LOG(3,(THIS_FILE, " error: retransmission received"));
test_complete = -125;
}
}
return PJ_TRUE;
} else if (pj_stricmp2(&branch_param, TEST4_BRANCH_ID) == 0 ||
pj_stricmp2(&branch_param, TEST5_BRANCH_ID) == 0 ||
pj_stricmp2(&branch_param, TEST6_BRANCH_ID) == 0)
{
/* TEST4_BRANCH_ID: absorbs retransmissions in TRYING state. */
/* TEST5_BRANCH_ID: retransmit last response in PROCEEDING state. */
/* TEST6_BRANCH_ID: retransmit last response in COMPLETED state. */
if (msg->type == PJSIP_REQUEST_MSG) {
/* On received request, create UAS. */
pjsip_transaction *tsx;
PJ_LOG(4,(THIS_FILE, " received request (probably retransmission)"));
status = pjsip_tsx_create_uas(&tsx_user, rdata, &tsx);
if (status != PJ_SUCCESS) {
app_perror(" error: unable to create transaction", status);
test_complete = -130;
return PJ_TRUE;
}
pjsip_tsx_recv_msg(tsx, rdata);
save_key(tsx);
if (pj_stricmp2(&branch_param, TEST4_BRANCH_ID) == 0) {
} else if (pj_stricmp2(&branch_param, TEST5_BRANCH_ID) == 0) {
send_response(rdata, tsx, TEST5_PROVISIONAL_CODE);
} else if (pj_stricmp2(&branch_param, TEST6_BRANCH_ID) == 0) {
PJ_LOG(4,(THIS_FILE, " sending provisional response"));
send_response(rdata, tsx, TEST6_PROVISIONAL_CODE);
PJ_LOG(4,(THIS_FILE, " sending final response"));
send_response(rdata, tsx, TEST6_STATUS_CODE);
}
} else {
/* Verify the response received. */
PJ_LOG(4,(THIS_FILE, " received response number %d", recv_count));
++recv_count;
if (pj_stricmp2(&branch_param, TEST4_BRANCH_ID) == 0) {
PJ_LOG(3,(THIS_FILE, " error: not expecting response!"));
test_complete = -132;
} else if (pj_stricmp2(&branch_param, TEST5_BRANCH_ID) == 0) {
if (rdata->msg_info.msg->line.status.code!=TEST5_PROVISIONAL_CODE) {
PJ_LOG(3,(THIS_FILE, " error: incorrect status code!"));
test_complete = -133;
}
if (recv_count > TEST5_RESPONSE_COUNT) {
PJ_LOG(3,(THIS_FILE, " error: not expecting response!"));
test_complete = -134;
}
} else if (pj_stricmp2(&branch_param, TEST6_BRANCH_ID) == 0) {
int code = rdata->msg_info.msg->line.status.code;
switch (recv_count) {
case 1:
if (code != TEST6_PROVISIONAL_CODE) {
PJ_LOG(3,(THIS_FILE, " error: invalid code!"));
test_complete = -135;
}
break;
case 2:
case 3:
if (code != TEST6_STATUS_CODE) {
PJ_LOG(3,(THIS_FILE, " error: invalid code %d "
"(expecting %d)", code, TEST6_STATUS_CODE));
test_complete = -136;
}
break;
default:
PJ_LOG(3,(THIS_FILE, " error: not expecting response"));
test_complete = -137;
break;
}
}
}
return PJ_TRUE;
} else if (pj_stricmp2(&branch_param, TEST7_BRANCH_ID) == 0 ||
pj_stricmp2(&branch_param, TEST8_BRANCH_ID) == 0)
{
/*
* TEST7_BRANCH_ID and TEST8_BRANCH_ID test the retransmission
* of INVITE final response
*/
if (msg->type == PJSIP_REQUEST_MSG) {
/* On received request, create UAS. */
pjsip_transaction *tsx;
status = pjsip_tsx_create_uas(&tsx_user, rdata, &tsx);
if (status != PJ_SUCCESS) {
app_perror(" error: unable to create transaction", status);
test_complete = -140;
return PJ_TRUE;
}
pjsip_tsx_recv_msg(tsx, rdata);
save_key(tsx);
if (pj_stricmp2(&branch_param, TEST7_BRANCH_ID) == 0) {
send_response(rdata, tsx, TEST7_STATUS_CODE);
} else {
send_response(rdata, tsx, TEST8_STATUS_CODE);
}
} else {
int code;
++recv_count;
if (pj_stricmp2(&branch_param, TEST7_BRANCH_ID) == 0)
code = TEST7_STATUS_CODE;
else
code = TEST8_STATUS_CODE;
if (recv_count==1) {
if (rdata->msg_info.msg->line.status.code != code) {
PJ_LOG(3,(THIS_FILE," error: invalid status code"));
test_complete = -141;
}
recv_last = rdata->pkt_info.timestamp;
} else {
pj_time_val now;
unsigned msec, msec_expected;
now = rdata->pkt_info.timestamp;
PJ_TIME_VAL_SUB(now, recv_last);
msec = now.sec*1000 + now.msec;
msec_expected = (1 << (recv_count-2)) * pjsip_cfg()->tsx.t1;
if (msec_expected > pjsip_cfg()->tsx.t2)
msec_expected = pjsip_cfg()->tsx.t2;
if (DIFF(msec, msec_expected) > MAX_ALLOWED_DIFF) {
PJ_LOG(3,(THIS_FILE,
" error: incorrect retransmission "
"time (%d ms expected, %d ms received",
msec_expected, msec));
test_complete = -142;
}
if (recv_count > 11) {
PJ_LOG(3,(THIS_FILE," error: too many responses (%d)",
recv_count));
test_complete = -143;
}
recv_last = rdata->pkt_info.timestamp;
}
}
return PJ_TRUE;
} else if (pj_stricmp2(&branch_param, TEST9_BRANCH_ID) == 0) {
/*
* TEST9_BRANCH_ID tests that the retransmission of INVITE final
* response should cease when ACK is received. Transaction also MUST
* terminate in T4 seconds.
*/
if (msg->type == PJSIP_REQUEST_MSG) {
/* On received request, create UAS. */
pjsip_transaction *tsx;
status = pjsip_tsx_create_uas(&tsx_user, rdata, &tsx);
if (status != PJ_SUCCESS) {
app_perror(" error: unable to create transaction", status);
test_complete = -150;
return PJ_TRUE;
}
pjsip_tsx_recv_msg(tsx, rdata);
save_key(tsx);
send_response(rdata, tsx, TEST9_STATUS_CODE);
} else {
++recv_count;
if (rdata->msg_info.msg->line.status.code != TEST9_STATUS_CODE) {
PJ_LOG(3,(THIS_FILE," error: invalid status code"));
test_complete = -151;
}
if (recv_count==1) {
recv_last = rdata->pkt_info.timestamp;
} else if (recv_count < 5) {
/* Let UAS retransmit some messages before we send ACK. */
pj_time_val now;
unsigned msec, msec_expected;
now = rdata->pkt_info.timestamp;
PJ_TIME_VAL_SUB(now, recv_last);
msec = now.sec*1000 + now.msec;
msec_expected = (1 << (recv_count-2)) * pjsip_cfg()->tsx.t1;
if (msec_expected > pjsip_cfg()->tsx.t2)
msec_expected = pjsip_cfg()->tsx.t2;
if (DIFF(msec, msec_expected) > MAX_ALLOWED_DIFF) {
PJ_LOG(3,(THIS_FILE,
" error: incorrect retransmission "
"time (%d ms expected, %d ms received",
msec_expected, msec));
test_complete = -152;
}
recv_last = rdata->pkt_info.timestamp;
} else if (recv_count == 5) {
pjsip_tx_data *tdata;
pjsip_sip_uri *uri;
pjsip_via_hdr *via;
status = pjsip_endpt_create_request_from_hdr(
endpt, &pjsip_ack_method,
rdata->msg_info.to->uri,
rdata->msg_info.from,
rdata->msg_info.to,
NULL,
rdata->msg_info.cid,
rdata->msg_info.cseq->cseq,
NULL,
&tdata);
if (status != PJ_SUCCESS) {
app_perror(" error: unable to create ACK", status);
test_complete = -153;
return PJ_TRUE;
}
uri=(pjsip_sip_uri*)pjsip_uri_get_uri(tdata->msg->line.req.uri);
uri->transport_param = pj_str("loop-dgram");
via = (pjsip_via_hdr*) pjsip_msg_find_hdr(tdata->msg, PJSIP_H_VIA, NULL);
via->branch_param = pj_str(TEST9_BRANCH_ID);
status = pjsip_endpt_send_request_stateless(endpt, tdata,
NULL, NULL);
if (status != PJ_SUCCESS) {
app_perror(" error: unable to send ACK", status);
test_complete = -154;
}
} else {
PJ_LOG(3,(THIS_FILE," error: too many responses (%d)",
recv_count));
test_complete = -155;
}
}
return PJ_TRUE;
} else if (pj_stricmp2(&branch_param, TEST10_BRANCH_ID) == 0 ||
pj_stricmp2(&branch_param, TEST11_BRANCH_ID) == 0 ||
pj_stricmp2(&branch_param, TEST12_BRANCH_ID) == 0)
{
int test_num, code1, code2;
if (pj_stricmp2(&branch_param, TEST10_BRANCH_ID) == 0)
test_num=10, code1 = 100, code2 = 0;
else if (pj_stricmp2(&branch_param, TEST11_BRANCH_ID) == 0)
test_num=11, code1 = 100, code2 = 200;
else
test_num=12, code1 = 200, code2 = 0;
if (rdata->msg_info.msg->type == PJSIP_REQUEST_MSG) {
/* On received response, create UAS. */
pjsip_transaction *tsx;
status = pjsip_tsx_create_uas(&tsx_user, rdata, &tsx);
if (status != PJ_SUCCESS) {
app_perror(" error: unable to create transaction", status);
test_complete = -150;
return PJ_TRUE;
}
pjsip_tsx_recv_msg(tsx, rdata);
save_key(tsx);
schedule_send_response(rdata, &tsx_key, code1, 1000);
if (code2)
schedule_send_response(rdata, &tsx_key, code2, 2000);
} else {
}
return PJ_TRUE;
}
return PJ_FALSE;
}
