i8254.c 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760
  1. /*
  2. * 8253/8254 interval timer emulation
  3. *
  4. * Copyright (c) 2003-2004 Fabrice Bellard
  5. * Copyright (c) 2006 Intel Corporation
  6. * Copyright (c) 2007 Keir Fraser, XenSource Inc
  7. * Copyright (c) 2008 Intel Corporation
  8. * Copyright 2009 Red Hat, Inc. and/or its affiliates.
  9. *
  10. * Permission is hereby granted, free of charge, to any person obtaining a copy
  11. * of this software and associated documentation files (the "Software"), to deal
  12. * in the Software without restriction, including without limitation the rights
  13. * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  14. * copies of the Software, and to permit persons to whom the Software is
  15. * furnished to do so, subject to the following conditions:
  16. *
  17. * The above copyright notice and this permission notice shall be included in
  18. * all copies or substantial portions of the Software.
  19. *
  20. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  21. * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  22. * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
  23. * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  24. * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  25. * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  26. * THE SOFTWARE.
  27. *
  28. * Authors:
  29. * Sheng Yang <sheng.yang@intel.com>
  30. * Based on QEMU and Xen.
  31. */
  32. #define pr_fmt(fmt) "pit: " fmt
  33. #include <linux/kvm_host.h>
  34. #include <linux/slab.h>
  35. #include "irq.h"
  36. #include "i8254.h"
  37. #ifndef CONFIG_X86_64
  38. #define mod_64(x, y) ((x) - (y) * div64_u64(x, y))
  39. #else
  40. #define mod_64(x, y) ((x) % (y))
  41. #endif
  42. #define RW_STATE_LSB 1
  43. #define RW_STATE_MSB 2
  44. #define RW_STATE_WORD0 3
  45. #define RW_STATE_WORD1 4
  46. /* Compute with 96 bit intermediate result: (a*b)/c */
  47. static u64 muldiv64(u64 a, u32 b, u32 c)
  48. {
  49. union {
  50. u64 ll;
  51. struct {
  52. u32 low, high;
  53. } l;
  54. } u, res;
  55. u64 rl, rh;
  56. u.ll = a;
  57. rl = (u64)u.l.low * (u64)b;
  58. rh = (u64)u.l.high * (u64)b;
  59. rh += (rl >> 32);
  60. res.l.high = div64_u64(rh, c);
  61. res.l.low = div64_u64(((mod_64(rh, c) << 32) + (rl & 0xffffffff)), c);
  62. return res.ll;
  63. }
  64. static void pit_set_gate(struct kvm *kvm, int channel, u32 val)
  65. {
  66. struct kvm_kpit_channel_state *c =
  67. &kvm->arch.vpit->pit_state.channels[channel];
  68. WARN_ON(!mutex_is_locked(&kvm->arch.vpit->pit_state.lock));
  69. switch (c->mode) {
  70. default:
  71. case 0:
  72. case 4:
  73. /* XXX: just disable/enable counting */
  74. break;
  75. case 1:
  76. case 2:
  77. case 3:
  78. case 5:
  79. /* Restart counting on rising edge. */
  80. if (c->gate < val)
  81. c->count_load_time = ktime_get();
  82. break;
  83. }
  84. c->gate = val;
  85. }
  86. static int pit_get_gate(struct kvm *kvm, int channel)
  87. {
  88. WARN_ON(!mutex_is_locked(&kvm->arch.vpit->pit_state.lock));
  89. return kvm->arch.vpit->pit_state.channels[channel].gate;
  90. }
  91. static s64 __kpit_elapsed(struct kvm *kvm)
  92. {
  93. s64 elapsed;
  94. ktime_t remaining;
  95. struct kvm_kpit_state *ps = &kvm->arch.vpit->pit_state;
  96. if (!ps->pit_timer.period)
  97. return 0;
  98. /*
  99. * The Counter does not stop when it reaches zero. In
  100. * Modes 0, 1, 4, and 5 the Counter ``wraps around'' to
  101. * the highest count, either FFFF hex for binary counting
  102. * or 9999 for BCD counting, and continues counting.
  103. * Modes 2 and 3 are periodic; the Counter reloads
  104. * itself with the initial count and continues counting
  105. * from there.
  106. */
  107. remaining = hrtimer_get_remaining(&ps->pit_timer.timer);
  108. elapsed = ps->pit_timer.period - ktime_to_ns(remaining);
  109. elapsed = mod_64(elapsed, ps->pit_timer.period);
  110. return elapsed;
  111. }
  112. static s64 kpit_elapsed(struct kvm *kvm, struct kvm_kpit_channel_state *c,
  113. int channel)
  114. {
  115. if (channel == 0)
  116. return __kpit_elapsed(kvm);
  117. return ktime_to_ns(ktime_sub(ktime_get(), c->count_load_time));
  118. }
  119. static int pit_get_count(struct kvm *kvm, int channel)
  120. {
  121. struct kvm_kpit_channel_state *c =
  122. &kvm->arch.vpit->pit_state.channels[channel];
  123. s64 d, t;
  124. int counter;
  125. WARN_ON(!mutex_is_locked(&kvm->arch.vpit->pit_state.lock));
  126. t = kpit_elapsed(kvm, c, channel);
  127. d = muldiv64(t, KVM_PIT_FREQ, NSEC_PER_SEC);
  128. switch (c->mode) {
  129. case 0:
  130. case 1:
  131. case 4:
  132. case 5:
  133. counter = (c->count - d) & 0xffff;
  134. break;
  135. case 3:
  136. /* XXX: may be incorrect for odd counts */
  137. counter = c->count - (mod_64((2 * d), c->count));
  138. break;
  139. default:
  140. counter = c->count - mod_64(d, c->count);
  141. break;
  142. }
  143. return counter;
  144. }
  145. static int pit_get_out(struct kvm *kvm, int channel)
  146. {
  147. struct kvm_kpit_channel_state *c =
  148. &kvm->arch.vpit->pit_state.channels[channel];
  149. s64 d, t;
  150. int out;
  151. WARN_ON(!mutex_is_locked(&kvm->arch.vpit->pit_state.lock));
  152. t = kpit_elapsed(kvm, c, channel);
  153. d = muldiv64(t, KVM_PIT_FREQ, NSEC_PER_SEC);
  154. switch (c->mode) {
  155. default:
  156. case 0:
  157. out = (d >= c->count);
  158. break;
  159. case 1:
  160. out = (d < c->count);
  161. break;
  162. case 2:
  163. out = ((mod_64(d, c->count) == 0) && (d != 0));
  164. break;
  165. case 3:
  166. out = (mod_64(d, c->count) < ((c->count + 1) >> 1));
  167. break;
  168. case 4:
  169. case 5:
  170. out = (d == c->count);
  171. break;
  172. }
  173. return out;
  174. }
  175. static void pit_latch_count(struct kvm *kvm, int channel)
  176. {
  177. struct kvm_kpit_channel_state *c =
  178. &kvm->arch.vpit->pit_state.channels[channel];
  179. WARN_ON(!mutex_is_locked(&kvm->arch.vpit->pit_state.lock));
  180. if (!c->count_latched) {
  181. c->latched_count = pit_get_count(kvm, channel);
  182. c->count_latched = c->rw_mode;
  183. }
  184. }
  185. static void pit_latch_status(struct kvm *kvm, int channel)
  186. {
  187. struct kvm_kpit_channel_state *c =
  188. &kvm->arch.vpit->pit_state.channels[channel];
  189. WARN_ON(!mutex_is_locked(&kvm->arch.vpit->pit_state.lock));
  190. if (!c->status_latched) {
  191. /* TODO: Return NULL COUNT (bit 6). */
  192. c->status = ((pit_get_out(kvm, channel) << 7) |
  193. (c->rw_mode << 4) |
  194. (c->mode << 1) |
  195. c->bcd);
  196. c->status_latched = 1;
  197. }
  198. }
  199. static void kvm_pit_ack_irq(struct kvm_irq_ack_notifier *kian)
  200. {
  201. struct kvm_kpit_state *ps = container_of(kian, struct kvm_kpit_state,
  202. irq_ack_notifier);
  203. int value;
  204. spin_lock(&ps->inject_lock);
  205. value = atomic_dec_return(&ps->pit_timer.pending);
  206. if (value < 0)
  207. /* spurious acks can be generated if, for example, the
  208. * PIC is being reset. Handle it gracefully here
  209. */
  210. atomic_inc(&ps->pit_timer.pending);
  211. else if (value > 0)
  212. /* in this case, we had multiple outstanding pit interrupts
  213. * that we needed to inject. Reinject
  214. */
  215. queue_kthread_work(&ps->pit->worker, &ps->pit->expired);
  216. ps->irq_ack = 1;
  217. spin_unlock(&ps->inject_lock);
  218. }
  219. void __kvm_migrate_pit_timer(struct kvm_vcpu *vcpu)
  220. {
  221. struct kvm_pit *pit = vcpu->kvm->arch.vpit;
  222. struct hrtimer *timer;
  223. if (!kvm_vcpu_is_bsp(vcpu) || !pit)
  224. return;
  225. timer = &pit->pit_state.pit_timer.timer;
  226. if (hrtimer_cancel(timer))
  227. hrtimer_start_expires(timer, HRTIMER_MODE_ABS);
  228. }
  229. static void destroy_pit_timer(struct kvm_pit *pit)
  230. {
  231. hrtimer_cancel(&pit->pit_state.pit_timer.timer);
  232. flush_kthread_work(&pit->expired);
  233. }
  234. static void pit_do_work(struct kthread_work *work)
  235. {
  236. struct kvm_pit *pit = container_of(work, struct kvm_pit, expired);
  237. struct kvm *kvm = pit->kvm;
  238. struct kvm_vcpu *vcpu;
  239. int i;
  240. struct kvm_kpit_state *ps = &pit->pit_state;
  241. int inject = 0;
  242. /* Try to inject pending interrupts when
  243. * last one has been acked.
  244. */
  245. spin_lock(&ps->inject_lock);
  246. if (ps->irq_ack) {
  247. ps->irq_ack = 0;
  248. inject = 1;
  249. }
  250. spin_unlock(&ps->inject_lock);
  251. if (inject) {
  252. kvm_set_irq(kvm, kvm->arch.vpit->irq_source_id, 0, 1);
  253. kvm_set_irq(kvm, kvm->arch.vpit->irq_source_id, 0, 0);
  254. /*
  255. * Provides NMI watchdog support via Virtual Wire mode.
  256. * The route is: PIT -> PIC -> LVT0 in NMI mode.
  257. *
  258. * Note: Our Virtual Wire implementation is simplified, only
  259. * propagating PIT interrupts to all VCPUs when they have set
  260. * LVT0 to NMI delivery. Other PIC interrupts are just sent to
  261. * VCPU0, and only if its LVT0 is in EXTINT mode.
  262. */
  263. if (kvm->arch.vapics_in_nmi_mode > 0)
  264. kvm_for_each_vcpu(i, vcpu, kvm)
  265. kvm_apic_nmi_wd_deliver(vcpu);
  266. }
  267. }
  268. static enum hrtimer_restart pit_timer_fn(struct hrtimer *data)
  269. {
  270. struct kvm_pit_timer *ktimer = container_of(data, struct kvm_pit_timer, timer);
  271. struct kvm_pit *pt = ktimer->kvm->arch.vpit;
  272. if (ktimer->reinject || !atomic_read(&ktimer->pending)) {
  273. atomic_inc(&ktimer->pending);
  274. queue_kthread_work(&pt->worker, &pt->expired);
  275. }
  276. if (pt->pit_state.is_periodic) {
  277. hrtimer_add_expires_ns(&ktimer->timer, ktimer->period);
  278. return HRTIMER_RESTART;
  279. } else
  280. return HRTIMER_NORESTART;
  281. }
  282. static void create_pit_timer(struct kvm *kvm, u32 val, int is_period)
  283. {
  284. struct kvm_kpit_state *ps = &kvm->arch.vpit->pit_state;
  285. struct kvm_pit_timer *pt = &ps->pit_timer;
  286. s64 interval;
  287. if (!irqchip_in_kernel(kvm) || ps->flags & KVM_PIT_FLAGS_HPET_LEGACY)
  288. return;
  289. interval = muldiv64(val, NSEC_PER_SEC, KVM_PIT_FREQ);
  290. pr_debug("create pit timer, interval is %llu nsec\n", interval);
  291. /* TODO The new value only affected after the retriggered */
  292. hrtimer_cancel(&pt->timer);
  293. flush_kthread_work(&ps->pit->expired);
  294. pt->period = interval;
  295. ps->is_periodic = is_period;
  296. pt->timer.function = pit_timer_fn;
  297. pt->kvm = ps->pit->kvm;
  298. atomic_set(&pt->pending, 0);
  299. ps->irq_ack = 1;
  300. hrtimer_start(&pt->timer, ktime_add_ns(ktime_get(), interval),
  301. HRTIMER_MODE_ABS);
  302. }
  303. static void pit_load_count(struct kvm *kvm, int channel, u32 val)
  304. {
  305. struct kvm_kpit_state *ps = &kvm->arch.vpit->pit_state;
  306. WARN_ON(!mutex_is_locked(&ps->lock));
  307. pr_debug("load_count val is %d, channel is %d\n", val, channel);
  308. /*
  309. * The largest possible initial count is 0; this is equivalent
  310. * to 216 for binary counting and 104 for BCD counting.
  311. */
  312. if (val == 0)
  313. val = 0x10000;
  314. ps->channels[channel].count = val;
  315. if (channel != 0) {
  316. ps->channels[channel].count_load_time = ktime_get();
  317. return;
  318. }
  319. /* Two types of timer
  320. * mode 1 is one shot, mode 2 is period, otherwise del timer */
  321. switch (ps->channels[0].mode) {
  322. case 0:
  323. case 1:
  324. /* FIXME: enhance mode 4 precision */
  325. case 4:
  326. create_pit_timer(kvm, val, 0);
  327. break;
  328. case 2:
  329. case 3:
  330. create_pit_timer(kvm, val, 1);
  331. break;
  332. default:
  333. destroy_pit_timer(kvm->arch.vpit);
  334. }
  335. }
  336. void kvm_pit_load_count(struct kvm *kvm, int channel, u32 val, int hpet_legacy_start)
  337. {
  338. u8 saved_mode;
  339. if (hpet_legacy_start) {
  340. /* save existing mode for later reenablement */
  341. saved_mode = kvm->arch.vpit->pit_state.channels[0].mode;
  342. kvm->arch.vpit->pit_state.channels[0].mode = 0xff; /* disable timer */
  343. pit_load_count(kvm, channel, val);
  344. kvm->arch.vpit->pit_state.channels[0].mode = saved_mode;
  345. } else {
  346. pit_load_count(kvm, channel, val);
  347. }
  348. }
  349. static inline struct kvm_pit *dev_to_pit(struct kvm_io_device *dev)
  350. {
  351. return container_of(dev, struct kvm_pit, dev);
  352. }
  353. static inline struct kvm_pit *speaker_to_pit(struct kvm_io_device *dev)
  354. {
  355. return container_of(dev, struct kvm_pit, speaker_dev);
  356. }
  357. static inline int pit_in_range(gpa_t addr)
  358. {
  359. return ((addr >= KVM_PIT_BASE_ADDRESS) &&
  360. (addr < KVM_PIT_BASE_ADDRESS + KVM_PIT_MEM_LENGTH));
  361. }
  362. static int pit_ioport_write(struct kvm_io_device *this,
  363. gpa_t addr, int len, const void *data)
  364. {
  365. struct kvm_pit *pit = dev_to_pit(this);
  366. struct kvm_kpit_state *pit_state = &pit->pit_state;
  367. struct kvm *kvm = pit->kvm;
  368. int channel, access;
  369. struct kvm_kpit_channel_state *s;
  370. u32 val = *(u32 *) data;
  371. if (!pit_in_range(addr))
  372. return -EOPNOTSUPP;
  373. val &= 0xff;
  374. addr &= KVM_PIT_CHANNEL_MASK;
  375. mutex_lock(&pit_state->lock);
  376. if (val != 0)
  377. pr_debug("write addr is 0x%x, len is %d, val is 0x%x\n",
  378. (unsigned int)addr, len, val);
  379. if (addr == 3) {
  380. channel = val >> 6;
  381. if (channel == 3) {
  382. /* Read-Back Command. */
  383. for (channel = 0; channel < 3; channel++) {
  384. s = &pit_state->channels[channel];
  385. if (val & (2 << channel)) {
  386. if (!(val & 0x20))
  387. pit_latch_count(kvm, channel);
  388. if (!(val & 0x10))
  389. pit_latch_status(kvm, channel);
  390. }
  391. }
  392. } else {
  393. /* Select Counter <channel>. */
  394. s = &pit_state->channels[channel];
  395. access = (val >> 4) & KVM_PIT_CHANNEL_MASK;
  396. if (access == 0) {
  397. pit_latch_count(kvm, channel);
  398. } else {
  399. s->rw_mode = access;
  400. s->read_state = access;
  401. s->write_state = access;
  402. s->mode = (val >> 1) & 7;
  403. if (s->mode > 5)
  404. s->mode -= 4;
  405. s->bcd = val & 1;
  406. }
  407. }
  408. } else {
  409. /* Write Count. */
  410. s = &pit_state->channels[addr];
  411. switch (s->write_state) {
  412. default:
  413. case RW_STATE_LSB:
  414. pit_load_count(kvm, addr, val);
  415. break;
  416. case RW_STATE_MSB:
  417. pit_load_count(kvm, addr, val << 8);
  418. break;
  419. case RW_STATE_WORD0:
  420. s->write_latch = val;
  421. s->write_state = RW_STATE_WORD1;
  422. break;
  423. case RW_STATE_WORD1:
  424. pit_load_count(kvm, addr, s->write_latch | (val << 8));
  425. s->write_state = RW_STATE_WORD0;
  426. break;
  427. }
  428. }
  429. mutex_unlock(&pit_state->lock);
  430. return 0;
  431. }
  432. static int pit_ioport_read(struct kvm_io_device *this,
  433. gpa_t addr, int len, void *data)
  434. {
  435. struct kvm_pit *pit = dev_to_pit(this);
  436. struct kvm_kpit_state *pit_state = &pit->pit_state;
  437. struct kvm *kvm = pit->kvm;
  438. int ret, count;
  439. struct kvm_kpit_channel_state *s;
  440. if (!pit_in_range(addr))
  441. return -EOPNOTSUPP;
  442. addr &= KVM_PIT_CHANNEL_MASK;
  443. if (addr == 3)
  444. return 0;
  445. s = &pit_state->channels[addr];
  446. mutex_lock(&pit_state->lock);
  447. if (s->status_latched) {
  448. s->status_latched = 0;
  449. ret = s->status;
  450. } else if (s->count_latched) {
  451. switch (s->count_latched) {
  452. default:
  453. case RW_STATE_LSB:
  454. ret = s->latched_count & 0xff;
  455. s->count_latched = 0;
  456. break;
  457. case RW_STATE_MSB:
  458. ret = s->latched_count >> 8;
  459. s->count_latched = 0;
  460. break;
  461. case RW_STATE_WORD0:
  462. ret = s->latched_count & 0xff;
  463. s->count_latched = RW_STATE_MSB;
  464. break;
  465. }
  466. } else {
  467. switch (s->read_state) {
  468. default:
  469. case RW_STATE_LSB:
  470. count = pit_get_count(kvm, addr);
  471. ret = count & 0xff;
  472. break;
  473. case RW_STATE_MSB:
  474. count = pit_get_count(kvm, addr);
  475. ret = (count >> 8) & 0xff;
  476. break;
  477. case RW_STATE_WORD0:
  478. count = pit_get_count(kvm, addr);
  479. ret = count & 0xff;
  480. s->read_state = RW_STATE_WORD1;
  481. break;
  482. case RW_STATE_WORD1:
  483. count = pit_get_count(kvm, addr);
  484. ret = (count >> 8) & 0xff;
  485. s->read_state = RW_STATE_WORD0;
  486. break;
  487. }
  488. }
  489. if (len > sizeof(ret))
  490. len = sizeof(ret);
  491. memcpy(data, (char *)&ret, len);
  492. mutex_unlock(&pit_state->lock);
  493. return 0;
  494. }
  495. static int speaker_ioport_write(struct kvm_io_device *this,
  496. gpa_t addr, int len, const void *data)
  497. {
  498. struct kvm_pit *pit = speaker_to_pit(this);
  499. struct kvm_kpit_state *pit_state = &pit->pit_state;
  500. struct kvm *kvm = pit->kvm;
  501. u32 val = *(u32 *) data;
  502. if (addr != KVM_SPEAKER_BASE_ADDRESS)
  503. return -EOPNOTSUPP;
  504. mutex_lock(&pit_state->lock);
  505. pit_state->speaker_data_on = (val >> 1) & 1;
  506. pit_set_gate(kvm, 2, val & 1);
  507. mutex_unlock(&pit_state->lock);
  508. return 0;
  509. }
  510. static int speaker_ioport_read(struct kvm_io_device *this,
  511. gpa_t addr, int len, void *data)
  512. {
  513. struct kvm_pit *pit = speaker_to_pit(this);
  514. struct kvm_kpit_state *pit_state = &pit->pit_state;
  515. struct kvm *kvm = pit->kvm;
  516. unsigned int refresh_clock;
  517. int ret;
  518. if (addr != KVM_SPEAKER_BASE_ADDRESS)
  519. return -EOPNOTSUPP;
  520. /* Refresh clock toggles at about 15us. We approximate as 2^14ns. */
  521. refresh_clock = ((unsigned int)ktime_to_ns(ktime_get()) >> 14) & 1;
  522. mutex_lock(&pit_state->lock);
  523. ret = ((pit_state->speaker_data_on << 1) | pit_get_gate(kvm, 2) |
  524. (pit_get_out(kvm, 2) << 5) | (refresh_clock << 4));
  525. if (len > sizeof(ret))
  526. len = sizeof(ret);
  527. memcpy(data, (char *)&ret, len);
  528. mutex_unlock(&pit_state->lock);
  529. return 0;
  530. }
  531. void kvm_pit_reset(struct kvm_pit *pit)
  532. {
  533. int i;
  534. struct kvm_kpit_channel_state *c;
  535. mutex_lock(&pit->pit_state.lock);
  536. pit->pit_state.flags = 0;
  537. for (i = 0; i < 3; i++) {
  538. c = &pit->pit_state.channels[i];
  539. c->mode = 0xff;
  540. c->gate = (i != 2);
  541. pit_load_count(pit->kvm, i, 0);
  542. }
  543. mutex_unlock(&pit->pit_state.lock);
  544. atomic_set(&pit->pit_state.pit_timer.pending, 0);
  545. pit->pit_state.irq_ack = 1;
  546. }
  547. static void pit_mask_notifer(struct kvm_irq_mask_notifier *kimn, bool mask)
  548. {
  549. struct kvm_pit *pit = container_of(kimn, struct kvm_pit, mask_notifier);
  550. if (!mask) {
  551. atomic_set(&pit->pit_state.pit_timer.pending, 0);
  552. pit->pit_state.irq_ack = 1;
  553. }
  554. }
  555. static const struct kvm_io_device_ops pit_dev_ops = {
  556. .read = pit_ioport_read,
  557. .write = pit_ioport_write,
  558. };
  559. static const struct kvm_io_device_ops speaker_dev_ops = {
  560. .read = speaker_ioport_read,
  561. .write = speaker_ioport_write,
  562. };
  563. /* Caller must hold slots_lock */
  564. struct kvm_pit *kvm_create_pit(struct kvm *kvm, u32 flags)
  565. {
  566. struct kvm_pit *pit;
  567. struct kvm_kpit_state *pit_state;
  568. struct pid *pid;
  569. pid_t pid_nr;
  570. int ret;
  571. pit = kzalloc(sizeof(struct kvm_pit), GFP_KERNEL);
  572. if (!pit)
  573. return NULL;
  574. pit->irq_source_id = kvm_request_irq_source_id(kvm);
  575. if (pit->irq_source_id < 0) {
  576. kfree(pit);
  577. return NULL;
  578. }
  579. mutex_init(&pit->pit_state.lock);
  580. mutex_lock(&pit->pit_state.lock);
  581. spin_lock_init(&pit->pit_state.inject_lock);
  582. pid = get_pid(task_tgid(current));
  583. pid_nr = pid_vnr(pid);
  584. put_pid(pid);
  585. init_kthread_worker(&pit->worker);
  586. pit->worker_task = kthread_run(kthread_worker_fn, &pit->worker,
  587. "kvm-pit/%d", pid_nr);
  588. if (IS_ERR(pit->worker_task)) {
  589. mutex_unlock(&pit->pit_state.lock);
  590. kvm_free_irq_source_id(kvm, pit->irq_source_id);
  591. kfree(pit);
  592. return NULL;
  593. }
  594. init_kthread_work(&pit->expired, pit_do_work);
  595. kvm->arch.vpit = pit;
  596. pit->kvm = kvm;
  597. pit_state = &pit->pit_state;
  598. pit_state->pit = pit;
  599. hrtimer_init(&pit_state->pit_timer.timer,
  600. CLOCK_MONOTONIC, HRTIMER_MODE_ABS);
  601. pit_state->irq_ack_notifier.gsi = 0;
  602. pit_state->irq_ack_notifier.irq_acked = kvm_pit_ack_irq;
  603. kvm_register_irq_ack_notifier(kvm, &pit_state->irq_ack_notifier);
  604. pit_state->pit_timer.reinject = true;
  605. mutex_unlock(&pit->pit_state.lock);
  606. kvm_pit_reset(pit);
  607. pit->mask_notifier.func = pit_mask_notifer;
  608. kvm_register_irq_mask_notifier(kvm, 0, &pit->mask_notifier);
  609. kvm_iodevice_init(&pit->dev, &pit_dev_ops);
  610. ret = kvm_io_bus_register_dev(kvm, KVM_PIO_BUS, KVM_PIT_BASE_ADDRESS,
  611. KVM_PIT_MEM_LENGTH, &pit->dev);
  612. if (ret < 0)
  613. goto fail;
  614. if (flags & KVM_PIT_SPEAKER_DUMMY) {
  615. kvm_iodevice_init(&pit->speaker_dev, &speaker_dev_ops);
  616. ret = kvm_io_bus_register_dev(kvm, KVM_PIO_BUS,
  617. KVM_SPEAKER_BASE_ADDRESS, 4,
  618. &pit->speaker_dev);
  619. if (ret < 0)
  620. goto fail_unregister;
  621. }
  622. return pit;
  623. fail_unregister:
  624. kvm_io_bus_unregister_dev(kvm, KVM_PIO_BUS, &pit->dev);
  625. fail:
  626. kvm_unregister_irq_mask_notifier(kvm, 0, &pit->mask_notifier);
  627. kvm_unregister_irq_ack_notifier(kvm, &pit_state->irq_ack_notifier);
  628. kvm_free_irq_source_id(kvm, pit->irq_source_id);
  629. kthread_stop(pit->worker_task);
  630. kfree(pit);
  631. return NULL;
  632. }
  633. void kvm_free_pit(struct kvm *kvm)
  634. {
  635. struct hrtimer *timer;
  636. if (kvm->arch.vpit) {
  637. kvm_io_bus_unregister_dev(kvm, KVM_PIO_BUS, &kvm->arch.vpit->dev);
  638. kvm_io_bus_unregister_dev(kvm, KVM_PIO_BUS,
  639. &kvm->arch.vpit->speaker_dev);
  640. kvm_unregister_irq_mask_notifier(kvm, 0,
  641. &kvm->arch.vpit->mask_notifier);
  642. kvm_unregister_irq_ack_notifier(kvm,
  643. &kvm->arch.vpit->pit_state.irq_ack_notifier);
  644. mutex_lock(&kvm->arch.vpit->pit_state.lock);
  645. timer = &kvm->arch.vpit->pit_state.pit_timer.timer;
  646. hrtimer_cancel(timer);
  647. flush_kthread_work(&kvm->arch.vpit->expired);
  648. kthread_stop(kvm->arch.vpit->worker_task);
  649. kvm_free_irq_source_id(kvm, kvm->arch.vpit->irq_source_id);
  650. mutex_unlock(&kvm->arch.vpit->pit_state.lock);
  651. kfree(kvm->arch.vpit);
  652. }
  653. }