1 line
6.5 KiB
Python
1 line
6.5 KiB
Python
ZnJvbSBkYXRldGltZSBpbXBvcnQgdGltZWRlbHRhCmZyb20gdW5pdHRlc3QubW9jayBpbXBvcnQgcGF0Y2gKCmZyb20gb2Rvby5maWVsZHMgaW1wb3J0IERhdGV0aW1lCmZyb20gb2Rvby50b29scyBpbXBvcnQgbXV0ZV9sb2dnZXIKCmZyb20gb2Rvby5hZGRvbnMuY2V0bWl4X3Rvd2VyX3NlcnZlci50ZXN0cy5jb21tb24gaW1wb3J0IFRlc3RUb3dlckNvbW1vbgoKCmNsYXNzIFRlc3RUb3dlckNvbW1hbmQoVGVzdFRvd2VyQ29tbW9uKToKICAgICIiIlRlc3Qgc3VpdGUgZm9yIHZlcmlmeWluZyB6b21iaWUgY29tbWFuZCBkZXRlY3Rpb24gYW5kIHJlbGF0ZWQKICAgIHF1ZXVlIGpvYiBjYW5jZWxsYXRpb24uCgogICAgVGVzdHMgaW4gdGhpcyBjbGFzcyB2ZXJpZnkgdGhhdCBjb21tYW5kcyB3aGljaCBoYXZlIGJlZW4gcnVubmluZwogICAgbG9uZ2VyIHRoYW4gdGhlIHRpbWVvdXQgYXJlIHByb3Blcmx5IGRldGVjdGVkIGFzIHpvbWJpZXMsIGFuZCB0aGVpcgogICAgYXNzb2NpYXRlZCBxdWV1ZSBqb2JzIGFyZSBjYW5jZWxsZWQuCiAgICAiIiIKCiAgICBAY2xhc3NtZXRob2QKICAgIGRlZiBzZXRVcENsYXNzKGNscyk6CiAgICAgICAgc3VwZXIoKS5zZXRVcENsYXNzKCkKICAgICAgICAjIFNldCBjb21tYW5kIHRpbWVvdXQgdG8gMTAgc2Vjb25kcwogICAgICAgIGNscy5lbnZbImlyLmNvbmZpZ19wYXJhbWV0ZXIiXS5zdWRvKCkuc2V0X3BhcmFtKAogICAgICAgICAgICAiY2V0bWl4X3Rvd2VyX3NlcnZlci5jb21tYW5kX3RpbWVvdXQiLCAiMTAiCiAgICAgICAgKQogICAgICAgICMgU2V0IG9sZCB0aW1lIHRvIDIwIHNlY29uZHMgYWdvIChvbGRlciB0aGFuIHRpbWVvdXQpCiAgICAgICAgIyB0byBzaW11bGF0ZSBydW5uaW5nIGNvbW1hbmQgaW4gcGFzdAogICAgICAgIG5vdyA9IERhdGV0aW1lLm5vdygpCiAgICAgICAgY2xzLm9sZF90aW1lID0gbm93IC0gdGltZWRlbHRhKHNlY29uZHM9MjApCgogICAgZGVmIF9wYXRjaF9jb21tYW5kX3J1bm5lcihzZWxmLCBjb21tYW5kX3R5cGUsIHJ1bm5lcl9tZXRob2QpOgogICAgICAgICIiIkhlbHBlciB0byBwYXRjaCBhIGNvbW1hbmQgcnVubmVyIHRvIHNpbXVsYXRlIGEgem9tYmllIGNvbW1hbmQuCgogICAgICAgIEFyZ3M6CiAgICAgICAgICAgIGNvbW1hbmRfdHlwZTogVHlwZSBvZiBjb21tYW5kIHJ1bm5lciB0byBwYXRjaCAoJ3NzaCcgb3IgJ3B5dGhvbl9jb2RlJykKICAgICAgICAgICAgcnVubmVyX21ldGhvZDogT3JpZ2luYWwgbWV0aG9kIHRvIHdyYXAKCiAgICAgICAgUmV0dXJuczoKICAgICAgICAgICAgQSBjb250ZXh0IG1hbmFnZXIgdGhhdCBhcHBsaWVzIHRoZSBwYXRjaAogICAgICAgICIiIgoKICAgICAgICBkZWYgX3dyYXBwZXIoKmFyZ3MsICoqa3dhcmdzKToKICAgICAgICAgICAgIyBNb2RpZnkgYXJncyB0byBkaXNhYmxlIGxvZyByZWNvcmQgZmluaXNoaW5nCiAgICAgICAgICAgIGFyZ3MgPSBsaXN0KGFyZ3MpCiAgICAgICAgICAgIGlmIGxlbihhcmdzKSA+IDE6CiAgICAgICAgICAgICAgICBhcmdzWzFdID0gRmFsc2UgICMgU2V0IGxvZ19yZWNvcmQgdG8gRmFsc2UKICAgICAgICAgICAgcmV0dXJuIHJ1bm5lcl9tZXRob2QoKmFyZ3MsICoqa3dhcmdzKQoKICAgICAgICByZXR1cm4gcGF0Y2gub2JqZWN0KAogICAgICAgICAgICBzZWxmLnJlZ2lzdHJ5WyJjeC50b3dlci5zZXJ2ZXIiXSwKICAgICAgICAgICAgZiJfY29tbWFuZF9ydW5uZXJfe2NvbW1hbmRfdHlwZX0iLAogICAgICAgICAgICBfd3JhcHBlciwKICAgICAgICApCgogICAgZGVmIF92ZXJpZnlfem9tYmllX2NvbW1hbmRfam9iX2NhbmNlbGxhdGlvbihzZWxmLCBjb21tYW5kX2FjdGlvbik6CiAgICAgICAgIiIiVmVyaWZ5IHpvbWJpZSBjb21tYW5kIGlzIGRldGVjdGVkIGFuZCBqb2IgaXMgY2FuY2VsbGVkLgoKICAgICAgICBBcmdzOgogICAgICAgICAgICBjb21tYW5kX2FjdGlvbjogQWN0aW9uIHR5cGUgKCdzc2hfY29tbWFuZCcgb3IgJ3B5dGhvbl9jb2RlJykKICAgICAgICAiIiIKICAgICAgICAjIGNoZWNrIHpvbWJpZSBjb21tYW5kIGxvZ3MKICAgICAgICBkb21haW4gPSBbCiAgICAgICAgICAgICgiaXNfcnVubmluZyIsICI9IiwgVHJ1ZSksCiAgICAgICAgICAgICgic3RhcnRfZGF0ZSIsICI9Iiwgc2VsZi5vbGRfdGltZSksCiAgICAgICAgICAgICgiY29tbWFuZF9hY3Rpb24iLCAiPSIsIGNvbW1hbmRfYWN0aW9uKSwKICAgICAgICBdCiAgICAgICAgem9tYmllX2NvbW1hbmRfbG9ncyA9IHNlbGYuZW52WyJjeC50b3dlci5jb21tYW5kLmxvZyJdLnNlYXJjaChkb21haW4pCgogICAgICAgIHNlbGYuYXNzZXJ0RXF1YWwoCiAgICAgICAgICAgIGxlbih6b21iaWVfY29tbWFuZF9sb2dzKSwgMSwgIlpvbWJpZSBjb21tYW5kIGxvZyBzaG91bGQgYmUgY3JlYXRlZCIKICAgICAgICApCiAgICAgICAgc2VsZi5hc3NlcnRUcnVlKAogICAgICAgICAgICB6b21iaWVfY29tbWFuZF9sb2dzLnF1ZXVlX2pvYl9pZCwKICAgICAgICAgICAgIlpvbWJpZSBjb21tYW5kIGxvZyBzaG91bGQgaGF2ZSBxdWV1ZSBqb2IiLAogICAgICAgICkKCiAgICAgICAgam9iID0gem9tYmllX2NvbW1hbmRfbG9ncy5xdWV1ZV9qb2JfaWQKICAgICAgICBzZWxmLmFzc2VydFRydWUoam9iLmV4aXN0cygpLCAiWm9tYmllIGNvbW1hbmQgam9iIHNob3VsZCBleGlzdCIpCgogICAgICAgIHNlbGYuYXNzZXJ0RXF1YWwoam9iLnN0YXRlLCAicGVuZGluZyIsICJab21iaWUgY29tbWFuZCBqb2Igc2hvdWxkIGJlIHBlbmRpbmciKQoKICAgICAgICAjIHJ1biBwcm9jZXNzIHRvIGtpbGwgem9tYmllIGNvbW1hbmQKICAgICAgICBzZWxmLnNlcnZlcl90ZXN0XzEuX2NoZWNrX3pvbWJpZV9jb21tYW5kcygpCgogICAgICAgICMgY2hlY2sgdGhhdCBjb21tYW5kIGxvZyBpcyBjYW5jZWxsZWQKICAgICAgICBzZWxmLmFzc2VydEVxdWFsKAogICAgICAgICAgICBqb2Iuc3RhdGUsICJjYW5jZWxsZWQiLCAiWm9tYmllIGNvbW1hbmQgam9iIHNob3VsZCBiZSBjYW5jZWxsZWQiCiAgICAgICAgKQoKICAgIGRlZiB0ZXN0X2NoZWNrX3pvbWJpZV9zc2hfY29tbWFuZF9xdWV1ZShzZWxmKToKICAgICAgICAiIiIKICAgICAgICBUZXN0IHRoYXQgem9tYmllIHNzaCBjb21tYW5kIGlzIGtpbGxlZCBhbmQgam9iIGlzIGNhbmNlbGxlZAogICAgICAgICIiIgogICAgICAgICMgQ3JlYXRlIHRlc3QgY29tbWFuZHMKICAgICAgICBzc2hfY29tbWFuZCA9IHNlbGYuQ29tbWFuZC5jcmVhdGUoCiAgICAgICAgICAgIHsKICAgICAgICAgICAgICAgICJuYW1lIjogIlRlc3QgU1NIIENvbW1hbmQiLAogICAgICAgICAgICAgICAgImNvZGUiOiAibHMgLWxhIiwKICAgICAgICAgICAgICAgICJhY3Rpb24iOiAic3NoX2NvbW1hbmQiLAogICAgICAgICAgICB9CiAgICAgICAgKQoKICAgICAgICAjIHBhdGNoIGNvbW1hbmQgcnVubmVyIHRvIG5vdCBmaW5pc2ggbG9nIHJlY29yZAogICAgICAgIGN4X3Rvd2VyX3NlcnZlcl9vYmogPSBzZWxmLnJlZ2lzdHJ5WyJjeC50b3dlci5zZXJ2ZXIiXQogICAgICAgIF9jb21tYW5kX3J1bm5lcl9zc2hfc3VwZXIgPSBjeF90b3dlcl9zZXJ2ZXJfb2JqLl9jb21tYW5kX3J1bm5lcl9zc2gKCiAgICAgICAgd2l0aCBzZWxmLl9wYXRjaF9jb21tYW5kX3J1bm5lcigic3NoIiwgX2NvbW1hbmRfcnVubmVyX3NzaF9zdXBlcik6CiAgICAgICAgICAgICMgcnVuIHpvbWJpZSBjb21tYW5kIHdpdGggbG9nIGNyZWF0aW9uIGluIHBhc3QKICAgICAgICAgICAgc2VsZi5zZXJ2ZXJfdGVzdF8xLnJ1bl9jb21tYW5kKAogICAgICAgICAgICAgICAgc3NoX2NvbW1hbmQsIGxvZz17InN0YXJ0X2RhdGUiOiBzZWxmLm9sZF90aW1lfQogICAgICAgICAgICApCgogICAgICAgICMgY2hlY2sgem9tYmllIGNvbW1hbmQgbG9ncwogICAgICAgIHNlbGYuX3ZlcmlmeV96b21iaWVfY29tbWFuZF9qb2JfY2FuY2VsbGF0aW9uKCJzc2hfY29tbWFuZCIpCgogICAgQG11dGVfbG9nZ2VyKCJweS53YXJuaW5ncyIpCiAgICBkZWYgdGVzdF9jaGVja196b21iaWVfcHl0aG9uX2NvbW1hbmRfcXVldWUoc2VsZik6CiAgICAgICAgIiIiCiAgICAgICAgVGVzdCB0aGF0IHpvbWJpZSBweXRob24gY29tbWFuZCBpcyBraWxsZWQgYW5kIGpvYiBpcyBjYW5jZWxsZWQKICAgICAgICAiIiIKICAgICAgICAjIENyZWF0ZSB0ZXN0IGNvbW1hbmRzCiAgICAgICAgcHl0aG9uX2NvbW1hbmQgPSBzZWxmLkNvbW1hbmQuY3JlYXRlKAogICAgICAgICAgICB7CiAgICAgICAgICAgICAgICAibmFtZSI6ICJUZXN0IFB5dGhvbiBDb21tYW5kIiwKICAgICAgICAgICAgICAgICJjb2RlIjogInByaW50KCd0ZXN0JykiLAogICAgICAgICAgICAgICAgImFjdGlvbiI6ICJweXRob25fY29kZSIsCiAgICAgICAgICAgIH0KICAgICAgICApCgogICAgICAgICMgcGF0Y2ggY29tbWFuZCBydW5uZXIgdG8gbm90IGZpbmlzaCBsb2cgcmVjb3JkCiAgICAgICAgY3hfdG93ZXJfc2VydmVyX29iaiA9IHNlbGYucmVnaXN0cnlbImN4LnRvd2VyLnNlcnZlciJdCiAgICAgICAgX2NvbW1hbmRfcnVubmVyX3B5dGhvbl9jb2RlX3N1cGVyID0gKAogICAgICAgICAgICBjeF90b3dlcl9zZXJ2ZXJfb2JqLl9jb21tYW5kX3J1bm5lcl9weXRob25fY29kZQogICAgICAgICkKCiAgICAgICAgd2l0aCBzZWxmLl9wYXRjaF9jb21tYW5kX3J1bm5lcigKICAgICAgICAgICAgInB5dGhvbl9jb2RlIiwgX2NvbW1hbmRfcnVubmVyX3B5dGhvbl9jb2RlX3N1cGVyCiAgICAgICAgKToKICAgICAgICAgICAgIyBydW4gem9tYmllIGNvbW1hbmQgd2l0aCBsb2cgY3JlYXRpb24gaW4gcGFzdAogICAgICAgICAgICBzZWxmLnNlcnZlcl90ZXN0XzEucnVuX2NvbW1hbmQoCiAgICAgICAgICAgICAgICBweXRob25fY29tbWFuZCwgbG9nPXsic3RhcnRfZGF0ZSI6IHNlbGYub2xkX3RpbWV9CiAgICAgICAgICAgICkKCiAgICAgICAgIyBjaGVjayB6b21iaWUgY29tbWFuZCBsb2dzCiAgICAgICAgc2VsZi5fdmVyaWZ5X3pvbWJpZV9jb21tYW5kX2pvYl9jYW5jZWxsYXRpb24oInB5dGhvbl9jb2RlIikK |