pikvm/pikvm#1041: Randomized recorder events

This commit is contained in:
Maxim Devaev 2023-07-07 14:03:44 +03:00
parent 26d807a4b9
commit 64728dc0e6
2 changed files with 59 additions and 6 deletions

View File

@ -154,37 +154,57 @@ export function Recorder() {
__checkType(event.event, "object", "Non-dict event"); __checkType(event.event, "object", "Non-dict event");
if (event.event_type === "delay") { if (event.event_type === "delay") {
__checkInt(event.event.millis, "Non-integer delay"); __checkUnsigned(event.event.millis, "Non-unsigned delay");
if (event.event.millis < 0) {
throw "Negative delay";
}
events_time += event.event.millis; events_time += event.event.millis;
} else if (event.event_type === "print") { } else if (event.event_type === "print") {
__checkType(event.event.text, "string", "Non-string print text"); __checkType(event.event.text, "string", "Non-string print text");
} else if (event.event_type === "key") { } else if (event.event_type === "key") {
__checkType(event.event.key, "string", "Non-string key code"); __checkType(event.event.key, "string", "Non-string key code");
__checkType(event.event.state, "boolean", "Non-bool key state"); __checkType(event.event.state, "boolean", "Non-bool key state");
} else if (event.event_type === "mouse_button") { } else if (event.event_type === "mouse_button") {
__checkType(event.event.button, "string", "Non-string mouse button code"); __checkType(event.event.button, "string", "Non-string mouse button code");
__checkType(event.event.state, "boolean", "Non-bool mouse button state"); __checkType(event.event.state, "boolean", "Non-bool mouse button state");
} else if (event.event_type === "mouse_move") { } else if (event.event_type === "mouse_move") {
__checkType(event.event.to, "object", "Non-object mouse move target"); __checkType(event.event.to, "object", "Non-object mouse move target");
__checkInt(event.event.to.x, "Non-int mouse move X"); __checkInt(event.event.to.x, "Non-int mouse move X");
__checkInt(event.event.to.y, "Non-int mouse move Y"); __checkInt(event.event.to.y, "Non-int mouse move Y");
} else if (event.event_type === "mouse_relative") { } else if (event.event_type === "mouse_relative") {
__checkMouseRelativeDelta(event.event.delta); __checkMouseRelativeDelta(event.event.delta);
__checkType(event.event.squash, "boolean", "Non-boolean squash"); __checkType(event.event.squash, "boolean", "Non-boolean squash");
} else if (event.event_type === "mouse_wheel") { } else if (event.event_type === "mouse_wheel") {
__checkType(event.event.delta, "object", "Non-object mouse wheel delta"); __checkType(event.event.delta, "object", "Non-object mouse wheel delta");
__checkInt(event.event.delta.x, "Non-int mouse delta X"); __checkInt(event.event.delta.x, "Non-int mouse delta X");
__checkInt(event.event.delta.y, "Non-int mouse delta Y"); __checkInt(event.event.delta.y, "Non-int mouse delta Y");
} else if (event.event_type === "atx_button") { } else if (event.event_type === "atx_button") {
__checkType(event.event.button, "string", "Non-string ATX button"); __checkType(event.event.button, "string", "Non-string ATX button");
} else if (event.event_type === "gpio_switch") { } else if (event.event_type === "gpio_switch") {
__checkType(event.event.channel, "string", "Non-string GPIO channel"); __checkType(event.event.channel, "string", "Non-string GPIO channel");
__checkType(event.event.state, "boolean", "Non-bool GPIO state"); __checkType(event.event.state, "boolean", "Non-bool GPIO state");
} else if (event.event_type === "gpio_pulse") { } else if (event.event_type === "gpio_pulse") {
__checkType(event.event.channel, "string", "Non-string GPIO channel"); __checkType(event.event.channel, "string", "Non-string GPIO channel");
} else if (event.event_type === "delay_random") {
__checkType(event.event.range, "object", "Non-object random delay range");
__checkUnsigned(event.event.range.min, "Non-unsigned random delay range min");
__checkUnsigned(event.event.range.max, "Non-unsigned random delay range max");
__checkRangeMinMax(event.event.range, "Invalid random delay range");
events_time += event.event.range.max;
} else if (event.event_type === "mouse_move_random") { // Hack for pikvm/pikvm#1041
__checkType(event.event.range, "object", "Non-object random mouse move range");
__checkInt(event.event.range.min, "Non-int random mouse move range min");
__checkInt(event.event.range.max, "Non-int random mouse move range max");
__checkRangeMinMax(event.event.range, "Invalid random mouse move range");
} else { } else {
throw `Unknown event type: ${event.event_type}`; throw `Unknown event type: ${event.event_type}`;
} }
@ -217,6 +237,19 @@ export function Recorder() {
} }
}; };
var __checkUnsigned = function(obj, msg) {
__checkInt(obj, msg);
if (obj < 0) {
throw msg;
}
};
var __checkRangeMinMax = function(obj, msg) {
if (obj.min > obj.max) {
throw msg;
}
};
var __checkArray = function (obj, msg) { var __checkArray = function (obj, msg) {
if (!Array.isArray(obj)) { if (!Array.isArray(obj)) {
throw msg; throw msg;
@ -237,8 +270,13 @@ export function Recorder() {
__setCounters(__events.length - index + 1, __events_time - time); __setCounters(__events.length - index + 1, __events_time - time);
let event = __events[index]; let event = __events[index];
if (event.event_type === "delay") { if (["delay", "delay_random"].includes(event.event_type)) {
__play_timer = setTimeout(() => __runEvents(index + 1, time + event.event.millis), event.event.millis); let millis = (
event.event_type === "delay"
? event.event.millis
: tools.getRandomInt(event.event.range.min, event.event.range.max)
);
__play_timer = setTimeout(() => __runEvents(index + 1, time + millis), millis);
return; return;
} else if (event.event_type === "print") { } else if (event.event_type === "print") {
@ -291,6 +329,15 @@ export function Recorder() {
} else if (["key", "mouse_button", "mouse_move", "mouse_wheel", "mouse_relative"].includes(event.event_type)) { } else if (["key", "mouse_button", "mouse_move", "mouse_wheel", "mouse_relative"].includes(event.event_type)) {
__ws.sendHidEvent(event); __ws.sendHidEvent(event);
} else if (event.event_type === "mouse_move_random") {
__ws.sendHidEvent({
"event_type": "mouse_move",
"event": {"to": {
"x": tools.getRandomInt(event.event.range.min, event.event.range.max),
"y": tools.getRandomInt(event.event.range.min, event.event.range.max),
}},
});
} }
index += 1; index += 1;

View File

@ -70,6 +70,12 @@ export var tools = new function() {
return btoa(text).replace("=", "_"); return btoa(text).replace("=", "_");
}; };
self.getRandomInt = function(min, max) {
min = Math.ceil(min);
max = Math.floor(max);
return Math.floor(Math.random() * (max - min + 1)) + min;
};
self.formatSize = function(size) { self.formatSize = function(size) {
if (size > 0) { if (size > 0) {
let index = Math.floor( Math.log(size) / Math.log(1024) ); let index = Math.floor( Math.log(size) / Math.log(1024) );